剑指 Offer 22-链表中倒数第k个节点

题目

思路

解法一

既然需要输出第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;
}

猜你喜欢

转载自blog.csdn.net/LWR_Shadow/article/details/126864054