#include <bits/stdc++.h>
using namespace std;
bool tab[1000000];
int main()
{
int n,k,w=0,r=0;
cin>>n>>k;
for(int i = 1; i<=n ;i ++)
{
int x;
cin>>x;
if(!tab[x]&&r<k)
{
w+=i;
r++;
tab[x]=1;
}
}
if(r<k)cout<<-1;
else cout<<w-(k+1)*k/2;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | #include <bits/stdc++.h> using namespace std; bool tab[1000000]; int main() { int n,k,w=0,r=0; cin>>n>>k; for(int i = 1; i<=n ;i ++) { int x; cin>>x; if(!tab[x]&&r<k) { w+=i; r++; tab[x]=1; } } if(r<k)cout<<-1; else cout<<w-(k+1)*k/2; } |
English