代码随想录Day3 | 203.移除链表元素 707.设计链表 206.反转链表

代码随想录Day3 | 203.移除链表元素 707.设计链表 206.反转链表

203.移除链表元素

题目： 给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。

思路： 因为要返回头节点，我们设置一个虚拟头节点在头节点的前面，使用两个指针，cur和pre，cur遍历链表，pre指向cur的前一个结点，当cur的值为val时，pre->next指向cur->next。

代码：

class Solution {
    public ListNode removeElements(ListNode head, int val) {
        if(head == null) return head;
        ListNode dummy = new ListNode(-1, head);
        ListNode pre = dummy, cur = head;
        while(cur != null){
            if(cur.val == val){
                cur = cur.next;
                pre.next = cur;
            }else{
                cur = cur.next;
                pre = pre.next;
            }
        }
        return dummy.next;
    }
}

时间复杂度O(n)，空间复杂度O(1)

707.设计链表

题目： 设计链表的实现。您可以选择使用单链表或双链表。单链表中的节点应该具有两个属性：val 和 next。val 是当前节点的值，next 是指向下一个节点的指针/引用。如果要使用双向链表，则还需要一个属性 prev 以指示链表中的上一个节点。假设链表中的所有节点都是0-index的。

在链表类中实现这些功能：

• get(index)：获取链表中第index个节点的值。如果索引无效，则返回-1。
• addAtHead(val)：在链表的第一个元素之前添加一个值为val的节点。插入后，新节点将成为链表的第一个节点。
• addAtTail(val)：将值为 val 的节点追加到链表的最后一个元素。
• addAtIndex(index,val)：在链表中的第 index 个节点之前添加值为val 的节点。如果 index 等于链表的长度，则该节点将附加到链表的末尾。如果 index 大于链表长度，则不会插入节点。如果index小于0，则在头部插入节点。
• deleteAtIndex(index)：如果索引index有效，则删除链表中的第index个节点。

思路： 首先创建一个节点类，存放链表的结点。初始化MyLinkedList时，创建一个虚拟头节点，并且使用一个size常量保存当前链表的长度，size初始化长度为0。

代码：

class listNode{
    int val;
    listNode next;
    public listNode(int val){
        this.val = val;
    }
    public listNode(int val, listNode next){
        this.val = val;
        this.next = next;
    }
}
class MyLinkedList {
    listNode dummy;
    int size;
    public MyLinkedList() {
        this.size = 0;
        this.dummy = new listNode(-1);
    }
    
    public int get(int index) {
        listNode tmp = this.dummy;
        if(index < 0 || index >= size) return -1;
        for(int i = 0; i <= index; i++){
            tmp = tmp.next;
        }
        return tmp.val;
    }
    
    public void addAtHead(int val) {
        listNode head = new listNode(val);
        head.next = this.dummy.next;
        this.dummy.next = head;
        this.size++;
    }
    
    public void addAtTail(int val) {
        listNode tmp = this.dummy;
        while(tmp.next != null){
            tmp = tmp.next;
        }
        tmp.next = new listNode(val);
        this.size++;
    }
    
    public void addAtIndex(int index, int val) {
        listNode tmp = this.dummy;
        if(index <= this.size && index >= 0){
            for(int i = 0; i < index; i++){
                tmp = tmp.next;
            }
            tmp.next = new listNode(val, tmp.next);
            this.size++;
        }
    }
    
    public void deleteAtIndex(int index) {
        listNode tmp = this.dummy;
        if(index < this.size && index >= 0){
            for(int i = 0; i < index; i++){
                tmp = tmp.next;
            }
            tmp.next = tmp.next.next;
            this.size--;
        }
    }
}

206.反转链表

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

示例：

输入：head = [1,2,3,4,5]
输出：[5,4,3,2,1]

思路： 当节点个数为0或1时，直接返回，然后定义pre为第一个节点，cur为第二个节点，让第一个节点的next指向null。然后开始让cur遍历链表，记录下cur的next节点tmp，然后让cur的next指向pre，完成反转。在更新pre和cur的位置完成下一个节点反转，直至cur为null，此时pre的位置就是反转之后的头节点。

代码：

class Solution {
    public ListNode reverseList(ListNode head) {
        if(head == null || head.next == null) return head;
        ListNode pre = head, cur = head.next;
        pre.next = null;
        while(cur != null){
            ListNode tmp = cur.next;
            cur.next = pre;
            pre = cur;
            cur = tmp;
        }
        return pre;
    }
}

时间复杂度O(n)，空间复杂度O(1)