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
#include <iostream>

using namespace std;

int ile_marki[500007];
int butelki[500007];
long long pref[500007];

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(NULL);
    cout.tie(NULL);
    long long n, k;
    cin >> n >> k;
    for(long long i = 1; i <= n; i++){
        pref[i] = pref[i-1]+(i-1);
    }
    long long lewa = 0;
    long long ileZL = 0;
    for(int i = 0; i < n; i++){
        cin >> butelki[i];
    }
    for(long long i = 0; i < k; i++){
        ile_marki[butelki[i]]++;
        if(ile_marki[butelki[i]] > 1){
            lewa += (k-1-i);
            ileZL++;
        }
    }
    lewa -= pref[ileZL];
    for(long long i = k; i < n && ileZL > 0; i++){
        ile_marki[butelki[i]]++;
        if(ile_marki[butelki[i]] == 1){
            lewa += (i-k+ileZL);
            ileZL--;
        }
    }
    if(ileZL != 0){
        cout << -1;
        return 0;
    }
    cout << lewa;
    return 0;
}