如何找出单链表中倒数第 k 个元素,有三种方法可供参考。
1. 寻找正数第 n-k+1 个元素
该方法思想是
(1)首先遍历一遍单链表,求出整个单链表的长度 n;
(2)将倒数第 k 个,转换为正数第 n - k +1个,然后遍历一次找到第 n - k+1 个 元素就可以得到结果。
具体实现如下:
public class FindReverseEle {
public int getLength(Node head) {//求链表长度
int length = 0;
Node tmp = head;
while(tmp != null) {
length++;
tmp = tmp.next;
}
return length;
}
public Node findElem(Node head, int k) {
int length = 0;
Node tmp = head;
Node pNode = head;
int i = 1;
while(tmp != null) {
length++;
tmp = tmp.next;
}
if(k < 1 || k > length) {
return null;
}
while(i < (length-k+1) && pNode != null) {
pNode = pNode.next;
i++;
}
return pNode;
}
}
该方法存在一个问题:
需要对链表进行两次遍历,第一次遍历用于求解单链表的长度,第二次遍历用于查找正数第 n - k 个元素。
2. 测试法
显然以上方法还可以进行优化。如果沿从头至尾的方向从链表中的某个元素开始,遍历 k 个元素后刚好达到链尾,那么该元素就是要找的倒数第 k 个元素。根据这一性质,可设置如下算法:
从头结点开始,依次对链表的每一个结点元素进行这样的测试,遍历 k 个元素,看是否到达表尾,直到找到那个倒数第 k 个元素。程序代码如下:
public Node multiTest(Node head, int k) {
Node tmp = head;
int len = getLength(tmp);
if(k < 1 || k > len) {
return null;
}
while(tmp != null) {
int count = 0;
Node pNode = tmp;
while(count < k && pNode != null) {
count++;
pNode = pNode.next;
}
if(count == k && pNode == null) {
return tmp;
}
tmp = tmp.next;
}
return null;
}
此种方法要对链表中的大部分元素,都要遍历 k 个元素,,如果链表的长度为 n,该算法的时间复杂度为 O( kn ),效率太低。
3. 两个指针(一个比另一个快 k 步)
下面介绍一种更高效的方法,只需要一次遍历链表即可查找到倒数第 k 个元素。由于单链只能从头到尾依次访问链表的各个节点,因此,如果要找出链表的倒数第 k 个元素的话,也只能从头到尾进行遍历查找。
该方法的思想是: 在查找过程中,设置两个指针,让其中一个指针比另一个指针先前移 k 步,然后两个指针同时往前移动。循环直到先行的指针值为 null 时,另一个指针的位置就是所要找的位置。程序代码如下:
public Node findElemDoblePointer(Node head, int k) {
if(k < 1 || k > getLength(head)) {
return null;
}
Node p1 = head;
Node p2 = head;
for(int i = 0; i < k ; i++) {
p1 = p1.next;
}
while(p1 != null) {
p1 = p1.next;
p2 = p2.next;
}
return p2;
}