算法60天_Day3

【算法60天：Day3】203.移除链表元素、206.反转链表

题目

移除链表元素(203)

删除链表中等于给定值 val 的所有节点。
链表https://leetcode.cn/problems/remove-linked-list-elements/

解题思路

虚拟头节点

class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        // 删除头结点
        while (head != NULL && head->val == val) { // 注意这里不是if
            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;
    }
};

反转链表(206)

给你单链表的头节点 head ，请你反转链表，并返回反转后的链表。

链接https://leetcode.cn/problems/reverse-linked-list/

解题思路

自己的思路是再定义一个新的链表，实现链表元素的反转，但其实这是对内存空间的浪费。
双指针法

class Solution {
public:
    ListNode* reverseList(ListNode* head) {
        ListNode* temp; // 保存cur的下一个节点
        ListNode* cur = head;
        ListNode* pre = NULL;
        while(cur) {
            temp = cur->next;  // 保存一下 cur的下一个节点，因为接下来要改变cur->next
            cur->next = pre; // 翻转操作
            // 更新pre 和 cur指针
            pre = cur;
            cur = temp;
        }
        return pre;
    }
};