POJ 5253 Fence Repair

原題鏈接

思路：倒着想，木塊已經全部切割，每次選擇最小的兩塊合併，直到合併爲一塊爲止。

AC代碼：

#include <iostream>
#include <algorithm>
#include <stdlib.h>
using namespace std;

int l[20001];
int n;
int main()
{
    int i,j,min0,min1;
    long long money=0;
    cin>>n;
    for(i=0;i<n;i++){
        cin>>l[i];
    }
    while(n>1){
        min0=0,min1=1;
        if(l[min0]>l[min1]) swap(min0,min1);
        for(j=2;j<n;j++){
            if(l[j]<l[min0]){
                min1=min0;
                min0=j;
            }
            else if(l[j]<l[min1]){
                min1=j;
            }
        }
        int t=l[min0]+l[min1];
        money+=t;
        if(min0==n-1) swap(min0,min1);
        l[min0]=t;
        l[min1]=l[n-1];
        n--;
    }
    cout<<money<<endl;
    return 0;
}

也可以用priority_queue，省去找最小木塊和次小木塊的步驟。

AC代碼：

#include <iostream>
#include <stdio.h> 
#include <algorithm>
#include <cstdlib>
#include <queue> 
#include <vector>
#include <functional>  //greater<int>
using namespace std;

priority_queue<int, vector<int>, greater<int> > l;
int n;
int main()
{
    int i,j,t;
    long long money=0;
    cin>>n;
    getchar();
    for(i=0;i<n;i++){
        cin>>t;
        getchar();
        l.push(t);
    }
    while(1){
        int min0=l.top();
        l.pop();
        int min1=l.top();
        l.pop();
        t=min0+min1;
        money+=t;
        if(l.empty()) break;
        l.push(t);
    }
    cout<<money<<endl;
    return 0;
}