#include <bits/stdc++.h>
using namespace std;
bool czy[500007];
int main()
{
ios_base::sync_with_stdio(0);
cout.tie(0);
cin.tie(0);
int a, b;
cin >> a >> b;
int tab[a];
for(int i = 0; i < a; i++)
{
cin >> tab[i];
}
int poz = 0;
int ile = 0;
long long wyn = 0;
for(int i = 0; i < a; i++)
{
if(czy[tab[i]] == 0)
{
if(ile < b)
{
wyn += (long long)(i - poz);
ile++;
poz++;
}
czy[tab[i]] = 1;
}
}
if(ile < b)
{
cout << "-1" << endl;
return 0;
}
cout << wyn << endl;
return 0;
}
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 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | #include <bits/stdc++.h> using namespace std; bool czy[500007]; int main() { ios_base::sync_with_stdio(0); cout.tie(0); cin.tie(0); int a, b; cin >> a >> b; int tab[a]; for(int i = 0; i < a; i++) { cin >> tab[i]; } int poz = 0; int ile = 0; long long wyn = 0; for(int i = 0; i < a; i++) { if(czy[tab[i]] == 0) { if(ile < b) { wyn += (long long)(i - poz); ile++; poz++; } czy[tab[i]] = 1; } } if(ile < b) { cout << "-1" << endl; return 0; } cout << wyn << endl; return 0; } |
English