#include<bits/stdc++.h>
using namespace std;
using ll = long long;
int main() {
ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
int n, k; cin>>n>>k;
vector<int> a(n); for(auto& i: a) cin>>i;
vector<bool> taken(n+1);
int cnt = 0;
ll ans = 0;
for(int i=0; i<n && cnt < k; ++i) {
if(!taken[a[i]]) {
taken[a[i]] = 1;
ans += i - cnt;
cnt++;
}
}
if(cnt < k) {
cout<<-1<<'\n';
}
else {
cout<<ans<<'\n';
}
}
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 | #include<bits/stdc++.h> using namespace std; using ll = long long; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, k; cin>>n>>k; vector<int> a(n); for(auto& i: a) cin>>i; vector<bool> taken(n+1); int cnt = 0; ll ans = 0; for(int i=0; i<n && cnt < k; ++i) { if(!taken[a[i]]) { taken[a[i]] = 1; ans += i - cnt; cnt++; } } if(cnt < k) { cout<<-1<<'\n'; } else { cout<<ans<<'\n'; } } |
English