码迷,mamicode.com
首页 > 其他好文 > 详细

栈的压入、弹出序列(剑指offer-21)

时间:2020-07-02 22:06:04      阅读:54      评论:0      收藏:0      [点我收藏+]

标签:tac   empty   lan   不可   开始   遍历   sequence   bool   比较   

题目描述

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

题目解析

借用一个额外的辅助栈
入栈1,2,3,4,5
出栈4,5,3,2,1
遍历压栈顺序,先将第一个放入栈中,这里是1,然后判断栈顶元素是不是出栈顺序的第一个元素,这里是4,很显然1≠4,所以我们继续压栈,
直到相等以后开始出栈,出栈一个元素,则将出栈顺序向后移动一位,直到不相等,这样循环等压栈顺序遍历完成,如果辅助栈还不为空,
说明弹出序列不是该栈的弹出顺序。最后辅助栈为空。如果不为空说明弹出序列不是该栈的弹出顺序。

题目解答

import java.util.*;
public class Solution {
    public boolean IsPopOrder(int [] pushA,int [] popA) {
      Stack<Integer> stack = new Stack<>();
      int pushindex = 0;
      int popindex = 0;
      
      while(pushindex < pushA.length){
          if(!stack.isEmpty() && stack.peek()==popA[popindex]){//栈顶元素等于出栈元素时,直接出栈
              stack.pop();
              popindex++;}
          else{
              stack.push(pushA[pushindex]);//栈顶元素不等于出栈元素时,再往栈里面放数据
              pushindex++;
          }
      }
        //while循环是为了防止当所有入栈的元素都压入栈的时候,栈顶元素和出栈序列的下标所指的数字没有来得及比较。
        while (!stack.isEmpty()) {
            if (stack.peek() == popA[popindex]) {
                stack.pop();
                popindex++;
            } else {
                return false;
            }
        }
        return true;
    }
}

简约写法

public boolean IsPopOrder(int[] pushSequence, int[] popSequence) {
    int n = pushSequence.length;
    Stack<Integer> stack = new Stack<>();
    for (int pushIndex = 0, popIndex = 0; pushIndex < n; pushIndex++) {
        stack.push(pushSequence[pushIndex]);
        while (popIndex < n && !stack.isEmpty() 
                && stack.peek() == popSequence[popIndex]) {
            stack.pop();
            popIndex++;
        }
    }
    return stack.isEmpty();
}

栈的压入、弹出序列(剑指offer-21)

标签:tac   empty   lan   不可   开始   遍历   sequence   bool   比较   

原文地址:https://www.cnblogs.com/yzhengy/p/13227033.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!