POJ:1738

楼教主男人八题第二题,5W个叶子节点合并最小费用

最小权费和二叉树:GW算法

#include <cstdio>
#include <cstring>

using namespace std;

#define maxn 50050
#define maxV 1000000001

int S = 1;
int L = 0;
long long V[maxn];
int LAST[maxn];
int NEXT[maxn];

int main(){
    int n;
    while(scanf("%d",&n)!=EOF && n){
        memset(V,0,sizeof(V));
        memset(LAST,0,sizeof(LAST));
        memset(NEXT,0,sizeof(NEXT));
        L = n;
        S = 1;
        for(int i=1;i<=L;i++){
            scanf("%lld",&V[i]);
            LAST[i] = i-1;
            NEXT[i] = i+1;
        }
        NEXT[L] = 0;
        V[0] = maxV;
        V[L+1] = maxV;
        long long ans = 0;
        int st = S;
        while(L>2){
            int i = st;
            while(i!=0){
                int a = LAST[i];
                int b = NEXT[i];
                if(V[a] <= V[b]){
                    V[a] += V[i];
                    ans += V[a];
                    NEXT[a] = b;
                    LAST[b] = a;
                    i = a;
                    break;
                }
                i = NEXT[i];
            }
            L--;
            int j = i;
            while(j!=0){
                int j_last = LAST[j];
                if(V[j_last] >= V[i]){
                    if(i==j) break;
                    int a = LAST[i];
                    int b = NEXT[i];
                    NEXT[a] = b;
                    LAST[b] = a;
                    NEXT[i] = j;
                    LAST[i] = j_last;
                    LAST[j] = i;
                    NEXT[j_last] = i;
                    if(j_last == 0) S = i;
                    break;
                }
                j = LAST[j];
            }
            for(int k=0;k<2;k++){
                if(i==S) break;
                i = LAST[i];
            }
            st = i;
        }
        if(L==2) printf("%lld\n",ans+V[S]+V[NEXT[S]]);
        if(L==1) printf("0\n");
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值