[array] leetcode - 41. First Missing Positive - Hard

leetcode - 41. First Missing Positive - Hard

descrition

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.

解析

注意题目的要求，时间复杂度： O(n)。如果没有时间复杂度的现在，我们可以对数组进行排序，并检查顺序数组中 positive 数的缺失情况即可，时间复杂度为O(nlog(n))。

在有时间复杂度限制的情况下，我们需要进一步分析题目的特点。

• 此处还需要注意的是，positive 是指那些大于 0 的数。
• 要找到第一个缺失的正整数（所谓的第一个正整数，是指从 1 开始计数，第一个缺失的正整数）

基本原理：对于 k 个正整数（允许重复），第一个缺失的值必然在区间 [1,k+1] 内。可以想象成，将 k 个球放到 k+1 个箱子里，那么必然有至少有一个箱子是空的。

对于长度为 n 的整型数组 arry[]，假设正整数的个数为 k 个，k<=n，那么缺失的值必然在区间 [1,k+1]内，我们可以将区间映射到 [0,k]（当 k=n 时，区间为[0,n-1]）。

（辅助理解：[1,k+1] 可以看成是从 1 开始顺序编号的箱子，直到 k+1，我们现在有 k 个正整数，正整数的数值表示起要放到几号桶，因为我们最多只有 k 个正整数，那么必然至少有一个桶时空的）

根据以上分析，对于任意正整数 1<=arry[i]<=n，我们可以从左边到右（从编号1开始到k）将其放到 arry[i]-1 的位置。满足 1<=arry[i]<=n 条件的数最多有 n 个。最后检查，如果存在 arry[i] != i+1 ，那么 i+1 就是缺失的第一个正整数，最极端的情况是 n 个数都满足 arry[i] == i+1，此时第一个缺失的正整数为 n + 1。（注意！！从左往右遍历）

具体实现如代码所示。注意，满足 1<=arry[i]<=n 条件的正整数有可能存在重复，因此需要检查即将要放置的目标位置是否已经满足条件，如果不满足条件才能放置。

code

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

class Solution{
public:
	// time-O(n), space-O(1)
	int firstMissingPositive(vector<int>& nums){
		int n = nums.size();
		// put the element to the right place
		// i must be increased from 0 to n, becasue we need to satisfy the lower number first
		for(int i=0; i<n; i++){
			while(nums[i] > 0 && nums[i] <= n && nums[nums[i]-1] != nums[i]){
				// note: nums[nums[i]-1] != nums[i] indicate the place nums[nums[i]-1] hasn't satisfied
				// so we can place the nums[i] to nums[nums[i]-1]
				swap(nums[i], nums[nums[i]-1]);
			}
		}

		// find the first missing positive
		for(int i=0; i<n; i++){
			if(nums[i] != (i+1))
				return i+1;
		}

		return n + 1;
	}
};

int main()
{
	return 0;
}