#include <iostream>
#include <vector>
using namespace std;
int count_distinct(vector<int> const& as) {
int count = 0;
vector<bool> used(as.size(), false);
for(auto e : as) {
if(!used[e]) {
used[e] = true;
count++;
}
}
return count;
}
int main(void) {
ios_base::sync_with_stdio(0);
cin.tie(0);
int n, k;
int ni = 1;
uint64_t result = 0;
vector<int> as{};
vector<bool> used{};
cin >> n >> k;
as.resize(n);
used.resize(n, false);
for(auto& e : as) {
cin >> e;
e--;
}
if(count_distinct(as) < k) {
cout << -1;
return 0;
}
used[as[0]] = true;
for(int i = 1; ni < k; i++) {
if(!used[as[i]]) {
used[as[i]] = true;
result += i - ni;
ni++;
}
}
cout << result;
return 0;
}
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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | #include <iostream> #include <vector> using namespace std; int count_distinct(vector<int> const& as) { int count = 0; vector<bool> used(as.size(), false); for(auto e : as) { if(!used[e]) { used[e] = true; count++; } } return count; } int main(void) { ios_base::sync_with_stdio(0); cin.tie(0); int n, k; int ni = 1; uint64_t result = 0; vector<int> as{}; vector<bool> used{}; cin >> n >> k; as.resize(n); used.resize(n, false); for(auto& e : as) { cin >> e; e--; } if(count_distinct(as) < k) { cout << -1; return 0; } used[as[0]] = true; for(int i = 1; ni < k; i++) { if(!used[as[i]]) { used[as[i]] = true; result += i - ni; ni++; } } cout << result; return 0; } |
English