题目

Farmer John 正在在计划自己的农场漫步。他的农场的结构就像一棵树：农场有N个谷仓（1<= N <=100,000），分别由N-1条路链接。这样，他便可以通过这些谷仓间的道路遍及各个谷仓。Farmer John想要选择一条路线：这条路线的起点和终点分别为农场中两个不同的谷仓，这条路线不能重复经过一条边两次。Farmer John担心这条路径可能会偏长，所以他想在路线上寻找一个休息点（当然这个休息点不能为起点或者终点）。

每条边的两旁都是牛群，要么是Charcolais（白毛），要么是Angus（黑毛）。Farmer John是一个聪明人，所以他想要在他通过小路的同时平衡小路两侧阴阳的力量。他要选择一条路径使得他从起点到休息站，和从休息站到终点这两段路上都满足路两边的Charcolais牛群和Angus牛群总数量相同。

Farmer John好奇他能找到多少条如上所述的平衡的路径。我们认为，当且仅当两条路线的边的集合不同时，这两条路径才被认为是不同的，否则认为是相同的路线。就算路线上有多个有效的“休息站”的位置能使路线平衡，我们也只记为一条路线。

请帮助计算有多少条不同的平衡路线。

分析

这是一道不错的点分治练手题目，不难很容易打。

但不知道为神马，旁边的几个逗逼ly、风筝都是很tm麻烦，细节很多。

真tm搞不懂+_+。

首先把黑白毛两种牛当做该边的边权-1或1，如果有两个点，到他们路径上某个点的距离为0，那么这就是一个合法路径。

对于一个以x为根的子树，经过x的路径\(（i，j），其中deep_{i}<=deep_{j}\)有三种：

显然，\(k\)为中间点。

那如何计算这种情况呢？

定义\(dis_{i}\)表示x到i距离，\(d_{i}\)表示在以x为根的子树中的i的祖先是否出现过一个k点，\(dis_{k}=dis_{i}\)，如果是，就为\(true\)，否则为\(false\)。

我们把每个儿子分开做，这样就不用判重了。

再定义\(b_{i}\)表示，在以x为根的子树中，有多少个节点的\(dis\)值为\(i\)；\(b1_{i}\)表示，在以x为根的子树中，有多少个节点的\(dis\)值为\(i\)，且\(d\)值为\(true\)。

接着就可以愉愉快快地求答案了。

每个儿子用两个递归，第一个来求以x为根的子树答案，第二个来求更新\(b\)和\(b1\)。

当前递归到\(l\)点时，如果\(d_{l}=true\)，那么答案加上\(b[-dis_{l}]\)，否则答案加上\(b1[-dis_{l}]\)

这是一种特殊情况，

显然，\(dis_{k},dis_{j}=0\)，这很容易搞。

事实上，可以加点小特判，跟上一种情况一起处理，不过细节有点多，

想一想就可以解决了。

这种情况是不用出处理的，

因为当处理\(lca(i,j)\)时，路径\((i,j)\)会被处理掉。

//code中并没有用到d数组，可以省去
#include <cmath>
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <queue>
const long long maxlongint=2147483647;
using namespace std;
long long d[200005],n,m,tot,dis[200005],next[200005],last[200005],to[200005],v[200005];
long long size[200005],mx[200005],root,f,ans,tt,N,b[200005],b1[200005],e[200005];
bool bz[200005],t[200005],g[200005];
long long bj(long long x,long long y,long long z)
{
    next[++tot]=last[x];
    last[x]=tot;
    to[tot]=y;
    v[tot]=z;
}
void getroot(long long x,long long fa)
{
    mx[x]=0;
    size[x]=1;
    for(long long i=last[x];i;i=next[i])
    {
        if(to[i]!=fa && bz[to[i]])
        {
            getroot(to[i],x);
            size[x]+=size[to[i]];
            mx[x]=max(mx[x],size[to[i]]);
        }
    }
    mx[x]=max(mx[x],f-size[x]);
    if (mx[x]<mx[root]) root=x;
}
long long dg1(long long x,long long fa)
{
    for(long long i=last[x];i;i=next[i])
    {
        long long j=to[i];
        if(j!=fa && bz[j])
        {
            if(t[dis[j]])
            {
                t[dis[j]]=false;
                b[dis[j]]++;
                dg1(j,x);
                t[dis[j]]=true;
            }
            else
            {
                b1[dis[j]]++;
                b[dis[j]]++;
                dg1(j,x);
            }
        }
    }
}
long long dg(long long x,long long fa)
{
    for(long long i=last[x];i;i=next[i])
    {
        long long j=to[i];
        if(j!=fa && bz[j])
        {
            dis[j]=dis[x]+v[i];
            if(t[dis[j]])
            {
                if(g[dis[j]])
                {
                    g[dis[j]]=false;
                    e[++tt]=dis[j];
                }
                t[dis[j]]=false;
                ans+=b1[2*N-dis[j]];
                dg(j,x);
                t[dis[j]]=true;
            }
            else
            {
                ans+=b[2*N-dis[j]];
                dg(j,x);
            }
        }
    }
}
long long solve(long long x,long long fa)
{
    bz[x]=false;
    b[N]=1;
    tt=0;
    for(long long i=last[x];i;i=next[i])
    {
        long long j=to[i];
        if(!bz[j] || j==fa) continue;
        dis[j]=v[i]+N;
        if(g[dis[j]])
        {
            g[dis[j]]=false;
            e[++tt]=dis[j];
        }
        t[dis[j]]=false;
        ans+=b1[2*N-dis[j]];
        dg(j,x);
        t[N]=false;
        b[dis[j]]++;
        dg1(j,x);
        t[N]=true;
        t[dis[j]]=true;
    }
    for(long long i=1;i<=tt;i++)
    {
        g[e[i]]=true;
        b[e[i]]=0;
        b1[e[i]]=0;
    }
    b[N]=0;
    for(long long i=last[x];i;i=next[i])
    {
        long long j=to[i];
        if(j!=fa && bz[j])
        {
            root=0;
            f=size[j];
            getroot(j,x);
            solve(root,x);
        }

    }

}
int main()
{
    scanf("%lld",&n);
    for(long long i=1;i<=n-1;i++)
    {
        long long x,y,z;
        scanf("%lld%lld%lld",&x,&y,&z);
        if(!z) z--;
        bj(x,y,z);
        bj(y,x,z);
    }
    mx[0]=maxlongint;
    f=n;
    N=n+1;
    memset(t,true,sizeof(t));
    memset(g,true,sizeof(g));
    memset(bz,true,sizeof(bz));
    root=0;
    getroot(1,0);
    ans=0;
    bz[0]=false;
    solve(root,0);
    printf("%lld",ans);
}