[백준] 20186. 수 고르기 (2020 정올 2차 초등부 1번)<C++>

www.acmicpc.net/problem/20186

20186번: 수 고르기

 

 

 

 

나의 풀이: 

 

 

가장 큰 수를 k개만큼 고르기만 하면 되는 문제이다.

 

 

가장 큰 수를 고르기 위해 정렬을 해준 후 뒤의 수 k개를 vector에 담아준다.

 

입력 받은 배열을 훑어보며 vector에 있는 수라면 더해준다.

 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
 
int n, k, cnt, sum;
vector<int> v;
int arr[100001];
int arr2[100001];
 
int big(int num) {
    for (int i = 0; i < v.size(); i++) {
        if (num == v[i]) {
            v[i] = 0;
            return 1;
        }
    }
    return 0;
}
 
int main() {
    cin >> n >> k;
    for (int i = 0; i < n; i++) {
        cin >> arr[i];
        arr2[i] = arr[i];
    }
    sort(arr2, arr2 + n);
 
    int c = 0;
    for (int i = n - 1; c < k; i--) {
        v.push_back(arr2[i]);
        c++;
    }
 
    for (int i = 0; i < n; i++) {
        if (big(arr[i]) && cnt < k) {
            sum += arr[i] - cnt;
            cnt++;
        }
    }
 
    cout << sum;
}