題目描述

刪除連結串列中等於給定值val的所有節點。

示例：

輸入： $1->2->6->3->4->5->6, val = 6$
輸出： $1->2->3->4->5$

思路 * 1：
申請一個頭結點cur，並使用一個變數dummpy指向其初始的位置，cur 和 head同時向前遍歷，head總比cur要快一個結點。當head中結點的值達到要求時（即head.val = val）,使較慢的cur.next直接指向head的下一個結點（當然也可以是null），此時cur不用繼續向前，head向前一步後，通過達到的結點後，cur繼續向前。最終返回dummpy.next即可。

程式碼 * 1：

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
 

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

複雜度分析

• 時間複雜度：$O（N）$
• 空間複雜度：$O（1）$

思路 * 2
還是可以將上面的思路改成遞迴的思路，很是簡便的程式碼量（但是資料量過大時，會造成棧的溢位）。

程式碼 * 2

 if (head == null) return null;
        head.next = removeElements(head.next, val);
        return head.val == val ? head.next : head;

複雜度分析

• 時間複雜度：$O（N）$
• 空間複雜度：$O（N）$

完整程式碼：

package leetcode203;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

/**
 * Created by 張超帥 on 2018/10/7.
 */
class ListNode {
    int val;
    ListNode next;
    ListNode( int val) {this.val = val;}
}
class Solution {
    public ListNode removeElements(ListNode head, int val) {
        /*ListNode cur = new ListNode(-1);
        ListNode dummpy = cur;
        while(head != null) {
            if(head.val == val) {
                cur.next = head.next;
            }else {
                cur.next = head;
                cur = cur.next;
            }
            head = head.next;
        }
        return dummpy.next;
    */
        if (head == null) return null;
        head.next = removeElements(head.next, val);
        return head.val == val ? head.next : head;
    }
}
public class leetcode203 {
    public static int[] stringToArrays(String input) {
        input = input.trim();
        input = input.substring(1,input.length() - 1);
        if(input == null){
            return new int[0];
        }
        String [] parts = input.split(",");
        int[] res = new int[parts.length];
        for(int i = 0; i < parts.length; i ++) {
            res[i] = Integer.parseInt(parts[i].trim());
        }

        return res;
    }
    public static ListNode stringToListNode(String input) {
        int[] nodes = stringToArrays(input);
        ListNode cur = new ListNode(-1);
        ListNode dummpy = cur; // 不能寫成 ListNode dummpy = cur.next;
        for(int node : nodes) {
            cur.next = new ListNode(node);
            cur = cur.next;
        }
        System.out.println(dummpy.next.val);
        return dummpy.next;
    }
    public static String listnodeTostring(ListNode head) {
        if(head == null) return "[]";
        String res = "";
        while(head != null) {
            res += head.val + ", ";
            head = head.next;
        }
        return "[" + res.substring(0, res.length() - 2) + "]";
    }
    public static void main(String[] args) throws IOException {
        BufferedReader in = new BufferedReader( new InputStreamReader(System.in));
        String line ;
        while((line = in.readLine()) != null) {
            ListNode head = stringToListNode(line);
            int num = Integer.parseInt(in.readLine());
            ListNode ret = new Solution().removeElements(head, num);
            String result = listnodeTostring(ret);
            System.out.println(result);
        }
    }
}