leetcode-75. 颜色分类

leetcode-75. 颜色分类

题目：

代码：

#include <iostream>
#include <vector>
using namespace std;

void sortColors(vector<int>& nums) {
	int flag=0,temp;
	for(int i=0;i<2;i++){
		for(int j=flag;j<nums.size();j++){
			if(nums[j]==i){
				temp=nums[flag];
				nums[flag]=nums[j];
				nums[j]=temp;
				flag++;
			}
		}
	}
}

int main(){
	vector<int> nums;
	string s;
	int tn;
	getline(cin,s);
	for(int i=0;i<s.size();i++){
		if(s[i]==' '){
			nums.push_back(tn);
			tn=0;
		}else{
			tn=tn*10+s[i]-'0';
		}
	}
	nums.push_back(tn);
	sortColors(nums);
	for(int i=0;i<nums.size();i++){
		cout<<nums[i]<<" ";
	}
    return 0;
}