题目
思路
解法一
既然需要输出第k个节点,就要先找到尾结点,然后从尾结点向前遍历。但是题目给的链表结构是单向链表,并不能找到某个节点的前向节点。因此需要借助额外的存储结构实现。涉及到倒序,栈就是一个不错的选择。我们可以从头遍历链表将每个节点压入栈中,然后再弹出k-1个节点。最后返回时再弹出第k个节点
public ListNode GetKthFromEnd(ListNode head, int k) {
if (head == null)
return head;
Stack<ListNode> stack = new();
ListNode cur = head;
while (cur != null)
{
stack.Push(cur);
cur = cur.next;
}
if (k > stack.Count)
return null;
for (int i = 1; i < k; i++)
{
stack.Pop();
}
return stack.Pop();
}
解法二
前面那种思路必须要先完整遍历一遍链表,然后再反向遍历k个元素,才能找到要求的节点。那么能否在第一遍遍历中就找到倒数第k个节点呢?答案肯定的。只需要两个指针就能轻松做到这一点。
首先定义两个指针「left」、「right」都指向头结点的位置
然后在遍历过程中,先让右指针走k-1步,然后左右指针再一起右移。最后当右指针遍历完成时,左指针就停留在倒数第k个位置
代码如下
public ListNode GetKthFromEnd2(ListNode head, int k)
{
if (head == null)
return head;
ListNode left = head;
ListNode right = head;
while (right.next != null)
{
if (k-1 <= 0)
{
left = left.next;
}
right = right.next;
k--;
}
return left;
}