力扣算法题 DAY3 链表 | 203.移除链表元素 、707.设计链表 、 206.反转链表

203.移除链表元素 

力扣题目链接(opens new window)

题意：删除链表中等于给定值 val 的所有节点。

示例 1： 输入：head = [1,2,6,3,4,5,6], val = 6 输出：[1,2,3,4,5]

示例 2： 输入：head = [], val = 1 输出：[]

示例 3： 输入：head = [7,7,7,7], val = 7 输出：[]

思路：设置虚拟头节点

代码：

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

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
class Solution {
    public ListNode removeElements(ListNode head, int val) {
        if(head == null){
            return head;
        }
        //因为删除可能涉及到头节点，所以设置dummy节点(虚拟头节点)，统一操作
        ListNode dummy = new ListNode(-1,head);  
        ListNode pre = dummy;
        ListNode cur = head;
        while(cur != null){
            if(cur.val == val){
                pre.next = cur.next;
            }else{
                pre = cur;
            }
            cur = cur.next;
        }
        return dummy.next;
    }
}

707.设计链表

力扣题目链接(opens new window)

题意：

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

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

思路：设置虚拟头节点

代码：

class ListNode {
    int val;
    ListNode next;
    ListNode(){}
    ListNode(int val) {
        this.val=val;
    }
}

class MyLinkedList {

    int size;  //存储链表元素的个数
    ListNode head; //设置虚拟头节点


    //初始化链表
    public MyLinkedList() {
        size = 0;
        head = new ListNode(0);
    }
    
    public int get(int index) {
        //index从0开始，无效返回-1
        if(index < 0 || index >= size){
            return -1;
        }
        ListNode cur = head; 
        //从头节点遍历到下标为index的节点
        for(int i = 0;i <= index;i ++){
            cur = cur.next;
        }
        return cur.val;
    }
    
    public void addAtHead(int val) {
        addAtIndex(0,val);
    }
    
    public void addAtTail(int val) {
        addAtIndex(size,val);
    }
    
    public void addAtIndex(int index, int val) {
        if(index > size){  //索引无效，不进行插入操作
            return;
        }
        if(index < 0){
            index = 0;  //插入到头节点之前
        }
        size ++; //增加列表存储元素的个数
        ListNode pre = head;  //找到要插入节点的前驱
        for(int i = 0;i < index;i ++){
            pre = pre.next;
        }
        ListNode toAdd = new ListNode(val); //要插入的节点
        toAdd.next = pre.next;  //将插入节点的后继节点设置为前驱节点的后继节点
        pre.next = toAdd;  //更新前驱节点的后继节点为插入节点
    }
    
    public void deleteAtIndex(int index) {
        //下标无效，不进行删除
        if(index < 0 || index >= size){
            return;
        }
        size --; //减少列表存储元素的个数
        if(index == 0){  //删除头节点
            head = head.next;
            return;
        }
        ListNode pre = head; 
        //找到要删除节点的前驱节点
        for(int i = 0;i < index;i ++){
            pre = pre.next;
        }
        pre.next = pre.next.next;  //将前驱节点的后继节点设置为要删除元素的后继节点
    }
}

/**
 * Your MyLinkedList object will be instantiated and called as such:
 * MyLinkedList obj = new MyLinkedList();
 * int param_1 = obj.get(index);
 * obj.addAtHead(val);
 * obj.addAtTail(val);
 * obj.addAtIndex(index,val);
 * obj.deleteAtIndex(index);
 */

206.反转链表

力扣题目链接(opens new window)

题意：反转一个单链表。

示例: 输入: 1->2->3->4->5->NULL  输出: 5->4->3->2->1->NULL

思路：双指针法

代码：

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

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode() {}
 *     ListNode(int val) { this.val = val; }
 *     ListNode(int val, ListNode next) { this.val = val; this.next = next; }
 * }
 */
 //双指针法反转链表指针方向
class Solution {
    public ListNode reverseList(ListNode head) {
        ListNode pre = null;
        ListNode cur = head;
        ListNode temp = null;  //记录cur指向的后继节点
        while(cur != null){
            temp = cur.next;
            cur.next = pre; //让后继节点指向前节点，反转指针方向
            //双指针继续向后遍历
            pre = cur; 
            cur = temp;
        }
        //cur指针指向了null，链表反转完毕,返回pre指针，指向了新的头结点
        return pre;
    }
}