PTA-求链式线性表的倒数第K项

给定一系列正整数,请设计一个尽可能高效的算法,查找倒数第K个位置上的数字。

输入格式:


输入首先给出一个正整数K,随后是若干非负整数,最后以一个负整数表示结尾(该负数不算在序列内,不要处理)。

输出格式:


输出倒数第K个位置上的数据。如果这个位置不存在,输出错误信息NULL。

输入样例:


4 1 2 3 4 5 6 7 8 9 0 -1

输出样例:

7

PTA-求链式线性表的倒数第K项_第1张图片

解题思路:


首先,我们使用一个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;
}

你可能感兴趣的:(数据结构,链表,数据结构)