LeetCode 41. 缺失的第一个正数(First Missing Positive)
题目描述
给定一个未排序的整数数组,找出其中没有出现的最小的正整数。
示例 1:
输入: [1,2,0]
输出: 3
示例 2:
输入: [3,4,-1,1]
输出: 2
示例 3:
输入: [7,8,9,11,12]
输出: 1
说明:
你的算法的时间复杂度应为O(n),并且只能使用常数级别的空间。
解题思路
本题让只能使用常数级别的空间,所以哈希的解法只能放弃了。考虑另一种思路,把数字依次放到他们对应的位置上,比如1放到第一个(数组第0个)位置上,2放到第二个(数组第1个)位置上,通过交换可得到,若该位置上的数不是正数或者大于数组的长度就跳过。这样调整完整个数组后,再从数组首位置开始判断,若对应位置上的数错误就返回该位置,若都摆放正确,就返回数组长度加1.
代码
1 class Solution { 2 public: 3 int firstMissingPositive(vector<int>& nums) { 4 int i = 0; 5 while(i < nums.size()){ 6 while(nums[i] > 0 && nums[i] <= nums.size() && nums[i] != i + 1 && nums[i] != nums[nums[i] - 1]) 7 swap(nums[i], nums[nums[i] - 1]); 8 i++; 9 } 10 for(int i = 0; i < nums.size(); i++) 11 if(nums[i] != i + 1) 12 return i + 1; 13 return nums.size() + 1; 14 } 15 };