面试题 02.02. 返回倒数第 k 个节点

题目

实现一种算法，找出单向链表中倒数第 k 个节点。返回该节点的值。

注意：本题相对原题稍作改动

示例：
输入： 1->2->3->4->5 和 k = 2
输出： 4

说明：
给定的 k保证是有效的。

题解

public int kthToLast(ListNode head, int k) {
    int count = 0;
    ListNode node = head;
    // 计算链表长度
    while (node != null) {
        count++;
        node = node.next;
    }

    int i = 0;
    node = head;
    // 找到倒数k位置的数字
    while (node != null) {
        if (i + k == count) {
            return node.val;
        }
        node = node.next;
        i++;
    }

    return 0;
}