LeetCode#75. Sort Colors

HAHAHA-

于 2017-04-18 18:16:19 发布

阅读量404

点赞数

分类专栏： LeetCode 文章标签： LeetCode 颜色排序

本文链接：https://blog.csdn.net/u012559634/article/details/70231167

版权

LeetCode 专栏收录该内容

134 篇文章 0 订阅

订阅专栏

题目：有三种颜色（红，白，蓝），分别用0，1，2代替，对这些颜色进行排序，使得相同的颜色相邻（其实就是给定三种数，每种的个数可能有多个，对这样的数组进行排序）
难度：Medium
思路：定义三个变量来记录每种数的个数，然后用一个for循环将三种数按照个数填入数组
代码：

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

        }
        for(int i = 0; i < nums.length; i++){
            if(i < red){
                nums[i] = 0;
            }else if(i < white+red){
                nums[i] = 1;
            }else{
                nums[i] = 2;
            }
        }
    }
}