首页 > 代码库 > 剑指OFFER之链表中倒数第k个节点(九度OJ1517)

剑指OFFER之链表中倒数第k个节点(九度OJ1517)

题目描述:

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

解题思路:

  首先我们考虑到,用普通的思想,获取链表的长度,然后输出第length-k个元素的值,就是倒数第k个元素的值了。但是考虑到面试的技巧,这种方法还是不行的,但是做题时可以AC的。代码如下,把所有的链表相关的东西删除,仍然可以AC。

#include <stdio.h>
#include <stdlib.h>
typedef struct node{
    int number;
    struct node * next;
}Node;
int main(){
    int n,k,i,temp;
    int flag = 0;
    Node *link = (Node *)malloc(sizeof(Node));
    link->next = NULL;
    while(scanf("%d %d",&n,&k)!=EOF && n>=0 && n<=1000 && k>=0 && k<=1000 ){
         flag = 0;
        for(i=0;i<n;i++){
            scanf("%d",&temp);
            if(i == n-k){
                flag = 1;
                printf("%d\n",temp);
            }
            Node *n = (Node *)malloc(sizeof(Node));
            n->next = link->next;
            link->next = n;
            n->number = temp;
        }
        if(!flag)
            printf("NULL\n");
    }
    return 0;
}
/**************************************************************
    Problem: 1517
    User: xhalo
    Language: C
    Result: Accepted
    Time:100 ms
    Memory:2496 kb
****************************************************************/

  但是考虑的复杂点,如果时间复杂度为O(n)呢,一次扫描是不可能获取长度,又输出的

  因此采取判断链表循环的思路,采用两个指针,第一个指针提前k-1步向下走,第二个指针再随着第一个指针一直走。知道第一个指针指向末尾,第二个指针的元素的值,便是我们所要求得的值。

    p1 = link->next;
    for(i=1;i<k;i++){
        p1 = p1->next;
    }
    p2 = link->next;
    while(p1->next!=NULL){
        p1 = p1->next;
        p2 = p2->next;
    }
    return p2->number;

  但是考虑到特殊情况:

1 如果输入的k是0,也就是说是个非法的值。解决办法

2 如果链表为空,根本找不到要查找的值。解决办法

 

    if(link->next == NULL || k == 0)
        return -1;

 

3 如果p1已经到了链表的末尾,而p2还没有开始,也就是说,链表的长度根本不够k个,怎么办呢?解决办法

    p1 = link->next;
    for(i=1;i<k;i++){
        if(p1->next == NULL)
            return -1;
        p1 = p1->next;
    }

全部代码:

#include <stdio.h>
#include <stdlib.h>
typedef struct node{
    int number;
    struct node * next;
}Node;
int getK(Node *link,int k);
int main(){
    int n,k,i,temp;
    int flag;
    while(scanf("%d %d",&n,&k)!=EOF && n>=0 && n<=1000 && k>=0 && k<=1000 ){
        Node *link = (Node *)malloc(sizeof(Node));
        link->next = NULL;
         flag = 0;
         Node *tail;
         tail = link;
        for(i=0;i<n;i++){
            scanf("%d",&temp);
            Node *n = (Node *)malloc(sizeof(Node));
            n->next = tail->next;
            tail->next = n;
            tail = tail->next;
            n->number = temp;
        }
        int numberK = getK(link,k);
        numberK ==-1?printf("NULL\n"):printf("%d\n",numberK);
    }
    return 0;
}
int getK(Node *link,int k){
    Node *p1,*p2;
    int i;
    if(link->next == NULL || k == 0)
        return -1;
    p1 = link->next;
    for(i=1;i<k;i++){
        if(p1->next == NULL)
            return -1;
        p1 = p1->next;
    }
    p2 = link->next;
    while(p1->next!=NULL){
        p1 = p1->next;
        p2 = p2->next;
    }
    return p2->number;
}
/**************************************************************
    Problem: 1517
    User: xhalo
    Language: C
    Result: Accepted
    Time:100 ms
    Memory:2496 kb
****************************************************************/