1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
#include <iostream>
#include <cstdint>
#include <vector>
#include <algorithm>
#include <iterator>


using namespace std;

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    int64_t n,k;
    cin>>n>>k;

    vector<int> a(n);
    copy_n(istream_iterator<int>(cin),n,begin(a));
    vector<bool> czy_byla(n+1,false);

    int64_t score=0;
    int unikalnych=0;

    for (int64_t i=0; i<n; i++){
        if (!czy_byla[a[i]]){
            czy_byla[a[i]]=true;
            score+=i;
            unikalnych++;
            if (unikalnych==k)
                break;
        }
    }

    if (unikalnych==k){
        cout << score - (k-1)*k/2<<endl;
    }else
        cout<<"-1\n";

    return 0;
}