PAT 1070. 结绳(25)

Posted on Edited on In

给定一段一段的绳子，你需要把它们串成一条绳。每次串连的时候，是把两段绳子对折，再如下图所示套接在一起。这样得到的绳子又被当成是另一段绳子，可以再次对折去跟另一段绳子串连。每次串连后，原来两段绳子的长度就会减半。

给定N段绳子的长度，你需要找出它们能串成的绳子的最大长度。

输入格式：

每个输入包含1个测试用例。每个测试用例第1行给出正整数N (2 <= N <= 104)；第2行给出N个正整数，即原始绳段的长度，数字间以空格分隔。所有整数都不超过104。

输出格式：

在一行中输出能够串成的绳子的最大长度。结果向下取整，即取为不超过最大长度的最近整数。

输入样例：

8
10 15 12 3 4 13 1 15

输出样例：

14

用贪心的思想来解，最长的绳子折叠的次数越少越好。这里用到了优先队列priority_queue，它保证每次插入到队列中的值都是按顺序排列的，因此我们可以尽可能的找到最短的两条线段，折成新的绳子，然后继续找两个最短的绳子做折叠，直到只剩下一条绳子。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
#include <queue>
#include <cstdio>
#include <functional>
#include <vector>

using namespace std;

int main() {
    int n = 0;
    scanf("%d", &n);
    priority_queue<int, vector<int>, greater<int> > q;
    for (int i = 0; i < n; i++) {
        int temp = 0;
        scanf("%d", &temp);
        q.push(temp);
    }

    while (q.size() > 1) {
        int a = q.top();
        q.pop();
        int b = q.top();
        q.pop();

        q.push((a + b) / 2);
    }
    printf("%d", q.top());
}