[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