这道题呢技巧性比较强,对于像我这样的新手第一次一般是想不出来这种方法的。
题目的大意是找出一个int数组中第一个缺失的正整数,比如:0,1,3,5. 这组int数组中,第一个缺失的正整数是2,当然最直观的思路是我从1到n一个一个到数组中去寻找,看第一个缺失的谁,这样复杂度是O(N^2),而这种思路代码比较简单,另一种比较新颖的思路是利用下标和正整数之间的关系,来完成寻找,最终是O(n)的复杂度。
利用正整数与下标之间的关系,假设如果现在的数组里有4个数,分别是1,2,4,5,那么,如果他们按照顺序排列起来我们就可以直接从头到尾扫一遍,通过比较nums[i] 和 下标i的关系,就知道哪个正数缺失,当第一次出现nums[i] != i+1时,就说明i+1是第一个缺失的正整数。
刚刚提到了排序,其实该题中,我们并不需要将所有的元素排一下,而是需要将对应的元素放入对应的位置上去,即使得符合条件的数字满足nums[i] == i+1;
整个数组分为两种情况,第一种是1 2 3 4 6,这样在第四位上会出现nums[i] != i+1的情况,返回的第一个缺失正数为 i+1即可,另外一种是 1 2 3 4 5,这样,数组中每一位都是符合nums[i] == i+1,所以第一个缺失的正整数应该是i+2;
基于以上的思想以下是代码:
class Solution {
public:
int firstMissingPositive(vector<int>& nums) {
if (nums.size() < 1) {
return 1;
}
for (int i=0; i<nums.size();) {
if (nums[i]>0 && nums[i]<=nums.size() && nums[i]!=i+1 && nums[i]!=nums[nums[i]-1]) {
swap(nums[i], nums[nums[i]-1]);
} else {
++i;
}
}
for (int i=0; i<nums.size(); ++i) {
if (nums[i] != i+1) {
return i+1;
} else if (i == nums.size()-1) {
return i+2;
}
}
return 0;
}
};