PAT 乙级练习 题解合集
题目
给定一段一段的绳子,你需要把它们串成一条绳。每次串连的时候,是把两段绳子对折,再如下图所示套接在一起。这样得到的绳子又被当成是另一段绳子,可以再次对折去跟另一段绳子串连。每次串连后,原来两段绳子的长度就会减半。
给定 N 段绳子的长度,你需要找出它们能串成的绳子的最大长度。
输入格式:
每个输入包含 1 个测试用例。每个测试用例第 1 行给出正整数 N (2≤N≤104 );第 2 行给出 N 个正整数,即原始绳段的长度,数字间以空格分隔。所有整数都不超过104 。
输出格式:
在一行中输出能够串成的绳子的最大长度。结果向下取整,即取为不超过最大长度的最近整数。
输入样例:
8
10 15 12 3 4 13 1 15
输出样例:
14
思路
以下是不严谨的直觉思考:一共给了我 N N N 段绳子,最后要串成 1 1 1 根绳子,那么一共需要串 N − 1 N - 1 N−1 次。
因为每串一次绳子,长度都要减半,所以在最后的绳子中,最先串的两根绳子只能贡献其原本长度的 0. 5 N − 1 0.5^{N-1} 0.5N−1 倍,而最后加入的绳子就可以贡献其原本长度的 0.5 0.5 0.5 倍,显然后加入的绳子比先加入的绳子贡献的权重更大。
所以每次都选取最短的两根绳子串起来,重复 N − 1 N - 1 N−1 次之后就能得到结果。
为什么我一开始要用小根堆呢?因为我原以为可能会发生最短的两根绳子串完之后的长度大于其他还没用过的绳子,但事实上这是不可能的。
假设长度为 N N N 的正整数数组中最小的数是 a a a,将其取出,再将剩下 N − 1 N-1 N−1 个元素中最小的数 b b b 取出。那么 a + b 2 \frac{a+b}{2} 2a+b 显然不大于剩下的 N − 2 N-2 N−2 个元素。
代码
想明白了的简单版本
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main() {
int n;
cin >> n;
vector<double> ropes(n);
for (int i = 0; i < n; ++i)
cin >> ropes[i];
sort(ropes.begin(), ropes.end());
double ans = ropes[0];
for (int i = 1; i < n; ++i)
ans = 0.5 * (ans + ropes[i]);
cout << static_cast<int>(ans) << endl;
return 0;
}
小根堆版本
#include <iostream>
#include <queue>
#include <vector>
using namespace std;
int main() {
int n;
double input, a, b;
priority_queue<double, vector<double>, greater<double> > ropes;
cin >> n;
while (n--) {
cin >> input;
ropes.push(input);
}
while (ropes.size() != 1) {
a = ropes.top();
ropes.pop();
b = ropes.top();
ropes.pop();
ropes.push(0.5 * (a + b));
}
cout << static_cast<int>(ropes.top()) << endl;
return 0;
}