1. 程式人生 > >[leetcode] 41. 缺失的第一個正數

[leetcode] 41. 缺失的第一個正數

mage leetcode rst etc length 刪除 輸入 元素 -m

41. 缺失的第一個正數

註意這題要求時間復雜度應為O(n),並且只能使用常數級別的空間。

挺有意思的一個題。

思路:
通過交換元素的位置使得正確的位置僅存放正確的數字,例如給定一個數字3那麽他應該在第三個位置,下標為2 。當數字>n 或 <1 時,不做操作。當出現重復數字時,置成-1

比如輸入3,4,-1,1 掃描一遍後應該是 1 -1 3 4
技術分享圖片

class Solution {
    public int firstMissingPositive(int[] nums) {
        int n = nums.length;
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] <= n && nums[i] > 0 && nums[i] != i + 1) {
                if (nums[i] == nums[nums[i] - 1]) {
                    // 如果出現重復,把這個數刪除(置-1)即可
                    nums[i] = -1;
                } else swap(i, nums[i] - 1, nums);
            }
        }
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] != i + 1) {
                return i + 1;
            }
        }
        return n + 1;
    }

    public void swap(int i, int j, int[] num) {
        int tmp = num[i];
        num[i] = num[j];
        num[j] = tmp;
    }
}

[leetcode] 41. 缺失的第一個正數