剑指Offer

剑指Offer - 九度1517 - 链表中倒数第k个结点
2013-11-30 02:57

题目描述：

输入一个链表，输出该链表中倒数第k个结点。
(hint: 请务必使用链表。)

输入：

输入可能包含多个测试样例，输入以EOF结束。
对于每个测试案例，输入的第一行为两个整数n和k(0<=n<=1000, 0<=k<=1000)：n代表将要输入的链表元素的个数,k代表要查询倒数第几个的元素。
输入的第二行包括n个数t(1<=t<=1000000)：代表链表中的元素。

输出：

对应每个测试案例，
若有结果，输出相应的查找结果。否则，输出NULL。

样例输入：

5 2
1 2 3 4 5
1 0
5

样例输出：

4
NULL

题意分析：
　　给定一个单向链表，找出倒数第k个节点并输出节点值。可以先让一个指针p1从链表头开始跑k歩，然后指针p2指向链表头，两指针同步向前走，当p1走到NULL时，p2即指向倒数第k点。
　　不过，既然是处理链表，必然要步步为营，检查指针是否指向NULL。对一切“->”进行指诊检查，确保不发生内存错误。野指针是严重错误，面试里敢犯这等错误的话，估计Offer也就无缘了。

// 651911    zhuli19901106    1517    Accepted    点击此处查看所有case的执行结果    1552KB    1018B    100MS
// 201311152057
#include <cstdio>
#include <cstdlib>
using namespace std;

struct Node{
    int val;
    struct Node *next;
    Node(int _val = 0): val(_val), next(NULL){}
};

int main()
{
    int n, k, i;
    int tmp;
    struct Node *root = NULL;
    struct Node *tail;
    struct Node *p1, *p2;
    
    root = new Node(0);
    while(scanf("%d%d", &n, &k) == 2){
        tail = root;
        for(i = 0; i < n; ++i){
            scanf("%d", &tmp);
            tail->next = new Node(tmp);
            tail = tail->next;
        }
        
        if(k > n || k < 1 || root->next == NULL){
            printf("NULL
");
            continue;
        }
        
        p1 = p2 = root->next;
        while(k > 0 && p1 != NULL){
            --k;
            p1 = p1->next;
        }
        while(p1 != NULL){
            p1 = p1->next;
            p2 = p2->next;
        }
        printf("%d
", p2->val);
        
        while(root->next != NULL){
            tail = root->next;
            root->next = tail->next;
            delete tail;
        }
    }
    
    delete root;
    root = NULL;
    
    return 0;
}