打卡信奥刷题（007）用C++信奥P9045[普及组/提高] [PA2021] 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$。

C++实现

#include

using namespace std;
int main(){
int book[5000];
int ans=0,n,k,d=0;
int a[]={3,3,3,1,2};
cin>>n>>k;
for(int i=0;i<=n;i++){
int tmp=a[i];
if(book[tmp]0){
book[tmp]=1;
ans+=i+1-d-1;
d++;
}
if(dk){
cout<<" "<<ans;
return 0;
}
}
}

后续：

接下来我会不断用C++来实现信奥比赛中的算法题、C++考级编程题实现、白名单赛事考题实现，感兴趣的请关注，我后续将继续分享相关内容