Leetcode之first missing positive

转自http://blog.csdn.net/nanjunxiao/article/details/12973173

Given an unsorted integer array, find the first missing positive integer.

For example,
Given [1,2,0] return 3,
and [3,4,-1,1] return 2.

Your algorithm should run in O(n) time and uses constant space.

[cpp]  view plain  copy

1. class Solution {  
2. public:  
3.     int firstMissingPositive(int A[], int n) {  
4.         // Note: The Solution object is instantiated only once and is reused by each test case.  
5.         if(NULL==A || n<1)                                                                                                                                     
6.             return 1;  
7.         int pos=0;  
8.         while(pos<n)  
9.         {     
10.             if(A[pos]>0 && A[pos] != pos+1 && A[pos]-1<n && A[pos]!=A[A[pos]-1])  
11.             {                                                                                                                                                  
12.                 swap(A[pos],A[A[pos]-1]);     
13.             }     
14.             else  
15.                 ++pos;  
16.         }     
17.         for(int i=0;i<n;++i)  
18.         {     
19.             if(i+1 != A[i])  
20.                 return i+1;                                                                                                                                    
21.         }     
22.         return n+1;  
23.     }  
24.      void swap(int &a,int &b)   
25.     {     
26.         int tmp = a;  
27.         a = b;  
28.         b= tmp;  
29.     }   
30. };  

思路：

虽然不能再另外开辟非常数级的额外空间，但是可以在输入数组上就地进行swap操作。

思路：交换数组元素，使得数组中第i位存放数值(i+1)。最后遍历数组，寻找第一个不符合此要求的元素，返回其下标。整个过程需要遍历两次数组，复杂度为O(n)。

下图以题目中给出的第二个例子为例，讲解操作过程。

妈蛋，这题挣扎好久。首先思路上，其次临界条件，这题和下面题异曲同工：

n个元素的数组，里面的数都是0~n-1范围内的，求数组中重复的某一个元素，没有返回-1, 要求时间性能O(n) 空间性能O(1)。

python代码:

class Solution(object):
    def firstMissingPositive(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if nums == None or len(nums) == 0:
            return 1
        
        length = len(nums)
        pos = 0
        
        while pos < length:
            if nums[pos] >= 1 and nums[pos] != pos + 1 and nums[pos] <= length and nums[pos] != nums[nums[pos] - 1]:
                self.swap(nums, pos, nums[pos] - 1)
            else:
                pos += 1
        
        for i in range(length):
            if nums[i] != i + 1:
                return i + 1
        return length + 1
    
    def swap(self, nums, pos, nums_pos):
        temp = nums[pos]
        nums[pos] = nums[nums_pos]
        nums[nums_pos] = temp