参考自:《剑指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);
ListNode result = FindKthToTail(head, 3); //4
System.out.println(result.val);
}
public 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;
}
}