给定一系列正整数,请设计一个尽可能高效的算法,查找倒数第K个位置上的数字。
输入格式:
输入首先给出一个正整数K,随后是若干非负整数,最后以一个负整数表示结尾(该负数不算在序列内,不要处理)。
输出格式:
输出倒数第K个位置上的数据。如果这个位置不存在,输出错误信息NULL。
输入样例:
4 1 2 3 4 5 6 7 8 9 0 -1
输出样例:
7
解题思路:
首先,我们使用一个list容器来存储输入的非负整数序列。我们通过循环读取非负整数,并将它们逐个插入到list容器中,直到读取到一个负整数为止。
接下来,我们需要计算出倒数第k个位置上的索引,可以通过用链表长度减去k来得到。如果y大于等于0,则表示倒数第k个位置存在。我们使用auto it = link.end();将迭代器it指向list容器的末尾,然后使用advance(it, -k);将迭代器it向前移动k个位置,最后通过*it取得倒数第k个位置上的数字。
如果y小于0,表示倒数第k个位置不存在。此时我们输出错误信息"NULL"。
完整代码:
#include <iostream>
#include <list>
#include <iterator>
using namespace std;
int main() {
list<int> link;
int k;
cin >> k;
for (int i = 0; ; i++) {
int n;
cin >> n;
if (n >= 0) {
link.push_back(n);
} else {
break;
}
}
int y = link.size() - k;
if (y >= 0) {
auto it = link.end();
advance(it, -k);
cout << *it << endl;
}
else {
cout << "NULL" << endl;
}
return 0;
}