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