Oranżada

[PA2021] Oranżada

题目描述

有一排共 $n$ 瓶橙汁，其中第 $i$ 瓶的品牌为 $a_i$。

你可以花费 $1$ 个单位的的代价交换两瓶相邻的橙汁。

求最小代价使得最左边 $k$ 瓶橙汁品牌两两不同。

输入格式

第一行，两个整数 $n,k$；

第二行，$n$ 个整数 $a_1,a_2,\cdots ,a_n$。

输出格式

一行，一个整数，若有解，输出最小代价；否则，输出 $-1$。

样例 #1

样例输入 #1

5 3
3 3 3 1 2

样例输出 #1

4

样例 #2

样例输入 #2

3 2
1 1 1

样例输出 #2

-1

提示说明

样例 #1 解释

最优方案为先交换位置 $3$ 和 $4$ 的瓶子、再交换位置 $4$ 和 $5$ 的瓶子，接着交换位置 $2$ 和 $3$ 的瓶子，最后交换位置 $3$ 和 $4$ 的瓶子，共 $4$ 次操作。

样例 #2 解释

显然无解。

数据范围

对于 $100\%$ 的数据，$1\le k,a_i\le n\le 5\times 10^5$。

代码内容

// #include <iostream>
// #include <algorithm>
// #include <cstring>
// #include <stack>//栈
// #include <deque>//队列
// #include <queue>//堆/优先队列
// #include <map>//映射
// #include <unordered_map>//哈希表
// #include <vector>//容器，存数组的数，表数组的长度
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const ll N=5e5+10;
bool vis[N];

int main()
{
    ll n,k;
    cin>>n>>k;

    ll p=0,c=0;
    for(ll i=1;i<=n;i++)
    {
        ll a;
        cin>>a;
        if(!vis[a])
        {
            c+=i-p-1;
            p++;
            vis[a]=1;
        }
        if(p==k)
        {
            cout<<c;
            return 0;
        }
    }
    
    cout<<-1;
    return 0;
}