一首歌的时间看懂荷兰三色旗问题
生活随笔
收集整理的這篇文章主要介紹了
一首歌的时间看懂荷兰三色旗问题
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
? ? ??給定一個包含紅色、白色和藍色,一共 n 個元素的數組,原地對它們進行排序,使得相同顏色的元素相鄰,并按照紅色、白色、藍色順序排列。
此題中,我們使用整數 0、 1 和 2 分別表示紅色、白色和藍色。
注意:
不能使用代碼庫中的排序函數來解決這道題。
示例:
輸入: [2,0,2,1,1,0] 輸出: [0,0,1,1,2,2]下面看下代碼:
/* 荷蘭三色旗問題解 */class Solution {public://p0追蹤0的最右邊界,p2追蹤2的最左邊界 void sortColors(vector<int>& nums) {// 對于所有 idx < p0 : nums[idx < p0] = 0// curr 是當前考慮元素的下標int p0 = 0, curr = 0;// 對于所有 idx > p2 : nums[idx > p2] = 2//p2首先指向數組最后一個下標int p2 = nums.size() - 1;while (curr <= p2) {if (nums[curr] == 0) {swap(nums[curr++], nums[p0++]);} else if (nums[curr] == 2) {swap(nums[curr], nums[p2--]);}else curr++;}} };?該算法用了用3個指針,p0追蹤0的最右邊界,p2追蹤2的最左邊界 ,curr表示當前下標。遇到等于0或2的時候就進行交換,并且對p0或p2進行操作,思路很巧妙。
時間復雜度:O(N)
空間復雜度:O(1)
?
參考地址:https://leetcode-cn.com/problems/sort-colors/solution/yan-se-fen-lei-by-leetcode/
?
總結
以上是生活随笔為你收集整理的一首歌的时间看懂荷兰三色旗问题的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 动态规划面试常考:最短路径和
- 下一篇: 巧妙异或思路解子集问题,面试官叫绝