【leetcode-数组】sort-colors

题目描述：给定n个物品 颜色为红、白、蓝  将n个物体排序 使得颜色相同的物体是相邻的 且 颜色顺序是 红、白、蓝 使用0、1、2来分别代表红、白、蓝

Given an array with n objects colored red, white or blue, sort them so that objects of the same color are adjacent, with the colors in the order red, white and blue. Here, we will use the integers 0, 1, and 2 to represent the color red, white, and blue respectively.

Note:
You are not suppose to use the library's sort function for this problem.

不可以使用库函数

Follow up:
A rather straight forward solution is a two-pass algorithm using counting sort.First, iterate the array counting number of 0's, 1's, and 2's, then overwrite array with total number of 0's, then 1's and followed by 2's.

Could you come up with an one-pass algorithm using only constant space?

解题思路：题目已经给了一种解题思路  两次循环 第一次循环记录0 1 2 的个数 第二次循环 根据得到的个数 重新复制数组 则可

但是 这道题 希望我们能够一次遍历就可以解决问题  解题思路就是 把0 往1左边放 2往1右边放  借用牛客 向宇左桌的解题思路：设置3个变量，分别代表数组前部head，当前遍历的位置 i，数组后部 tail

①当A[i] = 0时，必然属于数组前部，则交换A[i] 和 A[head] ,接着i++ , head++

②当A[i] = 1时，只需i++就好，因为只有排好了0和2,1自然就只能在中间了，故不作处理

③当A[i] = 2时，不然属于数组后部，则交换A[i]和A[tail]，接着twoindex--，不过此时就不能i++了因为，交换过去的A[i]有可能是0或者2，所以需要在下一个循环里判断，这样它的位置才能够正确

注意一点：循环什么时候结束？答案是 i<tail+1的时候 因为 tail之后的是已经确定了的2 当i遍历到tail指向的数时  整个数组已经排序好了  （我做的时候 因为 没有判断好循环截止的条件 wrong answer了很多次 哭）

代码如下：

public class SortColors {
	public static void main(String[] args) {
		int[] colors = new int[] {0,1,2,0,2,1,1,2};
		sortColor(colors);
		System.out.println(Arrays.toString(colors));
	}

	public static void sortColor(int[] colors) {
		int head=0;
		int tail= colors.length-1;
		int i=0;
		while(i < tail + 1) {
			if(colors[i]==0) {
				swap(colors,i,head);
				head++;
				i++;
			}else if(colors[i]==1) {
				i++;
			}else if(colors[i]==2) {
				swap(colors,i,tail);
				tail--;
			}
		}
	}

	public static void swap(int[] array,int i, int j) {
		int temp = array[j];
		array[j]=array[i];
		array[i]=temp;
		
	}
}

输出：[0, 0, 1, 1, 1, 2, 2, 2]

加加加油！！！！！！！！！！！