给定一系列正整数,请设计一个尽可能高效的算法,查找倒数第K个位置上的数字。
输入格式:
输入首先给出一个正整数K,随后是若干非负整数,最后以一个负整数表示结尾(该负数不算在序列内,不要处理)。
输出格式:
输出倒数第K个位置上的数据。如果这个位置不存在,输出错误信息NULL
。
输入样例:
4 1 2 3 4 5 6 7 8 9 0 -1
输出样例:
7
栈方法:把全部元素依次压入栈中,然后弹出k个元素 ,栈顶元素就是所求元素
#include<iostream>
#include<stack>
using namespace std;
int main()
{
stack<int>S;
int k,n;
cin>>k;
while(cin>>n&&n>=0)
S.push(n);
if(S.size()<k)cout<<"NULL";
else{
for(int i=1;i<k;i++)
S.pop();
cout<<S.top();
}
return 0;
}
向量方法:与栈方法类似,不同的是vector可以直接用数组方法表示,全部元素压入容器中后,直接输出元素个数-k位置上的元素即可
#include<iostream>
#include<vector>
using namespace std;
int main()
{
vector<int>V;
int m,n;
cin>>m;
while(cin>>n&&n>=0)
V.push_back(n);
if(V.size()<m)cout<<"NULL";
else cout<<V[V.size()-m];
return 0;
}