给定一系列正整数,请设计一个尽可能高效的算法,查找倒数第K个位置上的数字。
输入格式:
输入首先给出一个正整数K,随后是若干正整数,最后以一个负整数表示结尾(该负数不算在序列内,不要处理)。
输出格式:
输出倒数第K个位置上的数据。如果这个位置不存在,输出错误信息NULL
。
输入样例:
4 1 2 3 4 5 6 7 8 9 0 -1
输出样例:
7
#include <cstdio>
#include <vector>
using namespace std;
int main(){
int K, cnt = 0, num;
vector<int> V;
scanf("%d", &K);
while(1){
scanf("%d", &num);
if(num >= 0)
V.push_back(num), cnt++;
else break;
}
if(cnt-K < 0) printf("NULL\n");
else printf("%d\n", V[cnt-K]);
return 0;
}