题目
给你一个链表,删除链表的倒数第 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]
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/remove-nth-node-from-end-of-list
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
提示:
链表中结点的数目为 sz
1 <= sz <= 30
0 <= Node.val <= 100
1 <= n <= sz
题解思路
- 删除的节点要判断一下是否是第一个或者最后一个
代码
/*** 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) {vector<ListNode*> arr;ListNode* h = head;while(head){arr.push_back(head);head = head->next;}int k = arr.size();ListNode* pp = arr[k-n];if(pp==arr[0]){h = h->next;}else if(pp==arr[k-1]){ListNode* qq = arr[k-n-1];qq->next = nullptr;}else{ListNode* p = arr[k-n-1];ListNode* q = arr[k-n+1];p->next = q;}return h;}
};