LeetCode 075 Sort Colors

来源:互联网 发布:配电箱价格计算软件 编辑:程序博客网 时间:2024/05/16 02:10

有三种颜色,给他们排序,不用一般的排序算法。

由于只有三种数字,所有可以直接统计,然后再依次赋值即可。

class Solution {public:    void sortColors(vector<int>& nums) {        int temp[3];        temp[0]=temp[1]=temp[2]=0;        for(int ii=0;ii<nums.size();ii++){            if(nums[ii]==0)temp[0]++;            else if(nums[ii]==1)temp[1]++;            else temp[2]++;        }        int cnt=0;        while(temp[0]){            nums[cnt++]=0;            temp[0]--;        }        while(temp[1]){            nums[cnt++]=1;            temp[1]--;        }        while(temp[2]){            nums[cnt++]=2;            temp[2]--;        }        return ;    }};


0 0
原创粉丝点击