【BZOJ 3697】采药人的路径

题目链接：

　　TP

题解：

　　调了好久233。

　　大概想一想就是树分，然后考虑这样路径（u，v）的特征，以根节点（root）切开，u到root的阴阳差值，和v到root巧合互为相反数，然后考虑要有一个点可作为休息点，即u/v到root的路径中要有一点x与u/v到root的阴阳差值相同，然后维护一下就好。

　　注意的是阴阳差为0的特判……写挂了调好久,对拍也不好写，真是恶心。

代码:

　　

#define Troy 11/23
#define inf 0x7fffffff

#include <bits/stdc++.h>

using namespace std;

inline int read(){
    int s=0,k=1;char ch=getchar();
    while(ch<'0'|ch>'9')    ch=='-'?k=-1:0,ch=getchar();
    while(ch>47&ch<='9')    s=s*10+(ch^48),ch=getchar();
    return s*k;
}

const int N=1e5+10;

typedef long long ll;

int n;

struct edges{
    int v,w;edges *last;
}edge[N<<1],*head[N];int cnt;

inline void push(int u,int v,int w){
    edge[++cnt]=(edges){v,w,head[u]};head[u]=edge+cnt;
}

int tot,root,heavy[N],size[N],top,num,T[N<<1][2],nT[N<<1],pre[N<<1];

ll ans,t[N<<1][2];

bool vis[N];

inline void dfs(int x,int fa){
    size[x]=1,heavy[x]=0;
    for(edges *i=head[x];i;i=i->last)if(!vis[i->v]&&i->v!=fa){
        dfs(i->v,x);
        heavy[x]=max(heavy[x],size[i->v]);
        size[x]+=size[i->v];
    }
    heavy[x]=max(heavy[x],tot-size[x]);
    if(top>heavy[x])
        top=heavy[x],root=x;
}

#define g(s) t[s+n]
#define G(s) T[s+n]
#define f(s) pre[s+n]

inline void update(int x,int s,int fa){
    bool a=f(s)>0;
    if(G(s)[a]!=num)
        G(s)[a]=num,g(s)[a]=0;
    ++g(s)[a];
    ++f(s);
    for(edges *i=head[x];i;i=i->last)if(!vis[i->v]&&i->v!=fa)
        update(i->v,s+(i->w?1:-1),x);
    --f(s);
}

inline void calc(int x,int s,int fa){
    bool a=f(s)>0;
    ++f(s);
    if(G(s)[1]==num)
        ans+=g(s)[1];
    if(a&&G(s)[0]==num){
        ans+=g(s)[0];
        if(!s&&f(s)<=2)    ans-=g(s)[0];
    }
    for(edges *i=head[x];i;i=i->last)if(!vis[i->v]&&i->v!=fa)
        calc(i->v,s+(i->w?-1:1),x);
    --f(s);
}
inline void solve(int x){
    top=inf;
    dfs(x,x);
    vis[x=root]=true;    
    G(0)[0]=++num;
    g(0)[0]=1;
    for(edges *i=head[x];i;i=i->last)if(!vis[i->v]){
        calc(i->v,i->w?-1:1,x);
        update(i->v,i->w?1:-1,x);
    }
    for(edges *i=head[x];i;i=i->last)if(!vis[i->v]){
        tot=size[i->v];
        solve(i->v);
    }
}

int main(){
    n=read();
    for(int i=1,u,v,w;i^n;++i){
        u=read(),v=read(),w=read();
        push(u,v,w);
        push(v,u,w);
    }
    tot=n;
    f(0)=1;
    solve(1);
    printf("%lld
",ans);
}