#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; } |