HJ51 输出单向链表中倒数第k个结点

描述
输入一个单向链表，输出该链表中倒数第k个结点，链表的倒数第1个结点为链表的尾指针。

链表结点定义如下：

struct ListNode

{

int       m_nKey;

ListNode* m_pNext;

};



正常返回倒数第k个结点指针，异常返回空指针

本题有多组样例输入。


输入描述：
输入说明
1 输入链表结点个数
2 输入链表的值
3 输入k的值

输出描述：
输出一个整数

示例1
输入：
8
1 2 3 4 5 6 7 8
4

输出：
5

#include <stdio.h>
#include <malloc.h>

struct list
{
    int val;
    struct list *next;
};

int main(void)
{
    int cnt;
    int i;
    int k;
    struct list *p, *q, *head;
    while(scanf("%d", &cnt) != EOF)
    {
        head = (struct list *)malloc(sizeof(struct list));
        p = head;
        scanf("%d", &(p->val));
        for(i=0; i<cnt-1; i++)
        {
            p->next = (struct list *)malloc(sizeof(struct list));
            p = p->next;
            scanf("%d", &(p->val));
        }
        p->next = NULL;
        scanf("%d", &k);
        
        // 在此处判断是否非法
        if(k>cnt || k==0)
        {
            printf("0\n");
            goto END;
        }
        
        p = head;
        for(i=0; i<cnt-k; i++)
        {
            p = p->next;
        }
        
        printf("%d\n", p->val);
END:
        p = head;
        while(p)
        {
            q = p->next;
            free(p);
            p = q;
        }
    }
    return 0;
}