1. 程式人生 > >在連結串列中查詢倒數第k個元素

在連結串列中查詢倒數第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;
}