C++實現(xiàn)LeetCode(41.首個缺失的正數(shù))
[LeetCode] 41. First Missing Positive 首個缺失的正數(shù)
Given an unsorted integer array, find the smallest missing positive integer.
Example 1:
Input: [1,2,0]
Output: 3
Example 2:
Input: [3,4,-1,1]
Output: 2
Example 3:
Input: [7,8,9,11,12]
Output: 1
Note:
Your algorithm should run in O(n) time and uses constant extra space.
這道題讓我們找缺失的首個正數(shù),由于限定了 O(n) 的時間,所以一般的排序方法都不能用,最開始博主沒有看到還限制了空間復(fù)雜度,所以想到了用 HashSet 來解,這個思路很簡單,把所有的數(shù)都存入 HashSet 中,然后循環(huán)從1開始遞增找數(shù)字,哪個數(shù)字找不到就返回哪個數(shù)字,如果一直找到了最大的數(shù)字(這里是 nums 數(shù)組的長度),則加1后返回結(jié)果 res,參見代碼如下:
解法一:
// NOT constant space class Solution { public: int firstMissingPositive(vector<int>& nums) { unordered_set<int> st(nums.begin(), nums.end()); int res = 1, n = nums.size(); while (res <= n) { if (!st.count(res)) return res; ++res; } return res; } };
但是上面的解法不是 O(1) 的空間復(fù)雜度,所以需要另想一種解法,既然不能建立新的數(shù)組,那么只能覆蓋原有數(shù)組,思路是把1放在數(shù)組第一個位置 nums[0],2放在第二個位置 nums[1],即需要把 nums[i] 放在 nums[nums[i] - 1]上,遍歷整個數(shù)組,如果 nums[i] != i + 1, 而 nums[i] 為整數(shù)且不大于n,另外 nums[i] 不等于 nums[nums[i] - 1] 的話,將兩者位置調(diào)換,如果不滿足上述條件直接跳過,最后再遍歷一遍數(shù)組,如果對應(yīng)位置上的數(shù)不正確則返回正確的數(shù),參見代碼如下:
解法二:
class Solution { public: int firstMissingPositive(vector<int>& nums) { int n = nums.size(); for (int i = 0; i < n; ++i) { while (nums[i] > 0 && nums[i] <= n && nums[nums[i] - 1] != nums[i]) { swap(nums[i], nums[nums[i] - 1]); } } for (int i = 0; i < n; ++i) { if (nums[i] != i + 1) return i + 1; } return n + 1; } };
到此這篇關(guān)于C++實現(xiàn)LeetCode(41.首個缺失的正數(shù))的文章就介紹到這了,更多相關(guān)C++實現(xiàn)首個缺失的正數(shù)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
C++數(shù)據(jù)結(jié)構(gòu)二叉搜索樹的實現(xiàn)應(yīng)用與分析
從這篇博客開始,我就要和大家介紹有關(guān)二叉搜索樹的知識,它還衍生出了兩棵樹——AVL樹和紅黑樹,在后面兩篇博客我都會介紹。今天先從二叉搜索樹開始引入2022-02-02