【代码随想录】-02链表

【代码随想录】 链表

01 移除链表元素

【题目】：
删除链表中等于给定值 val 的所有节点。
【解题思路】：
找到结点–>更改上一结点的指向–>删除结点
一般是使用List->next来找结点，即找要被删除的结点的上一结点，这样才可以将链表连接起来
【注意】：
在Solution中，为什么不是用struct ListNode而是ListNode ，在C++中，结构体和类的定义可以直接在函数内部使用，不需要提前声明
参考链接：203.移除链表元素

#include <iostream>
#include <vector>
using namespace std;
#define N 10;

 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:
    
    /* 方法1：使用链表本身的头结点
    链表的其他节点都是通过前一个节点来移除当前节点,所以只要改变上一个节点的指向就行，
    但对于头结点来说，它没有上一个节点，所以要对它单独处理，只要将头结点后移即可
    */
    ListNode* removeElements(ListNode* head, int val) {
        //删除头结点
        while (head != NULL && head->val == val)
        {
            ListNode* tmp = head;
            head = head->next;//若要删除头结点，链表本身的头结点要后移
            delete tmp;
        }
        //删除非头结点
        ListNode* cur = head;
        while (cur != NULL && cur->next != NULL) {
            if (cur->next->val == val) {
                ListNode* tmp = cur->next;//指向该结点
                cur->next = cur->next->next;//重新指向
                delete tmp;
            }
            else {
                cur = cur->next;//继续往下走
            }
        }
        return head;
    }
    /* 方法2：设置一个虚拟头结点在进行移除节点操作：
        这样就不用单独考虑链表本身头结点的特殊性了
    */
    ListNode* removeElements2(ListNode* head, int val) {
        ListNode* tmp = new ListNode();
        tmp->next = head;
        while (tmp->next) {
            if (tmp->next->val == val) {
                ListNode* cur = tmp->next;//指向要删除的结点
                tmp->next = cur->next;//重新指向
                delete cur;
            }
            else {
                tmp = tmp->next;
            }
        }
        return head;
    }

};
ListNode* createList(int arr[], int length) {
    ListNode* List = new ListNode();
    ListNode* p = List;
    
    for (int i = 0; i < length; i++)
    {
        p->val = arr[i];
        p->next = new ListNode();
        p = p->next;
    }
    return List;
}
void printList(ListNode* List)
{
    ListNode* p = List;
    while (p) {
        cout << p->val<<" ";
        p = p->next;
    }
    cout << endl;
}
int main() {
    int a[] = {1,2,3,4,5};
    Solution sol;
    ListNode* head = createList(a,5);
    printList(head);
    int val = 2;
    ListNode* result = sol.removeElements(head, val);
    printList(head);
    
    result = sol.removeElements(head, 3);
    printList(head);
    return 0;
}