#include<bits/stdc++.h>
using namespace std;
int32_t main(){
ios::sync_with_stdio(false);
int n,k;
cin >> n >> k;
vector<int> tab(n);
for(int i=0;i<n;i++)
cin >> tab[i];
vector<int> seen(n+1);
long long diff = 0, d=0;
long long res = 0;
for(int i=0;i<n;i++) {
if(!seen[tab[i]]) {
seen[tab[i]] = 1;
diff++;
res += d;
}
else
d++;
if(diff == k) {
cout<<res;
return 0;
}
}
cout<<-1;
}
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 | #include<bits/stdc++.h> using namespace std; int32_t main(){ ios::sync_with_stdio(false); int n,k; cin >> n >> k; vector<int> tab(n); for(int i=0;i<n;i++) cin >> tab[i]; vector<int> seen(n+1); long long diff = 0, d=0; long long res = 0; for(int i=0;i<n;i++) { if(!seen[tab[i]]) { seen[tab[i]] = 1; diff++; res += d; } else d++; if(diff == k) { cout<<res; return 0; } } cout<<-1; } |
English