AOJ 593分组玩游戏：利用STL 快速排序

n个小朋友在老师的带领下玩游戏。
其中需要分组，老师打算根据大家的衣服颜色分组。
已知衣服颜色用0-9的数字表示。
老师决定每k个人分为一组。每一组的小朋友必须身穿同一种颜色的衣服。
请你告诉老师，是否能成功分组。

Input

仅一组数据
第一行两个整数n,k (1<=n,k<=100)
每二行n个整数a1,a2,a3……an,表示大家的衣服颜色。 (0<=ai<=9)

Output

如果能分组为输出"Yes",否则输出"No";

Sample Input

Original

Transformed

6 2
3 4 5 3 4 5

Sample Output

Yes

将这个问题抽象为一个简单的数学问题：一个序列是否可以平均分为K份，其中保证每一份的数字都是一样的。

我们很容易想到了排序的方法，好在STL 为我们提供了快速排序的sort函数，首先将这N个人所代表的衣服的颜色推进迭代器color中，然后用sort（color.begin(),color.end()）进行默认从小到大排序。排序之后我们只需要判断排序之后的序列分成的K段中，每一段的起点和终点的值相等就说明YES，否则就不能按照颜色进行分组。代码如下：

  

//分组玩游戏 AOJ 593
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main()
{
   int n,k,temp;
   vector<int>color;
   cin>>n>>k;
   for(int i=0;i<n;i++){
	   cin>>temp;
	   color.push_back(temp);
   }   
     if(n%k!=0){
	   cout<<"No"<<endl;
       return 0;
   }
   else {
   sort(color.begin(),color.end());
   for(vector<int>::iterator iter=color.begin();iter<color.end();iter+=k)
	    if(*iter!=*(iter+k-1)) {
			cout<<"No"<<endl;
			return 0;
		}	 
     cout<<"Yes"<<endl;		
   }   
 
  return 0;
}