参考自:《剑指Offer——名企面试官精讲典型编程题》
题目:链表中倒数第k个结点
输入一个链表,输出该链表中倒数第k个结点。为了符合大多数人的习惯,本题从1开始计数,即链表的尾结点是倒数第1个结点。例如一个链表有6个结点,从头结点开始它们的值依次是1、2、3、4、5、6。这个链表的倒数第3个结点是值为4的结点。
主要思路:使用双指针,前面的指针先走k-1步,后面的指针再从头结点开始移动,当前面的指针到达尾部时,后面的指针指向的就是倒数第k个结点。其中要注意结点为空,或者结点总数目小于k的特殊情况。
关键点:双指针遍历
时间复杂度:O(n)
public class FindKthNodeFromTail {
public static void main(String[] args) {
int[] data = {1, 2, 3, 4, 5, 7};
ListNode head = LinkedListData.generateDataByArray(data);
// 4
ListNode result = findKthToTail(head, 3);
System.out.println(result.val);
}
private static ListNode findKthToTail(ListNode head, int k) {
if (head == null || k <= 0) {
return null;
}
ListNode ahead = head;
//前面的指针先走k-1步
for (int i = 0; i < k - 1; i++) {
if (ahead == null) {
return null;
}
ahead = ahead.next;
}
ListNode behind = head;
//前面的指针到达尾部
while (ahead.next != null) {
behind = behind.next;
ahead = ahead.next;
}
return behind;
}
}
class ListNode {
int val;
ListNode next = null;
ListNode(int val) {
this.val = val;
}
}
class LinkedListData {
/**
* 根据数组生成链表
*
* @param data
* @return
*/
public static ListNode generateDataByArray(int[] data) {
ListNode head = new ListNode(data[0]);
ListNode currentNode = head;
for (int i = 1; i < data.length; i++) {
ListNode next = new ListNode(data[i]);
currentNode.next = next;
currentNode = next;
}
return head;
}
}