75. 顏色分類(LeetCode)
阿新 • • 發佈:2021-07-08
- 題目條件
給定一個包含紅色、白色和藍色,一共n 個元素的陣列,原地對它們進行排序,使得相同顏色的元素相鄰,並按照紅色、白色、藍色順序排列。
此題中,我們使用整數 0、1 和 2 分別表示紅色、白色和藍色。
示例:
輸入:nums = [2,0,2,1,1,0]
輸出:[0,0,1,1,2,2]
- 題目條件分析
- 資料元素有限,只有3種
- 要求原地排序,不申請多餘的空間,考慮使用交換
- 解題思路
- 可以定義兩個下標zero和two,其中[0,zero]中包含的都是0,(zero,two)之間包含的都是1,[tow,nums.length-1]包含都是2;
- 在遍歷元素時進行比較,如果元素=1,i++,如果元素=0,交換i和zero的位置,i++,如果元素==2,交換i和two的位置,此時i不需要++,因為還不知道two所在位置的元素是多少
編碼如下:
public void sortColors(int[] nums) { int zero = -1; // [0,zero] == 0 int two = nums.length; // [two, nums.length-1] == 2 int cur = 0; while (cur < two) { if (nums[cur] == 1) { cur++; } else if (nums[cur] == 0) { zero++; NumberUtils.swap(nums, cur, zero); cur++; } else { two--; NumberUtils.swap(nums, cur, two); } } }