75. Sort Colors

Posted 2020-07-04 阿怪123

public class Solution {
    public void sortColors(int[] nums) {
        int len=nums.length;
        int red=0;
        int white=0;
        int blue=0;
        for(int i=0;i<len;i++)
        {
            if(nums[i]==0)
                red++;
            else if(nums[i]==1)
                white++;
            else
                blue++;
        }
        for(int i=0;i<len;i++)
        {
            if(red!=0)
            {
                red--;
                nums[i]=0;
            }
            else if(white!=0)
            {
                white--;
                nums[i]=1;
            }
            else
            {
                blue--;
                nums[i]=2;
            }
        }
    }
}