删除链表的节点 - 剑指offer简单

删除链表的节点
题目链接
给定单向链表的头指针和一个要删除的节点的值，定义一个函数删除该节点。
返回删除后的链表的头节点。
注意：此题对比原题有改动

示例 1:

输入: head = [4,5,1,9], val = 5
输出: [4,1,9]
解释: 给定你链表中值为 5 的第二个节点，那么在调用了你的函数之后，该链表应变为 4 -> 1 -> 9.

示例 2:

输入: head = [4,5,1,9], val = 1
输出: [4,5,9]
解释: 给定你链表中值为 1 的第三个节点，那么在调用了你的函数之后，该链表应变为 4 -> 5 -> 9.

说明：

题目保证链表中节点的值互不相同
若使用 C 或 C++ 语言，你不需要 free 或 delete 被删除的节点

解题思路

• C++链表
• 需要找到被删除节点的前一个节点node
• node->next = node->next->next

链表结构体

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
 };

创建链表

ListNode* build(vector<int>nums){
    ListNode* ans = NULL;
    for(int i=0;i<nums.size();i++){
        if(ans==NULL){
            ans = new ListNode(nums[i]);
            continue;
        }
        ListNode* last_node = ans;
        while(last_node->next!=NULL){
            last_node = last_node->next;
        }
        last_node->next = new ListNode(nums[i]);
    }
    return ans;
}

打印链表

void print_list(ListNode* list_arr){
    while(list_arr!=NULL){
        cout<<list_arr->val<<" ";
        list_arr = list_arr->next;
    }cout<<endl;
}

删除节点

  ListNode* deleteNode(ListNode* head, int val) {
        if(head->val==val)return head->next;
        ListNode* prior=head;
        while(prior->next->val!=val){
            prior = prior->next;
        }
        prior->next = prior->next->next;
        return head;
    }

解题代码

#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
#include<stack>
#include<string>
#include<cstring>
#include<cstdlib>
using namespace std;

 struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
 };

class Solution {
public:
    ListNode* deleteNode(ListNode* head, int val) {
        if(head->val==val)return head->next;
        ListNode* prior=head;
        while(prior->next->val!=val){
            prior = prior->next;
        }
        prior->next = prior->next->next;
        return head;
    }
};
ListNode* build(vector<int>nums){
    ListNode* ans = NULL;
    for(int i=0;i<nums.size();i++){
        if(ans==NULL){
            ans = new ListNode(nums[i]);
            continue;
        }
        ListNode* last_node = ans;
        while(last_node->next!=NULL){
            last_node = last_node->next;
        }
        last_node->next = new ListNode(nums[i]);
    }
    return ans;
}
void print_list(ListNode* list_arr){
    while(list_arr!=NULL){
        cout<<list_arr->val<<" ";
        list_arr = list_arr->next;
    }cout<<endl;
}
int main()
{
    vector<int>nums = {4,5,1,9};
    int val=9;
    ListNode* list_arr = build(nums);
    print_list(list_arr);
    Solution so = Solution();
    ListNode* ans = so.deleteNode(list_arr,val);
    print_list(ans);
    return 0;
}