1. 程式人生 > 其它 >75. 顏色分類(LeetCode)

75. 顏色分類(LeetCode)

  • 題目條件

給定一個包含紅色、白色和藍色,一共n 個元素的陣列,原地對它們進行排序,使得相同顏色的元素相鄰,並按照紅色、白色、藍色順序排列。

此題中,我們使用整數 0、1 和 2 分別表示紅色、白色和藍色。

示例:

輸入:nums = [2,0,2,1,1,0]

輸出:[0,0,1,1,2,2]

  • 題目條件分析
  1. 資料元素有限,只有3種
  2. 要求原地排序,不申請多餘的空間,考慮使用交換
  • 解題思路
  1. 可以定義兩個下標zero和two,其中[0,zero]中包含的都是0,(zero,two)之間包含的都是1,[tow,nums.length-1]包含都是2;
  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);
		}
	}
}