bzoj2435[Noi2011] 道路修建

题目链接:bzoj2435
题目大意:
给你一棵树。树边有权值。求出每条边的边权乘以边的两端的点个数之差的绝对值之和。

题解:
..树形dp
没有想到noi还有这样水的题
然后RE了三遍…还以为爆longlong还是中途爆了什么。
原来是爆栈!
被逼着学了一点点手工栈。
但是。真的只是一点点。凭着YY我打出了奇怪的东西。(虽然是对的= =
然后注释掉的就是平常的dfs
后来学了点真·手工栈又打一遍

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
#define maxn 1000100

struct node
{
    int x,y,next;LL c;
}a[maxn*2];int len,first[maxn];
int n,siz[maxn];LL f[maxn];
int fa[maxn],sta[maxn],upe[maxn];
LL myabs(LL x){return (x<0)?-x:x;}
void ins(int x,int y,LL c)
{
    len++;a[len].x=x;a[len].y=y;a[len].c=c;
    a[len].next=first[x];first[x]=len;
}
void dfs()//奇怪的东西
{
    int head,tail;
    head=tail=1;sta[1]=1;fa[1]=0;
    while (head<=tail)
    {
        int x=sta[head++];
        siz[x]=1;f[x]=0;
        for (int k=first[x];k!=-1;k=a[k].next)
        {
            int y=a[k].y;
            if (y==fa[x]) continue;
            fa[y]=x;sta[++tail]=y;upe[y]=k;
        }
    }
    for (int i=tail;i>=0;i--)
    {
        int x,y;y=sta[i];x=fa[y];
        siz[x]+=siz[y];
        f[x]+=f[y]+a[upe[y]].c*myabs((LL)((n-siz[y])-siz[y]));
    }
}
//void dfs()//真·手工栈
// {
    // int tp=1;sta[tp]=1;fa[1]=0;siz[1]=1;
    // for (int i=1;i<=n;i++) upe[i]=first[i];
    // while (tp)
    // {
        // int x=sta[tp];
        // while (upe[x]!=-1 && a[upe[x]].y==fa[x]) upe[x]=a[upe[x]].next;
        // if (upe[x]==-1)
        // {
            // tp--;
            // if (fa[x])
            // {
                // siz[fa[x]]+=siz[x];
                // ans+=f[x]*myabs((LL)((n-siz[x])-siz[x]));
                //这里直接用个ans存了因为f[i]要被利用去存边权(懒得再开个数组
            // }
            // continue;
        // }
        // int k=upe[x];fa[a[k].y]=x;
        // sta[++tp]=a[k].y;
        // siz[a[k].y]=1;f[a[k].y]=a[k].c;
        // upe[x]=a[k].next;
    // }
// }

// void dfs(int x,int fa)//普通版本
// {
    // siz[x]=1;f[x]=0;
    // for (int k=first[x];k!=-1;k=a[k].next)
    // {
        // int y=a[k].y;
        // if (y==fa) continue;
        // dfs(y,x);
        // f[x]+=f[y]+a[k].c*myabs((LL)((n-siz[y])-siz[y]));
        // siz[x]+=siz[y];
    // }
// }
int main()
{
    //freopen("a.in","r",stdin);
    //freopen("a.out","w",stdout);
    int i,x,y;LL c;len=0;
    scanf("%d",&n);
    memset(first,-1,sizeof(first));
    for (i=1;i<n;i++)
    {
        scanf("%d%d%lld",&x,&y,&c);
        ins(x,y,c);ins(y,x,c);
    }dfs();
    printf("%lld\n",f[1]);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值