14、剑指offer--链表中倒数第k个结点

题目描述

输入一个链表，输出该链表中倒数第k个结点。

 

解题思路：

方法一：

定义两个指针，一个指针先向前移动k-1步（在移动过程中判断是否越界）

第二个指针指向头，然后两个指针同时向前移动，第一个指针指向最后一个节点的时候，第二个指针指向第k个结点

 

方法二：先全部遍历一遍，找到链表中结点数目，然后进行k和结点数目length的比较，把特殊错误的情况处理出来，然后再用一中的方法用两个指针，找到倒数第k个结点。

#include <iostream>
#include <malloc.h>
using namespace std;
struct ListNode {
    int val;
    struct ListNode *next;
    ListNode(int x) :
            val(x), next(NULL) {
    }
};
ListNode *CreateList(int n)
{
    ListNode *head;
    ListNode *p,*pre;
    int i;
    head=(ListNode *)malloc(sizeof(ListNode));
    head->next=NULL;
    pre=head;
    for(i=1;i<=n;i++)
    {
        p=(ListNode *)malloc(sizeof(ListNode));
        cin>>p->val;
        pre->next=p;
        pre=p;
    }
    p->next=NULL;

    return head->next;
}
/*-------------------------输出链表-----------------------------------*/
void PrintList(ListNode *h)
{
    ListNode *p;

    p=h;//不带空的头结点
    while(p)
    {
        cout<<p->val<<" ";
        p=p->next;
        cout<<endl;
    }
}
class Solution {
public:
    ListNode* FindKthToTail1(ListNode* pListHead, unsigned int k) {
        if(pListHead == NULL || k == 0)
            return NULL;
        ListNode *pHead = pListHead;
        ListNode *pBehind = NULL;
        for(int i=0;i<k-1;i++)
        {
            if(pHead->next!=NULL)
            {
                pHead = pHead->next;
            }
            else
            {
                return NULL;
            }
        }
        pBehind = pListHead;//当pHead指向链表第k-1和节点是，pBehind指向链表的头
        while(pHead->next!=NULL)
        {
            pHead = pHead->next;
            pBehind = pBehind->next;
        }
        return pBehind;
    }
    ListNode* FindKthToTail2(ListNode* pListHead, unsigned int k) {
        if(pListHead == NULL || k<=0)
            return NULL;
        int length = 0;
        ListNode *pNode = pListHead;
        while(pNode != NULL)
        {
            pNode = pNode->next;
            length++;
        }
        if(k == length)
            return pListHead;
        else if(k>length)
            return NULL;
        else//取k%length
        {
            ListNode *p1 = pListHead;
            ListNode *p2 = pListHead;
            for(int i=0;i<(k-1)%length;i++)
            {
                p2 = p2->next;
            }
            while(p2->next != NULL)
            {
                p1 = p1->next;
                p2 = p2->next;
            }
            return p1;
        }

    }
};
int main()
{
    int n1;
    int k;
    ListNode *h1;
    cout<<"输入链表1的结点数目"<<endl;
    cin>>n1;
    h1 = CreateList(n1);
    cout<<"链表1为："<<endl;
    PrintList(h1);
    cout<<"输入需要的k"<<endl;
    cin>>k;
    Solution s;
    cout<<"方法一：倒数第"<<k<<"个结点的值为"<<endl;
    cout<<s.FindKthToTail1(h1,k)->val<<endl;
    cout<<"方法二：倒数第"<<k<<"个结点的值为"<<endl;
    cout<<s.FindKthToTail2(h1,k)->val<<endl;
    return 0;
}