给定一个单向链表,查找出从尾部开始的第k个节点,也就是倒数第k个节点,这里假定是从1开始计数的,即尾节点是第一个节点。
如图,假如要查找的是倒数第3个节点,一种思路就是从头结点开始遍历,遍历到尾部,得出链表的长度n,然后再从头开始遍历到 n-k+1个节点,就是倒数第k个,这么做要遍历链表两次。
还有一个思路,定义两个指针,p1st,p2d,开始都指向头结点,如果p1st指向尾部节点时,刚好p2d指向倒数第k个节点,那么p1st和p2nd之间就间隔k-1,所以就让p1st先走k-1,然后p2nd再从头结点开始遍历,同时p1st继续遍历,这样在p1st到达尾部时,p2nd刚好指向倒数第k个节点。
#include <iostream>
#include "ListCommon.h"
using namespace std;
ListNode* searchKthNodeFromTail(ListNode* pHead, int kth){
//输入了非法值,
if(pHead == NULL || kth == 0){
return NULL;
}
PrintList(pHead);
ListNode* p1stHead = pHead;
ListNode* p2ndHead = pHead;
for(int i = 0; i < kth-1; i++){
if(p1stHead->m_pNext != NULL){
p1stHead = p1stHead->m_pNext;
}else{//要查找的节点在超出了尾部范围。
return NULL;
}
}
while(p1stHead->m_pNext != NULL){//同时移动
p1stHead = p1stHead->m_pNext;
p2ndHead = p2ndHead->m_pNext;
}
cout << "the "<<kth<< "th node from tail is :";
PrintListNode(p2ndHead);
return p2ndHead;
}
int main(int argc,char* args[]){
ListNode* pNode1 = CreateListNode(1);
ListNode* pNode2 = CreateListNode(2);
ListNode* pNode3 = CreateListNode(3);
ListNode* pNode4 = CreateListNode(4);
ListNode* pNode5 = CreateListNode(5);
ConnectListNodes(pNode1,pNode2);
ConnectListNodes(pNode2,pNode3);
ConnectListNodes(pNode3,pNode4);
ConnectListNodes(pNode4,pNode5);
searchKthNodeFromTail(pNode1,4);//查找倒数第4个节点
DestoryList(pNode1);
return 0;
}
测试结果:
PC:~/algorithm$ g++ ListCommon.cpp SearchKthNodeFromTail.cpp -o SearchKthNodeFromTail
PC:~/algorithm$ ./SearchKthNodeFromTail
print list begin ---
1
2
3
4
5
print list end
the 4th node from tail is :The value of node is 2