给定一系列正整数,请设计一个尽可能高效的算法,查找倒数第K个位置上的数字。
输入格式说明:
输入首先给出一个正整数K,随后是若干正整数,最后以一个负整数表示结尾(该负数不算在序列内,不要处理)。
输出格式说明:
输出倒数第K个位置上的数据。如果这个位置不存在,输出错误信息“NULL”。
样例输入与输出:
序号 | 输入 | 输出 |
1 |
4 1 2 3 4 5 6 7 8 9 0 -1 |
7 |
2 |
6 3 6 7 8 2 -2 |
NULL |
看题目的标题知道其实这是在考验链表的理解的,然而我直接就用了向量。。。只是为了过题。
#include<iostream>
#include<vector>
using namespace std;
int main(){
vector<int> vec;
int k ;
cin>>k;
int n;
while(cin>>n && n>=0){
vec.push_back(n);
}
if(k>vec.size()){
cout<<"NULL"<<endl;
}else{
cout<<vec[vec.size()-k]<<endl;
}
return 0;
}