leetcode-75 Sort Colors

问题描述

地址：https://leetcode.com/problems/sort-colors/
描述：
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.

翻译：
给出一个数组，数组中的对象代表红色、白色、蓝色，给他们排序，使得相同的颜色彼此相邻。
这里我们使用0,1,2代表颜色红、白、蓝。
注意：不允许使用函数库

问题解析

这个题目下述解法中使用奇淫巧技，一直也没有想到比较通用的好方法。
思路：
（1）定义三个变量 r w b分别代表red, white 和 blue的个数
（2）一次遍历数组，当遇到0时， r++；当遇到1时w++；当遇到2时b++，最终得到r w和b的值
（3）再次遍历数组，根据题目中给出的顺序进行输出

虽然这样写能够解决这个问题，但是感觉这样的解法并不通用。

解析代码

public void sortColors(int[] A) {  
        int i, r, w, b;  
        r = w = b = 0;  
        for (i = 0; i < A.length; i++) {  
            if (A[i] == 0) r++;  
            else  if (A[i] == 1) w++;  
            else b++;  
        }  
        for (i = 0; i < A.length; i++) {  
            if (i < r) A[i] = 0;  
            else if (i < r + w) A[i] = 1;  
            else A[i] = 2;  
        }  
    }