给你一个链表,删除链表的倒数第 n 个结点,并且返回链表的头结点。
示例 1:
输入:head = [1,2,3,4,5], n = 2
输出:[1,2,3,5]
示例 2:
输入:head = [1], n = 1
输出:[]
示例 3:
输入:head = [1,2], n = 1
输出:[1]
提示:
链表中结点的数目为 sz
1 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz
进阶:你能尝试使用一趟扫描实现吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-nth-node-from-end-of-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路
想要一趟扫描完成,最容易想到的方法就是一个指针数组。因为最大长度只有30,所以建立一个大小为30的数组就足矣了(一般人都凑个整建32个),毕竟一个指针才几个字节大小
用一个指针数组为这个链表的每个元素都建立一个索引,这样就能用类似数组的方法来操作链表了。如此,就能用一次遍历完成这道题的解答
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* removeNthFromEnd(ListNode* head, int n) {
ListNode *p[30],*h=head;
int i=0,size=0;
while(h!=nullptr){
p[i]=h;
i++;
size++;
h=h->next;
}
int index=size-n;
// cout<<index<<":"<<p[index]->val; //test
if(index==0){
head=head->next;
}
else if(n==1){
p[size-2]->next=nullptr;
}
else{
p[index-1]->next=p[index+1];
}
return head;
}
};