SDUTOJ 1743 - 最优合并问题

#include<bits/stdc++.h>
using namespace std;

bool cmp(int a, int b)
{
    return a>b;
}

int main()
{
    int k;
    int a[1010], b[1010];
    int minval = 0, maxval = 0;
    cin>>k;
    for (int i = 0; i < k; i++){
        cin>>a[i];
        b[i] = a[i];
    }
    sort(a, a+k); // 对a数组进行升序排序
    sort(b, b+k, cmp); // 对b数组进行降序排序
    for (int i = 0; i < k-1; i++){
        // 计算最少比较次数
        // 更新合并后的序列a[i+1]
        a[i+1] += a[i];
        minval += a[i+1]-1; // 即minval = a[i]+a[i+1]-1
        sort(a+i+1, a+k); // 重新升序排序

        // 计算最多比较次数
        b[i+1] += b[i];
        maxval += b[i+1]-1;
        sort(b+i+1, b+k, cmp); // 重新降序排序
    }
    cout<<maxval<<' '<<minval<<endl;
    return 0;
}