interview recall

给一个unsorted  positive integer array , 没有duplicate, 输出是一个integer 和 这个integer 后面第一个大于该integer的那个数,这道题可以用stack来做,

代码如下:

#include<iostream>
#include<stack>

using namespace std;

const int maxn = 100;

int a[maxn],b[maxn];
stack<int> astack;
stack<int> aposstack;
int main(){
	int n;
	cin >> n;
	for(int i = 0; i < n; ++i){
		cin >> a[i];
	}
	for(int i = 0; i < n; ++i){
		while(!astack.empty() && a[i] > astack.top()){
			int pos = aposstack.top();
			aposstack.pop();
			b[pos] = i;
			astack.pop();
		}
		astack.push(a[i]);
		aposstack.push(i);
	}
	while(!astack.empty()){
		int pos = aposstack.top();
		b[pos] = -1;
		aposstack.pop();
		astack.pop();
	}
	//print b[i];
	return 0;
}


你可能感兴趣的:(interview recall)