剑指Offer之链表中倒数第k个结点

题目描述：

输入一个链表，输出该链表中倒数第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

【代码】

/*********************************
*   日期：2013-11-20
*   作者：SJF0115
*   题号: 题目1517：链表中倒数第k个结点
*   来源：http://ac.jobdu.com/problem.php?pid=1517
*   结果：AC
*   来源：剑指Offer
*   总结：
**********************************/
#include<iostream>
#include <stdio.h>
#include <malloc.h>
#include <string.h>
using namespace std;

typedef struct ListNode{
    int value;
    struct ListNode *next;
}ListNode;

int FindKthToTail(ListNode*head,int k){
    //容错处理
    if(head == NULL || k <= 0){
        return NULL;
    }
    else{
        ListNode *p,*pre;
        //带头节点的链表
        pre = p = head->next;
        //第一个指针向前走k-1步，第二个指针保持不变
        for(int i = 0;i < k-1;i++){
            p = p->next;
        }
        //两个指针同时向前遍历
        while(p->next != NULL){
            pre = pre->next;
            p = p->next;
        }
        return pre->value;
    }
}

int main()
{
	int i,n,k;
	while(scanf("%d %d",&n,&k) != EOF){
	    ListNode *head,*p,*pre;
	    head = (ListNode*)malloc(sizeof(ListNode));
	    head->next = NULL;
        pre = head;
	    for(i = 0;i < n;i++){
	        p = (ListNode*)malloc(sizeof(ListNode));
	        scanf("%d",&p->value);
	        p->next = NULL;
            pre->next = p;
            pre = p;
	    }
	    //全部输出
	    if(n < k){
            printf("NULL\n");
	    }
	    else{
	        //倒数K个
            int num = FindKthToTail(head,k);
            if(num == NULL){
                printf("NULL\n");
            }
            else{
                printf("%d\n",num);
            }
	    }
	}
	return 0;
}