PTA - 求链式线性表的倒数第K项(堆栈解法)

求链式线性表的倒数第K项(堆栈解法)

题目：给定一系列正整数，请设计一个尽可能高效的算法，查找倒数第K个位置上的数字。

输入格式：

输入首先给出一个正整数K，随后是若干正整数，最后以一个负整数表示结尾（该负数不算在序列内，不要处理）。

输出格式：

输出倒数第K个位置上的数据。如果这个位置不存在，输出错误信息NULL。

输入样例：

4 1 2 3 4 5 6 7 8 9 0 -1

输出样例：

7

---

#include <iostream>
#include <strack>

using namespace std;

int main()
{
	int x;
	cin >> x;
	
	stack<int> INT_STACK;
	int count = 0;
	
	int n;
	while(cin >> n)
	{
		if (n < 0)  // 负整数表示结尾(该负数不算在序列内，不要处理)。  
			break;
		
		else
		{
			INT_STACK.push(n);  // 向堆栈压入这若干个整数 
			count++;            // 整数的个数 
		} 
	}
	
	if (x > count)
		cout << "NULL" << endl;
	
	else
	{
		for (int i = 1; i < x; i++)
		{
			INT_STACK.pop();   // 移除栈顶元素，直到到达x位置所在的元素为止 
		}
		
		cout << INT_STACK.top();  // 返回栈顶元素，记x所在位置的元素的值 
	}
	
	return 0;
}