#include <bits/stdc++.h>
using namespace std;
#define int long long
#define st first
#define nd second
#define bg begin
#define ed end
#define pb push_back
#define all(r) bg(r),ed(r)
#define fast ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0)
int32_t main(){
fast;
int n,k;
cin>>n>>k;
unordered_map<int, bool>mapa;
vector<int>ind;
int count=0;
int res=0;
for(int i=0;i<n;i++){
int x;
cin>>x;
if(mapa[x]==0){
mapa[x]=1;
res+=i-count;
count++;
}
if(count==k){
cout<<res<<'\n';
return 0;
}
}
cout<<-1<<'\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 27 28 29 30 31 32 33 34 35 | #include <bits/stdc++.h> using namespace std; #define int long long #define st first #define nd second #define bg begin #define ed end #define pb push_back #define all(r) bg(r),ed(r) #define fast ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0) int32_t main(){ fast; int n,k; cin>>n>>k; unordered_map<int, bool>mapa; vector<int>ind; int count=0; int res=0; for(int i=0;i<n;i++){ int x; cin>>x; if(mapa[x]==0){ mapa[x]=1; res+=i-count; count++; } if(count==k){ cout<<res<<'\n'; return 0; } } cout<<-1<<'\n'; } |
English