bzoj3697 点分治

解题思路

祝各位狗年快乐蛤^(*￣w￣)^
研读hzwer的题解半小时，终于明白这题怎么做了……
把边权为0的边的边权改成-1，然后点分治，再考虑经过当前点x，在x子树中的这样的路径。
开两个数组$f(i,0/1)$和$g(i,0/1)$
f的意思是，从x子树中某个点y走到x的路径，路径上的边权和为i，0和1代表这条路径是否有一个与x相连的前缀路径，路径边权和也为i，这样的路径有多少条。之所以要判断这条路径有没有边权为i的前缀路径，是因为如果有的话，就可以在该路径前缀处建立休息站了。
对了，由于i可能是负数，所以我们把所有权值都加上一个n
g的意思是从x走到y的路径，0与1的意思同上。
看完样例，发现起点终点互换并不能增加一种方案，所以在考虑x的每一个子节点时，我们从当前子节点z的子树里找一个起点，再从已经看过的子节点里找一个终点。
喜闻乐见的上代码

ans+=(g[n][0]-1)*f[n][0];
//休息站建到x处，g[n][0]中首先放了一个终点就在x的方案，要去掉
//之所以是g[n][0]和f[n][0]，是为了避免重复计算
for(int j=-lim;j<=lim;++j)
    ans+=g[n-j][0]*f[n+j][1]+g[n-j][1]*f[n+j][0]+g[n-j][1]*f[n+j][1];
//起点到x的路径上无休息站，x到终点有；起点到x有，x到终点无；起点到x有，x到终点也有
for(int j=n-lim;j<=n+lim;++j)
    g[j][0]+=f[j][0],g[j][1]+=f[j][1],f[j][0]=f[j][1]=0;

代码

#include<bits/stdc++.h>
using namespace std;
int read() {
    int q=0;char ch=' ';
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') q=q*10+ch-'0',ch=getchar();
    return q;
}
#define LL long long
const int N=100005,inf=0x3f3f3f3f;
int n,tot,rt,mx,SZ,lim;
int h[N],ne[N<<1],to[N<<1],w[N<<1],sum[N],vis[N],T[N<<1],dep[N],dis[N];
LL ans,f[N<<1][2],g[N<<1][2];
void add(int x,int y,int z) {to[++tot]=y,ne[tot]=h[x],h[x]=tot,w[tot]=z;}
void getrt(int x,int las) {
    int bl=0; sum[x]=1;
    for(int i=h[x];i;i=ne[i])
        if(to[i]!=las&&!vis[to[i]])
        getrt(to[i],x),sum[x]+=sum[to[i]],bl=max(bl,sum[to[i]]);
    bl=max(bl,SZ-sum[x]);
    if(bl<mx) mx=bl,rt=x;
}
void dfs(int x,int las) {
    lim=max(lim,dep[x]);
    if(T[dis[x]]) ++f[dis[x]][1];
    else ++f[dis[x]][0];
    ++T[dis[x]];
    for(int i=h[x];i;i=ne[i])
        if(to[i]!=las&&!vis[to[i]])
        dep[to[i]]=dep[x]+1,dis[to[i]]=dis[x]+w[i],dfs(to[i],x);
    --T[dis[x]];
}
void work(int x) {
    int LIM=0; vis[x]=1,g[n][0]=1;
    for(int i=h[x];i;i=ne[i])
        if(!vis[to[i]]) {
        dis[to[i]]=n+w[i],dep[to[i]]=1,lim=1;
        dfs(to[i],x),LIM=max(LIM,lim);
        ans+=(g[n][0]-1)*f[n][0];
        for(int j=-lim;j<=lim;++j)
            ans+=g[n-j][0]*f[n+j][1]+g[n-j][1]*f[n+j][0]+g[n-j][1]*f[n+j][1];
        for(int j=n-lim;j<=n+lim;++j)
            g[j][0]+=f[j][0],g[j][1]+=f[j][1],f[j][0]=f[j][1]=0;
    }
    for(int i=n-LIM;i<=n+LIM;++i) g[i][0]=g[i][1]=0;
    for(int i=h[x];i;i=ne[i])
        if(!vis[to[i]])
        SZ=sum[to[i]],mx=inf,getrt(to[i],x),work(rt);
}
int main()
{
    int x,y,z;
    n=read();
    for(int i=1;i<n;++i)
        x=read(),y=read(),z=(read()==0?-1:1),add(x,y,z),add(y,x,z);
    mx=inf,SZ=n,getrt(1,0),work(rt);
    printf("%lld\n",ans);
    return 0;
}