poj_3253_priority queue

描述

农夫约翰想要修篱墙，他需要N块木板，第i块板长Li。然后他买了一块很长的板子，足够他分成N块。忽略每次锯板子带来的损失。

约翰忘记买锯子了，于是像Don借。Don要收费，每次锯一下，就要收一次板子长度的钱。Don让约翰自己选择锯的方案。

求问约翰最少要花多少钱。

 

思路

类似Huffman编码的思想。

 

代码

#include<iostream>
#include<cstdio>
#include<queue>
#include<functional>

using namespace std;


typedef long long ll;
const int MAX_N = 20000;
int N, L[MAX_N];

void solve()
{
    ll ans = 0;
    priority_queue<int, vector<int>, greater<int>> que;
    for (int i = 0; i<N; i++)
    {
        que.push(L[i]);
    }

    while (que.size()>1)
    {
        int l1, l2;
        l1 = que.top();
        que.pop();
        l2 = que.top();
        que.pop();

        ans += l1 + l2;
        que.push(l1 + l2);

    }
    printf("%lld\n", ans);
}

int main()
{
    cin >> N;
    for (int i = 0; i < N; i++)
    {
        cin >> L[i];
    }
    solve();
    return 0;
}