- A多样化的组队
在学校的一个班级里面有n个学生,关于Codehorses的第i名学生的排名是ai。你需要建立一个包含k个学生的组,这些成员的排名都是不同的。
如果不可能组这么一个合适的队,就输出“NO”(不包括引号)。否则就输出”YES”,然后输出k个不同的1到n整数那些你建立的需要在组中表示出来的学生。在队中的所有学生的排名都不同。你可以以任何形式输出,如果有多个答案,就输出它们中的任何一个。
假设学生从1到n排序。
提示:
在第一组案例中所有可能的结果为:
1,2,5
2,3,5
2,4,5
注意顺序无关紧要。 通过的代码:
#include<iostream>
#include<algorithm>
#include<set>
using namespace std;
set<int>s;
int b[120],c[120];
int main(){
int n,k;cin>>n>>k;
for(int i=1;i<=n;i++){int a;cin>>a;s.insert(a);b[a]=i;}
if(s.size()<k)puts("NO");
else {
puts("YES");int t=0;
set<int>::iterator it=s.begin();
while(it!=s.end()){
c[++t]=b[*it];
it++;
}
for(int i=1;i<=k;i++)cout<<c[i]<<" ";
}
}