[leetcode]First Missing Positive

此题思路不易想到。

首先思路参照网上,因为是O(n),很自然想到用hash或bitmap,但不能用更多的存储,就用该数组本身吧。用正负来表示是否存在,所以第一遍的时候把负数变成不可能的正数,此处设为Integer.MAX_VALUE。

编写中错误几次,1.是length为0;2.同一个数出现两遍,那么本来负过一次可能会错误的又负负得正,所以要判断;3.要用index0表示1,这样才表示的全。

public class Solution {
    public int firstMissingPositive(int[] A) {
        // Start typing your Java solution below
        // DO NOT write main() function
        if (A.length == 0) return 1;
        
        for (int i = 0; i < A.length; i++) {
            if (A[i] <= 0) { A[i] = Integer.MAX_VALUE; }
        }
        
        for (int i = 0; i < A.length; i++) {
            int idx = A[i] > 0 ? A[i] : -A[i];
            if (idx <= A.length && A[idx-1] > 0) {
                A[idx - 1] = -A[idx - 1];
            }
        }
        
        for (int i = 0; i < A.length; i++) {
            if (A[i] > 0) {
                return i+1;
            }
        }
        
        return A.length+1;
    }
}

python3,此处用了len(nums) + 1表示不可能的数

class Solution:
    def firstMissingPositive(self, nums: List[int]) -> int:
        # remove negative ones and ones outside length
        for i in range(len(nums)):
            if nums[i] <= 0:
                nums[i] = len(nums) + 1
        for i in range(len(nums)):
            num = abs(nums[i])
            if num > 0 and num <= len(nums):
                # flip nums[num - 1]
                nums[num - 1] = -abs(nums[num - 1])
        for i in range(len(nums)):
            if nums[i] > 0:
                return i + 1
        return len(nums) + 1

  

posted @ 2013-08-07 21:20  阿牧遥  阅读(274)  评论(0编辑  收藏  举报