#include <bits/stdc++.h>
using namespace std;
vector <bool> czy;
vector <bool> wyst;
int main(){
ios_base::sync_with_stdio(0);
cin.tie(0);
long long n,k;
cin>>n>>k;
vector <int> v(n);
czy.resize(n);
wyst.resize(n);
for(int i=0; i<n; i++){
cin>>v[i];
v[i]--;
wyst[v[i]]=1;
}
long long wynik=0;
for(int i=0; i<n; i++){
if(wyst[i]){
wynik++;
}
}
if(wynik<k){
cout<<"-1";
return 0;
}
wynik=0;
int ile=0;
long long a=0;
while(ile<k){
if(!czy[v[a]]){
wynik+=a;
ile++;
}
czy[v[a]]=1;
a++;
}
wynik-=(k-1)*k/2;
cout<<wynik;
}
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 | #include <bits/stdc++.h> using namespace std; vector <bool> czy; vector <bool> wyst; int main(){ ios_base::sync_with_stdio(0); cin.tie(0); long long n,k; cin>>n>>k; vector <int> v(n); czy.resize(n); wyst.resize(n); for(int i=0; i<n; i++){ cin>>v[i]; v[i]--; wyst[v[i]]=1; } long long wynik=0; for(int i=0; i<n; i++){ if(wyst[i]){ wynik++; } } if(wynik<k){ cout<<"-1"; return 0; } wynik=0; int ile=0; long long a=0; while(ile<k){ if(!czy[v[a]]){ wynik+=a; ile++; } czy[v[a]]=1; a++; } wynik-=(k-1)*k/2; cout<<wynik; } |
English