1. 程式人生 > 實用技巧 >283 移動零

283 移動零

給定一個數組 nums,編寫一個函式將所有 0 移動到陣列的末尾,同時保持非零元素的相對順序。

示例:

輸入: [0,1,0,3,12]
輸出: [1,3,12,0,0]

說明:

必須在原陣列上操作,不能拷貝額外的陣列。
儘量減少操作次數。

兩種解法,都用到了雙指標

計算每個非零元素之前 0 的個數,只做非零數與 0 之間的交換。時間複雜度 \(O(N)\),空間複雜度 \(O(1)\)

    public void moveZeroes(int[] nums) {
        int zeroNum = 0;
        int temp;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] == 0) {
                zeroNum++;
            } else {
                temp = nums[i];
                nums[i] = 0;
                nums[i - zeroNum] = temp;
            }
        }
    }

將非零數放到指示交換位置的指標所指處。遍歷完後將後續位置填充 0 。每個位置最多遍歷兩次,時間複雜度 \(O(N)\),空間複雜度 \(O(1)\)

    public void moveZeroes(int[] nums) {
        int i = 0;
        int temp;
        for (int j = 0; j < nums.length; j++) {
            if (nums[j] != 0) {
                nums[i] = nums[j];
                i++;
            }
        }
        while (i < nums.length) {
            nums[i] = 0;
            i++;
        }
    }