3697: 采药人的路径

题目链接

题目大意：一棵树，边权为0/1，求有多少路径满足01个数相等，且存在一个点（不能为起点或终点），这个点到起点和终点分别满足01个数相等

题解：淀粉质点分治

ore menci

我的收获：淀粉质！

#include <bits/stdc++.h>
using namespace std;

const int N=100005;
const int INF=1e9;

int n,temp,root,mxdeep;
int t,head[N];
int sz[N],vf[N];
int dep[N],dis[N];
bool vis[N];
long long ans;

struct data{
    long long a[N<<1];
    long long &operator [] (int x){return a[x+N];}
}g[2],f[2],T; 

struct edge{int to,nex,val;}e[N<<1];

void add(int u,int v,int w){e[t].to=v,e[t].val=w,e[t].nex=head[u],head[u]=t++;}

void dfsroot(int x,int fa){
    sz[x]=1;vf[x]=0; 
    for(int i=head[x];i!=-1;i=e[i].nex){
        int v=e[i].to;
        if(v!=fa&&!vis[v]) dfsroot(v,x),sz[x]+=sz[v],vf[x]=max(vf[x],sz[v]);
    }
    vf[x]=max(vf[x],temp-sz[x]);
    if(vf[x]<vf[root]) root=x;
}

void dfs(int x,int fa)
{
    mxdeep=max(mxdeep,dep[x]);
    if(T[dis[x]]) f[1][dis[x]]++;
    else f[0][dis[x]]++;
    T[dis[x]]++;
    for(int i=head[x];i!=-1;i=e[i].nex)
    if(!vis[e[i].to]&&e[i].to!=fa)
    {
        dep[e[i].to]=dep[x]+1;
        dis[e[i].to]=dis[x]+e[i].val;
        dfs(e[i].to,x);
    }  
    T[dis[x]]--;
}

void solve(int x)
{ 
    int mx=0;
    g[0][0]=1;vis[x]=1;

    for(int i=head[x];i!=-1;i=e[i].nex)
    if(!vis[e[i].to])  
    {
        dis[e[i].to]=e[i].val;
        dep[e[i].to]=1;
        mxdeep=1;dfs(e[i].to,0);mx=max(mx,mxdeep);
        ans+=(g[0][0]-1)*f[0][0];
        for(int j=-mxdeep;j<=mxdeep;j++)
            ans+=g[1][-j]*f[1][j]+g[0][-j]*f[1][j]+g[1][-j]*f[0][j];
        for(int j=-mxdeep;j<=mxdeep;j++){
            g[0][j]+=f[0][j];
            g[1][j]+=f[1][j];
            f[0][j]=f[1][j]=0;
        }
    }

    for(int i=-mx;i<=mx;i++) g[0][i]=g[1][i]=0;

    for(int i=head[x];i!=-1;i=e[i].nex)
    if(!vis[e[i].to])  
    {
        root=0;temp=sz[e[i].to];
        dfsroot(e[i].to,0);solve(root);
    }
}

void work()
{
    vf[0]=n,temp=n;
    dfsroot(1,0);solve(root);
    printf("%lld",ans);
}

void init()
{
    scanf("%d",&n);
    memset(head,-1,sizeof(head));
    for(int u,v,w,i=1;i<n;i++){
        scanf("%d%d%d",&u,&v,&w);
        if(!w) w=-1;add(u,v,w),add(v,u,w);
    }
}

int main()
{
    init();
    work();
    return 0;
}