NC14291 Cut

题目链接

题目

题目描述

给你一个长度为 $n$ 的序列，你每次可以将一个序列分割成两个连续的的子序列，
分割的代价为原序列的总和。
现在允许你在初始时将序列重新排列一次。
问分割成 $n$ 个长度为 $1$ 的序列的最大总代价是多少？

输入描述

第一行一个数 $n$ 表示原序列的长度；
接下来一行 $n$ 个数 $a_i$ 表示原序列的第 $i$ 个数。
$2<=n<=100000$
$0<=a_i<=10000$

输出描述

一行一个整数表示答案。

示例1

输入

4
3 2 4 1

输出

26

说明

[3,2,4,1]重排->[1,2,3,4]->[1],[2,3,4]->[1],[2],[3,4]->[1],[2],[3],[4]。

示例2

输入

4
1 1 1 1

输出

9

题解

方法一

知识点：贪心，数学。

显然，每次把最小的取出来，大的加的机会就更多。

不妨再进一步思考，实际上从小到大取，第 $i$ 个数会加 $i$ 次。特别地，最后一个数只有 $n-1$ 次，因为最后一次没有取。

所以有 $ans = \sum_{i=1}^n i \cdot a_i - a_n$ 。

时间复杂度 $O(n)$

空间复杂度 $O(n)$

方法二

知识点：贪心，优先队列。

贪心思路还是一样的，不过这次用优先队列傻瓜式维护即可。

不过需要注意的是，为了方便我们假设一个等价情形，元素还未合并，要求每次合并两段序列，使得最后花费最多。显然是从大的先合并，答案和从小开始的分割序列等价。于是，很容易用优先队列维护。

时间复杂度 $O(n \log n)$

空间复杂度 $O(n)$

代码

方法一

#include <bits/stdc++.h>
#define ll long long
 
using namespace std;
 
int a[100007];
 
int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int n;
    cin >> n;
    for (int i = 1;i <= n;i++) cin >> a[i];
    sort(a + 1, a + 1 + n);
    ll ans = -a[n];
    for (int i = 1;i <= n;i++) {
        ans += 1LL * i * a[i];
    }
    cout << ans << '\n';
    return 0;
}

方法二

#include <bits/stdc++.h>
#define ll long long
 
 
using namespace std;
 
int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int n;
    cin >> n;
    priority_queue<ll> pq;
    for (int i = 0, tmp;i < n;i++) cin >> tmp, pq.push(tmp);
    ll ans = 0;
    for (int i = 0;i < n - 1;i++) {
        ll sum = pq.top();
        pq.pop();
        sum += pq.top();
        pq.pop();
        pq.push(sum);
        ans += sum;
    }
    cout << ans << '\n';
    return 0;
}