在連結串列中查詢倒數第k個元素
阿新 • • 發佈:2019-02-12
第一種思路:
- 首先對連結串列進行遍歷,計算出連結串列總共的元素個數
- 然後再次遍歷連結串列,找到第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;
}