题目链接
https://vjudge.net/problem/553795/origin
题目大意
给你n块已知染色所需墨水数的白色帆布,开始染色前可以改变帆布
排列顺序,问你把所有帆布染成不同颜色所需最少墨水数(白色不算)
染色方法:
pick+颜色 c+数 f
即选取所有颜色为c的帆布进行染色, 前f个帆布染成颜色x,剩下的染成颜色y(x, y为机器自动选取的保证前面未出现过的两种不同颜色)
(这题本弱鸡好不容易搞懂题意的啊QAQ)
核心思想
我们先模拟一遍机器的染色过程,发现如果有n块帆布,我们必须要染n-1次,有的帆布染色次数多,有的帆布染色次数少,所以我们要考虑的就是开始染布前我们该如何排列各个帆布才能使较大的帆布染色次数更少?
首先如果从正面考虑,至少我是觉得很难想出解决办法,如果我们从反面想,问题就简单了,到最后必定是把两块颜色相同的布给染成不同的布,那么我们就从反面想,我先染两块布,再把它们并成一块,这样就少了一块布,再重复我们就能把所有布都染色了
所以我们选择优先队列来解决该问题, 每次都从优先队列中取出最小的两块帆布,将它们染色后并成一块再放回,重复该步骤直到只剩一块帆布,这时所有染色所用墨水数之和就是最少的了
代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll; //会爆int
priority_queue<ll, vector<ll>, greater<ll> >cnt;
int main()
{
int T;
cin>>T;
while(T--)
{
while(!cnt.empty())cnt.pop();
int n;
scanf("%d", &n);
for(int i=0; i<n; ++i)
{
ll a;
scanf("%I64d", &a);
cnt.push(a);
}
ll ans=0;
while(cnt.size()>=2)
{
ll temp = cnt.top();
cnt.pop();
ans += ( temp + cnt.top() );
temp += cnt.top();
cnt.pop();
cnt.push(temp);
}
printf("%I64d\n", ans);
}
return 0;
}