剑指offer 22:链表中倒数第k个节点 -- java、python实现

题目

• 剑指offer 22. 链表中倒数第k个节点

---

输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。

例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。

示例：

给定一个链表: 1->2->3->4->5, 和 k = 2.

返回链表 4->5.

题解

法1 – java实现

• 1.遍历1遍链表,记住有多个元素
• 2.再次遍历链表并判断当前位置是否是倒数第k个位置

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode getKthFromEnd(ListNode head, int k) {
        int i = 0;
        ListNode p = head;
        int len = getLength(head);
        while(p != null){
            if(i == (len - k)){
                head = p;
                break;
            }
            i++;
            p = p.next;
        }
        return head;
    }
    public int getLength(ListNode head){
        ListNode p = head;
        int len = 0;
        while(p != null){
            len++;
            p = p.next;
        }
        return len;
    } 
}

法2（官方题解,妙啊） – java、python实现

• 定义指针p和prek，表示prek指向p的前k个节点的元素
• 当p遍历结束时，prek即指向倒数第k个元素

1.java实现

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode getKthFromEnd(ListNode head, int k) {
        ListNode p = head;
        ListNode prek = head;
        for(int i = 0; i < k; i++){
            p = p.next; 
        }
        while(p!=null){
            p = p.next;
            prek = prek.next;
        }
        return prek;
    } 
}

2.python实现

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def getKthFromEnd(self, head: ListNode, k: int) -> ListNode:
        prek,p = head,head
        # python的for循环有点东西
        for _ in range(k):
            p = p.next
        while p!= None:
            p = p.next
            prek = prek.next
        return prek