1. 程式人生 > >C#LeetCode刷題之#203-刪除連結串列中的節點(Remove Linked List Elements)

C#LeetCode刷題之#203-刪除連結串列中的節點(Remove Linked List Elements)

問題

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

輸入: 1->2->6->3->4->5->6, val = 6

輸出: 1->2->3->4->5

Remove all elements from a linked list of integers that have value val.

Input:  1->2->6->3->4->5->6, val = 6

Output: 1->2->3->4->5

示例

public class Program {

    public static void Main(string[] args) {
        var head = new ListNode(1) {
            next = new ListNode(2) {
                next = new ListNode(6) {
                    next = new ListNode(3) {
                        next = new ListNode(4) {
                            next = new ListNode(5) {
                                next = new ListNode(6)
                            }
                        }
                    }
                }
            }
        };

        var res = RemoveElements(head, 6);
        ShowArray(res);

        Console.ReadKey();
    }

    private static void ShowArray(ListNode list) {
        var node = list;
        while(node != null) {
            Console.Write($"{node.val} ");
            node = node.next;
        }
        Console.WriteLine();
    }

    private static ListNode RemoveElements(ListNode head, int val) {
        //建立一個根節點,以便統一處理所有節點,否則需要單獨處理邊界,會很麻煩
        var virtualHead = new ListNode(0);
        virtualHead.next = head;
        //臨時節點,用於存放每次移動的指標
        var next = virtualHead;
        while(next.next != null) {
            if(next.next.val == val) {
                //找到需要刪除的節點,將指標往後移動2位
                next.next = next.next.next;
            } else {
                //找不到,直接往後移動指標
                next = next.next;
            }
        }
        //返回除了根節點之外的所有節點
        return virtualHead.next;
    }

    public class ListNode {
        public int val;
        public ListNode next;
        public ListNode(int x) { val = x; }
    }

}

以上給出1種演算法實現,以下是這個案例的輸出結果:

1 2 3 4 5

分析:

顯而易見,以上演算法的時間複雜度為: O(n)