1366：栈的压入、弹出序列 @jobdu

题目1366：栈的压入、弹出序列

时间限制：1 秒

内存限制：32 兆

特殊判题：否

提交：1157

解决：426

题目描述：

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列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。

样例输入：

51 2 3 4 54 5 3 2 151 2 3 4 54 3 5 1 2

样例输出：

YesNo

答疑：

解题遇到问题?分享解题心得?讨论本题请访问： http://t.jobdu.com/thread-8089-1-1.html

类似贪心的策略，一旦有机会就一直弹栈

import java.io.BufferedInputStream;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.Scanner;
import java.util.Stack;


public class S22 {
	public static void main(String[] args) throws FileNotFoundException {
		BufferedInputStream in = new BufferedInputStream(new FileInputStream("in.in"));
		System.setIn(in);
		Scanner cin = new Scanner(System.in);
		
		while (cin.hasNextInt()) {
			int n = cin.nextInt();
			int[] input = new int[n];
			int[] out = new int[n];
			for(int i=0; i<n; i++){
				input[i] = cin.nextInt();
			}
			for(int i=0; i<n; i++){
				out[i] = cin.nextInt();
			}
			valid(input, out);
		}
	}

	private static void valid(int[] input, int[] out) {
		Stack<Integer> stack = new Stack<Integer>();
		int i=0, o=0;
		while(true){
			if(i>=input.length && o>=out.length && stack.isEmpty()){
				System.out.println("Yes");
				break;
			}
			
			if(i < input.length){
				stack.push(input[i]);
				i++;
				while(!stack.isEmpty() && stack.peek()==out[o]){		// 不停地检查看是否有机会弹栈
					stack.pop();
					o++;
				}
			}else{
				if(o < out.length){
					if(stack.peek()!=out[o]){
						System.out.println("No");
						break;
					}
					if(stack.isEmpty()){
						System.out.println("No");
						break;
					}
					if(stack.peek() == out[o]){
						stack.pop();
						o++;
					}
				}
			}
		}
	}
}