A. Diverse Team

题目链接

Codeforces Round #486 (Div. 3)--A. Diverse Team

题解

基础题,题意大概是从n个数中找出k个不同的数,输出他们的位置。
因为数据规模非常小,直接暴力就好。

代码

#include <iostream>
#include <vector>
using namespace std;
int n, k;
int a[105];
bool b[105];
vector<int>sx;
int main() {
    cin >> n >> k;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        if (!b[a[i]]) {
            b[a[i]] = true;
            sx.push_back(i);
        }
    }
    if (sx.size() < k) {
        cout << "NO";
    }
    else {
        cout << "YES" << endl;
        for (int i = 0; i < k; i++) {
            cout << sx[i]<<' ';
        }
    }
}

标签: implementation, brute force

添加新评论