在鏈表中查找倒數第k個元素

第一種思路：

• 首先對鏈表進行遍歷，計算出鏈表總共的元素個數
• 然後再次遍歷鏈表，找到第n-k個元素，即爲倒數第k個元素
• 以上方法需要對鏈表進行遍歷兩次。

第二種思路：

• 兩個鏈表指針，前一個指針指向第k+i個元素，後一個指針指向第i個元素，兩個指針間隔k-2個元素
• 這樣只需遍歷一次即可找到倒數第k個元素

注意：

• 鏈表使用結束後，記得釋放每個申請的內存。這很重要！！！

//文件名：listnode.h
#pragma once
#ifndef LISTNODE_H
#define LISTNODE_H

#include <iostream>

using namespace std;

//定義鏈表結構體
template<class T>
struct ListNode
{
    T val;
    ListNode* next;

    ListNode():val(0),next(NULL){}
    ListNode(T num):val(num),next(NULL){}
};

// template<class T>
// ListNode<T>* Init(ListNode<T>* p)
// {
//     //返回頭結點
//     p = new ListNode<T>();
//     return p;
// }

//構造鏈表
template<class T>
ListNode<T>* InitListNode(ListNode<T>* p,int n)
{
    ListNode<T>* head = p;
    for (int i=0;i<n;i++)
    {
        ListNode<T>* pNode = new ListNode<T>(i);
        p->next = pNode;
        p = p->next;
    }

    return head;
}

//釋放鏈表內存
template<class T>
void DeleteNode(ListNode<T>* p)
{
    ListNode<T>* head = p;

    //釋放鏈表內存的過程相當於逐個刪除頭結點後面一個幾點
    while (head->next != NULL)
    {
        cout<<"刪除節點:"<<head->next->val<<endl;;
        ListNode<T>* dp = p->next;
        head->next = dp->next;
        delete dp;
        dp = NULL;
    }

    //刪除頭結點
    if (head)
    {
        delete head;
        head = NULL;
    }
}

//查找倒數第k個元素
template<class T>
T FindLastKElem(ListNode<T>* p,int k)
{
    ListNode<T>* pFront;
    ListNode<T>* pBack;

    pFront = p;
    pBack = p;
    //倒數第3個，則兩個指針之間有1個元素
    for (int i=0;i<k-1;i++)
    {
        if (pFront->next != NULL)
            pFront = pFront->next;//向後移動k個
        else
        {
            cout<<"沒有倒數第k個元素"<<endl;
            return (T)0;
        }
    }

    while (pFront->next != NULL)
    {
        pFront = pFront->next;
        pBack = pBack->next;
    }

    return pBack->val;
}

#endif

// LinkList.cpp : 定義控制檯應用程序的入口點。
//

#include "stdafx.h"
#include "listnode.h"
#include <Windows.h>

int _tmain(int argc, _TCHAR* argv[])
{
    for (int i=0;i<100;i++)
    {
        ListNode<int> *head = new ListNode<int>();

        InitListNode(head,10);
        int m = FindLastKElem(head,3);
        cout<<m<<endl;

        //釋放內存
        DeleteNode(head);

        Sleep(50);
    }

    return 0;
}