#include <bits/stdc++.h>
using namespace std;
#define nd second
#define st first
#define pii pair<int,int>
#define pb push_back
typedef long long ll;
int main(){
ios_base::sync_with_stdio(0);
cin.tie(); cout.tie();
int n, k;
cin>>n>>k;
vector<int>v(n);
for(int i=0; i<n; i++)
cin>>v[i];
set<int>s;
ll cnt=0, ans=0;
for(int i=0; i<n; i++){
if(s.find(v[i])!=s.end()) continue;
if(cnt==k) break;
ans+=(i-cnt);
cnt++;
s.insert(v[i]);
}
if(cnt<k) cout<<"-1";
else cout<<ans;
}
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 | #include <bits/stdc++.h> using namespace std; #define nd second #define st first #define pii pair<int,int> #define pb push_back typedef long long ll; int main(){ ios_base::sync_with_stdio(0); cin.tie(); cout.tie(); int n, k; cin>>n>>k; vector<int>v(n); for(int i=0; i<n; i++) cin>>v[i]; set<int>s; ll cnt=0, ans=0; for(int i=0; i<n; i++){ if(s.find(v[i])!=s.end()) continue; if(cnt==k) break; ans+=(i-cnt); cnt++; s.insert(v[i]); } if(cnt<k) cout<<"-1"; else cout<<ans; } |
English