栈的压入压出

我不是女神ヾ 2022-03-20 02:46 181阅读 0赞

**何海涛：《剑指Offer：名企面试官精讲典型编程题》：九度OJ**

**题目描述：[http://ac.jobdu.com/problem.php?cid=1039&pid=8][http_ac.jobdu.com_problem.php_cid_1039_pid_8]**

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序，序列4，5,3,2,1是该压栈序列对应的一个弹出序列，但4,3,5,1,2就不可能是该压栈序列的弹出序列。

**输入：**

每个测试案例包括3行：

第一行为1个整数n(1<=n<=100000)，表示序列的长度。

第二行包含n个整数，表示栈的压入顺序。

第三行包含n个整数，表示栈的弹出顺序。

**输出：**

对应每个测试案例，如果第二个序列是第一个序列的弹出序列输出Yes，否则输出No。

**样例输入：**

1 2 3 4 5

4 5 3 2 1

1 2 3 4 5

4 3 5 1 2

**样例输出：**

Yes

代码AC：

思想：扫描 + 入栈 + 弹栈：当前的输出==栈顶元素，弹栈，若当前输出的不是栈顶元素，那么就要将元素入栈直到此元素，如果入栈中没有发现此元素，那么就是错误的序列。

#include <stdio.h>
    #include <stdlib.h>
    
    int main()
    {
        int n, *stack1, *stack2, *stack, i, j, top = -1, flag, f;
        
        while( scanf("%d", &n) != EOF )
        {
               stack1 = ( int * )malloc( sizeof( int ) * n );
               stack2 = ( int * )malloc( sizeof( int ) * n );
               stack  = ( int * )malloc( sizeof( int ) * n );
               
               for( i = 0; i < n; i++ )
               {
                    scanf("%d", &stack1[i]);
               }
               
               for( i = 0; i < n; i++ )
               {
                    scanf("%d", &stack2[i]);
               }
               
               top = -1;
               i = 0;
               j = 0;           
               flag = 1;
               
               while( stack1[i] != stack2[j] )
               {
                      stack[++top] = stack1[i];
                      
                      if( ( ++i ) >= n )
                      {
                          printf("No\n");
                          flag = 0;
                          break;
                      }
               }
               
               if( !flag )
               {
                   continue;
               }
                
               i++;
               j++;
               
               while( j < n )
               {
                      f = 0;
                      
                      if( top > -1 )
                      {
                          if( stack2[j] == stack[top] )
                          {
                              top--;
                              j++;
                          }
                          else
                          {
                              f = 1;
                          }
                      }
                      else
                      {
                          f = 1;
                      }
                      
                      if( f )
                      {
                          while( stack2[j] != stack1[i] )
                          {
                                 stack[++top] = stack1[i];
                                 i++;
                                 if( i >= n )
                                 {
                                     printf("No\n");
                                     flag = 0;
                                     break;
                                 }
                          }
                          if( flag )
                          {
                              stack[++top] = stack1[i];
                          }
                      }
                      
                      if( !flag )
                      {
                          break;
                      }
               }
               
               if( !flag )
               {
                   continue;
               }
               
               printf("Yes\n");
        }
        return 0;
    }

[http_ac.jobdu.com_problem.php_cid_1039_pid_8]: http://ac.jobdu.com/problem.php?cid=1039&pid=8