41. first-missing-positive

      这道题呢技巧性比较强,对于像我这样的新手第一次一般是想不出来这种方法的。

      题目的大意是找出一个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;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_29592167/article/details/83110053