链表中倒数最后k个结点
描述
输入一个长度为 n 的链表,设链表中的元素的值为 ai ,返回该链表中倒数第k个节点。
如果该链表长度小于k,请返回一个长度为 0 的链表。
数据范围:0≤n≤10 5 ,0≤ai ≤
1
0
9
10^9
109 ,0≤k≤
1
0
9
10^9
109
要求:空间复杂度 O(n),时间复杂度 O(n)
进阶:空间复杂度 O(1),时间复杂度 O(n)
例如输入{1,2,3,4,5},2时,对应的链表结构如下图所示:
其中蓝色部分为该链表的最后2个结点,所以返回倒数第2个结点(也即结点值为4的结点)即可,系统会打印后面所有的节点来比较。
示例1
输入:{1,2,3,4,5},2
返回值:{4,5}
说明:返回倒数第2个节点4,系统会打印后面所有的节点来比较。
示例2
输入:{2},8
返回值:{}
思路
这是一个比较简单的题,很容易就可以想到多种思路
思路一:首先遍历所有整个链表,得到长度n,若n>k,遍历得到n-k个节点输出
思路二:将所有的节点放到数组中,取出第k个节点(空间复杂度为O(n))
思路三:将所有节点压栈,弹出顶上的k个节点(空间复杂度为O(n))
…
本题的代码采用一种比较巧妙的思路:快慢指针
设置一个快指针,先走k步
之后快慢指针同时向后移动
当快指针到达结尾时慢指针所在的位置就是倒数第k个元素
代码
public ListNode FindKthToTail (ListNode pHead, int k) {
ListNode FastPtr = pHead;
ListNode SlowPtr = pHead;
int i=0;
while(FastPtr!=null && i<k){
FastPtr = FastPtr.next;
i++;
}
if(i!=k){
return null;
}
while(FastPtr!=null){
FastPtr = FastPtr.next;
SlowPtr = SlowPtr.next;
}
return SlowPtr;
}
删除链表的倒数第n个节点
描述
给定一个链表,删除链表的倒数第 n 个节点并返回链表的头指针
例如,给出的链表为: 1→2→3→4→5,n=2.
删除了链表的倒数第 n 个节点之后,链表变为1→2→3→5.
数据范围: 链表长度 0≤n≤1000,链表中任意节点的值满足 0≤val≤100
要求:空间复杂度 O(1),时间复杂度O(n)
备注:
题目保证 n 一定是有效的
示例1
输入:{1,2},2
返回值:{2}
思路
与上一题相同,只是需要单独记录下slow指针的上一指针。为了防止删除的节点是链表的第一个节点,引入了一个dummy节点。
ListNode dummy = new ListNode(-1);
dummy.next = head;
ListNode LastSlow = null;
ListNode FastPtr = dummy;
ListNode SlowPtr = dummy;
int i=0;
while(FastPtr!=null && i<n){
FastPtr = FastPtr.next;
i++;
}
if(i!=n){
return null;
}
while(FastPtr!=null){
LastSlow = SlowPtr;
FastPtr = FastPtr.next;
SlowPtr = SlowPtr.next;
}
LastSlow.next = SlowPtr.next;
return dummy.next;
此外,也可以考虑判断fast.next是否为空,这样可以直接得到slow的上一节点,只是要把边界条件考虑清楚