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
#include <bits/stdc++.h>
using namespace std;
int rozne[1000000];
int butelki[1000000];
int main()
{
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	
	int wsk1=1, ile=0, miejsce=0;
	int n, k,  i;
	cin>>n>>k;
	for( i=0; i<n; i++)
	{
		cin>>butelki[i];
	}
	rozne[0]=1;
	i=1;
	while(wsk1<k && i<n)
	{
		if(butelki[i-1]!=butelki[i])
		{
			wsk1++;
			rozne[i]=wsk1;
		}
		i++;
	}
	if(wsk1!=k)
	{
		cout<<(-1)<<endl;
	}
	else
	{
		for(i=0; i<n; i++)
		{
			if(rozne[i]!=0)
			{
				ile=ile+i-miejsce;
				miejsce++;
			}
		}
		cout<<ile<<endl;
	}
	return 0;
}