剑指Offer-题22(Java版):链表中倒数第k个结点

参考自:《剑指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;
    }
}

猜你喜欢

转载自blog.csdn.net/m0_37862405/article/details/80199214