/** * Definition for singly-linked list. * struct ListNode { * int val; * ListNode *next; * ListNode(int x) : val(x), next(NULL) {} * }; */ class Solution { public: ListNode* getKthFromEnd(ListNode* head, int k) { } };完善代码
时间: 2023-08-09 09:09:24 浏览: 147
根据题目要求,我们需要实现一个函数 `getKthFromEnd`,该函数接受一个链表头节点 `head` 和一个整数 `k`,返回倒数第 k 个节点。
我们可以使用双指针的方法来解决这个问题。具体步骤如下:
1. 定义两个指针 `fast` 和 `slow`,初始时都指向链表的头节点 `head`。
2. 将 `fast` 指针向前移动 k 个位置。
3. 同时移动 `fast` 和 `slow` 指针,直到 `fast` 指针达到链表末尾(即 `fast->next` 为 NULL)。
4. 返回 `slow` 指针所指向的节点。
实现代码如下:
```cpp
ListNode* getKthFromEnd(ListNode* head, int k) {
ListNode* fast = head;
ListNode* slow = head;
// 将 fast 指针向前移动 k 个位置
for (int i = 0; i < k; i++) {
fast = fast->next;
}
// 同时移动 fast 和 slow 指针
while (fast) {
fast = fast->next;
slow = slow->next;
}
return slow;
}
```
使用例子:
```cpp
// 创建链表 1->2->3->4->5
ListNode* head = new ListNode(1);
head->next = new ListNode(2);
head->next->next = new ListNode(3);
head->next->next->next = new ListNode(4);
head->next->next->next->next = new ListNode(5);
int k = 2;
ListNode* result = getKthFromEnd(head, k);
while (result) {
cout << result->val << " ";
result = result->next;
}
```
输出结果为:
```
4 5
```
这样就实现了获取倒数第 k 个节点的功能。注意,这里假设链表中至少有 k 个节点。如果链表长度不足 k,需要根据题目要求进行处理。
阅读全文