给定一系列正整数,请设计一个尽可能高效的算法,查找倒数第K个位置上的数字。
输入格式:
输入首先给出一个正整数K,随后是若干正整数,最后以一个负整数表示结尾(该负数不算在序列内,不要处理)。
输出格式:
输出倒数第K个位置上的数据。如果这个位置不存在,输出错误信息NULL
。
输入样例:
4 1 2 3 4 5 6 7 8 9 0 -1
输出样例:
7
#include<iostream>
#include<stack>
using namespace std;
int main()
{
stack<int> q;
int flag=1,k,count=0;
cin>>k;
while(flag)
{
int t;
cin>>t;
if(t<0)
flag=0;
else
{
q.push(t);
count++;
}
}
if(k>count)
cout<<"NULL";
else
{
for(int i=1;i<k;i++)
q.pop();
cout<<q.top();
}
}