【PTA-数据结构】7-19 求链式线性表的倒数第K项 (20 分)--容器/链表/堆栈

题目

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

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

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

思路

本题要求尽可能高效的算法。
有两个思路：
一个是链表/容器存储，两个指针跑，找到所求元素：
引入结构体或者是容器存储。
容器结果：
看到其他博主使用的是头插法，而不是双指针倒序。
而且先让指针按输入的K跑起来，如果跑到了null。那么确实输出NULL，不存在，否则就输出数据

另一个考虑到既然是倒数第K项，那么就pop()出来数据。采用建立堆栈的方法。
引入stack存储。

容器结合双指针代码

#include <iostream>
#include <vector>
using namespace std;
int main()
{
    vector<int> a;
    vector<int>::iterator p,p1,p2;
    int N;
    cin>>N;
    while(true)//压入元素到容器中，遇到负数break
    {
        int t;
        cin>>t;
        if(t>=0)
        {
            a.push_back(t);
        }
        else break;
    }
    p1=p2=a.begin();
    p1+=N;//双指针之间的位置之差
    int i=0;
    for(p=a.begin();p!=a.end();++p)//获取容器内元素个数
    {
        ++i;
    }
    if(N>i)//如果查询数据超出了容器元素个数范围，抛出NULL
    {
        cout<<"NULL";
        return 0;
    }
    for(;p1!=a.end();++p1,++p2);//双指针
        cout<<*p2;
}

单向链表头插法顺序找到k

#include <iostream>
using namespace std;
typedef struct Node *node;
struct Node
{
    int num;
    node next;
};

int main()
{
    int N;
    cin>>N;
    node h;
    h=(node)malloc(sizeof(struct Node));//头节点
    h->next=NULL;
    while(true)
    {
        int t;
        cin>>t;
        if(t>=0)
        {
            node n=(node)malloc(sizeof(Node));
            n->num=t;
            n->next=h->next;//新结点追加到头结点与头结点之后结点之间
            h->next=n;//新结点追加到头结点之后
        }
        else break;
    }
    for(int i=0;i<N;++i)//按输入的倒序index，让h指针跑起来
    {
        h=h->next;
    }
    if(h)
            cout<<h->num;
    else
        cout<<"NULL";
}

堆栈存储

#include <iostream>
#include <stack>
using namespace std;
int main()
{
    stack<int> s;
    int n;
    cin>>n;
    while(true)//将数据压入栈
    {
        int t;
        cin>>t;
        if(t>=0)
        {
            s.push(t);
        }
        else break;
    }
    if(n>s.size())//越界的k
    {
        cout<<"NULL";
        return 0;
    }
    for(int i=n;i>1;--i)//将k-1个数据pop出来
    {
        s.pop();
    }
    cout<<s.top();//输出顶部
}