题目描述:
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.
分析:
题意:给定一个无序整型数组,返回它缺失的第一个正整数。假设给定数组大小为n,时间复杂度要求为O(n)。
思路:如上图所示,我列出了一些情况。假设给定数组大小为n,如果数组涵盖了1→n,则第一个缺失的是n + 1;如果1→n中某个数字缺失了,假设为i,则第一个缺失的是i。我们考虑数组中处于1→n之间的数字,由于数字范围处于数组大小之间,可以采用桶排序的方法进行排序,大于n或者小于1的数字保持在初始位置。那么,排完序后,我们只要顺序遍历一遍,找到第一个nums[i] != i + 1的数(因为一般情况下,1→n的对应下标恰好为0→n - 1),i + 1即为第一个缺失的正整数。
桶排序和顺序遍历,时间复杂度为O(n)。
代码:
#include <bits/stdc++.h>
using namespace std;
// bucket sort
class Solution {
public:
int firstMissingPositive(vector<int>& nums) {
int n = nums.size();
// Exceptional Case:
if(n == 0){
return 1;
}
for(int i = 0; i <= n - 1;){
if((nums[i] >= 1) && (nums[i] <= n) && (nums[i] != i + 1) && (nums[i] != nums[nums[i] - 1])){
swap(nums[i], nums[nums[i] - 1]);
}
else{
i++;
}
}
// get answer
for(int i = 0; i <= n - 1; i++){
if(nums[i] != i + 1){
return i + 1;
}
}
return n + 1;
}
};