2435: [Noi2011]道路修建(树上操作)

2435: [Noi2011]道路修建

题目：传送门

---

 

题解：

　　 建完边之后以1为根建树，统计深度和各个点的子树大小（包括自己）

　　 询问的时候：答案=长度*abs(n-深度大的点的子树大小*2)

　　 ans+=a[i].c*abs(n-tot[y]*2)

 

---

代码：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long LL;
struct node
{
    int x,y,c,next;
}a[2110000];int len,last[1110000];
int n,tot[1110000],dep[1110000];
void ins(int x,int y,int c)
{
    len++;a[len].x=x;a[len].y=y;a[len].c=c;
    a[len].next=last[x];last[x]=len;
}
void dfs(int x,int fa)
{
    tot[x]=1;dep[x]=dep[fa]+1;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa)
        {
            dfs(y,x);
            tot[x]+=tot[y];
        }
    }
}
int main()
{
    scanf("%d",&n);len=0;memset(last,0,sizeof(last));
    for(int i=1;i<n;i++)
    {
        int x,y,c;scanf("%d%d%d",&x,&y,&c);
        ins(x,y,c);ins(y,x,c);
    }
    dfs(1,0);
    LL ans=0;
    for(int i=1;i<=len;i+=2)
    {
        int x=a[i].x,y=a[i].y;
        if(dep[x]>dep[y])swap(x,y);
        ans+=LL(a[i].c)*abs(n-tot[y]*2);
    }
    printf("%lld\n",ans);
    return 0;
}