数组——sort colors

井底的笨鸟

于 2016-05-16 20:26:13 发布

阅读量390

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/jingsuwen1/article/details/51427881

版权

LeetCode 专栏收录该内容

97 篇文章

订阅专栏

题意：对一个容量为n含有红色（用0表示），白色（用1表示），和蓝色（用2表示）的数组按升序排序。。。

 public void sortColors(int[] A) {
        if(A == null||A.length == 0)
            return;
        //计数排序;
        int red=0;int white=0;int blue=0;
        for(int i=0;i<A.length;i++)
          {
            if(A[i] == 0)
              { red++; }
            else if(A[i] == 1)
              { white++; }
            else
              { blue++; }
           }
        for(int i=0;i<A.length;i++)
           {
            if(red>0)
               {
                A[i]=0;
                red--;}
            else if(white>0)
               {
                A[i]=1;
                white--;}
            else
               {
                A[i]=2;
                blue--; }
           }
    }