计蒜客 2020 蓝桥杯大学模拟赛(三)程序设计:养猫

  •  1000ms
  •  262144K

众所周知的,小明家里有好多猫,经过一次“猫口普查”,我们得到了以下信息:

小明家里有 nn 只猫,第 ii 只猫的体重是 a_iai​。然而小明热爱虐猫(并不),他决定对这些猫做一些有♂趣的事情

经过了两年半的练习之后,这些猫已经能完全听懂小明的指令了。

小明的指令分成两个阶段,具体步骤如下:

  1. 指定一种颜色,所有这种颜色的猫都会从猫窝里跑出来,此时小明需要付出总共为这些猫的体重的代价。例如,现在小明有三只红色的猫,体重分别为 {1,2,3}1,2,3,还有两只蓝色的猫,体重分别为 {7,8}7,8。此时如果小明声明的颜色是红色,那么所有红色的猫会出来,小明需要支付的代价为 66。如果小明声明的颜色是蓝色,那么所有蓝色的猫会出来,小明需要支付的代价为 1515。

  2. 对于现在出来的这些猫,小明选择其中的一部分,将他们染成一种新的颜色。然后放回所有的猫。

初始时所有猫都是白色。现在小明想知道,如果想要使得这些猫两两颜色都不同,最少需要花费多少代价。

输入格式

第一行一个整数 nn,表示一共有 nn 只猫,接下来一行 nn 个数 a_iai​ 意义如题所示

输出格式

输出一行一个整数表示最小代价

数据范围

# n A_iAi​
1\sim51∼5 \leq 10≤10 \leq 100≤100
6\sim106∼10 \leq 1000≤1000 \leq 1000≤1000
11\sim1511∼15 \leq 100000≤100000 \leq 1000≤1000

样例输入复制

4
6 7 70 25

样例输出复制

159

思路:贪心+哈夫曼树

#include 

using namespace std;

typedef long long ll;

int main() 
{
	int n;
	cin>>n;	
	priority_queue,greater> que;
	for(int i = 0; i < n; i++) 
	{
		ll c;
		cin >> c;
		que.push(c);
	}
	ll ans = 0;
	while(que.size() > 1) 
	{
		ll a = que.top();
		que.pop();
		ll b = que.top();
		que.pop();
		que.push(a + b);
		ans += a + b;
	}
	cout << ans << endl;
}

 

你可能感兴趣的:(蓝桥杯)