n个小朋友在老师的带领下玩游戏。
其中需要分组,老师打算根据大家的衣服颜色分组。
已知衣服颜色用0-9的数字表示。
老师决定每k个人分为一组。每一组的小朋友必须身穿同一种颜色的衣服。
请你告诉老师,是否能成功分组。
其中需要分组,老师打算根据大家的衣服颜色分组。
已知衣服颜色用0-9的数字表示。
老师决定每k个人分为一组。每一组的小朋友必须身穿同一种颜色的衣服。
请你告诉老师,是否能成功分组。
Input
仅一组数据
第一行两个整数n,k (1<=n,k<=100)
每二行n个整数a1,a2,a3……an,表示大家的衣服颜色。 (0<=ai<=9)
第一行两个整数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;
}