leetcode_75. Sort Colors

版權聲明：本文爲博主原創文章，未經博主允許不得轉載。 https://blog.csdn.net/gxh27954/article/details/60886169

https://leetcode.com/problems/sort-colors/?tab=Description

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.

題意：給你一個數組，只有0，1，2，讓你從小到大排序。

思路：直接記錄0，1，2出現的個數就行了

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int size=nums.size();
        int x=0,y=0,z=0;
        for(int i=0;i<size;i++){
            if(nums[i]==0){
                x++;
            }
            if(nums[i]==1){
                y++;
            }
            if(nums[i]==2){
                z++;
            }
        }
        int j=0;
        for(int i=0;i<x;i++){
            nums[j++]=0;
        }
        for(int i=0;i<y;i++){
            nums[j++]=1;
        }
        for(int i=0;i<z;i++){
            nums[j++]=2;
        }
    }
};