思路很简单:每次找segments中最短的两截拿出来合成一截,然后当做一个新的segment放进去。最后只剩下一个segment时就是最终的longest rope。做法跟霍夫曼编码一模一样。
是一道比较简单的贪婪算法,因为绳子每次进行拼接都会缩短一半,所以很显然越长的越放在最后,只用拼接一次,长度也只损耗一半。
每次需要拿出最短的两截,很显然需要用到优先队列,即最小最大堆。C++的STL已经帮我们实现了priority_queue,直接使用即可。优先队列的用法,以及设置自定义比较规则,可以参考下面的博客:
https://blog.csdn.net/lijiang1991/article/details/52727478
最后贴上AC代码
#include<stdio.h>
#include<algorithm>
#include<math.h>
#include<queue>
#include<functional>
using namespace std;
int main()
{
priority_queue<double,vector<double>,greater<double>> segments;
int N;
scanf("%d", &N);
for (int i = 0; i < N; i++)
{
int segment;
scanf("%d", &segment);
segments.push(segment);
}
while (segments.size() > 1)
{
double a = segments.top();
segments.pop();
double b = segments.top();
segments.pop();
double c = a / 2 + b / 2;
segments.push(c);
}
printf("%d\n", (int)segments.top());
return 0;
}