BZOJ3697

3697: 采药人的路径

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 773   Solved: 271
[ Submit][ Status][ Discuss]

Description

采药人的药田是一个树状结构，每条路径上都种植着同种药材。
采药人以自己对药材独到的见解，对每种药材进行了分类。大致分为两类，一种是阴性的，一种是阳性的。
采药人每天都要进行采药活动。他选择的路径是很有讲究的，他认为阴阳平衡是很重要的，所以他走的一定是两种药材数目相等的路径。采药工作是很辛苦的，所以他希望他选出的路径中有一个可以作为休息站的节点（不包括起点和终点），满足起点到休息站和休息站到终点的路径也是阴阳平衡的。他想知道他一共可以选择多少种不同的路径。

Input

第1行包含一个整数N。
接下来N-1行，每行包含三个整数a_i、b_i和t_i，表示这条路上药材的类型。

Output

输出符合采药人要求的路径数目。

Sample Input

7
1 2 0
3 1 1
2 4 0
5 2 0
6 3 1
5 7 1

Sample Output

1

HINT

对于100%的数据，N ≤ 100,000。

这个题真TM烦...

首先把0都变成-1,这样yy平衡就变成了和为0

如果路径上有一个休息站的话,

要么休息站在根,此时满足该点到根的距离为0

要么休息站在路径上,此时就需要满足该点到根的长度等于路径端点到根的长度

点分治,然后记一个h表示点到根的距离为x时有h(x)个点

记一个q表示点到根的路径上有一个点A满足该点到根的长度等于A到根的长度

然后就是恶心的计算...不说了...

PS:把距离为0分开算好些

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define inf (1<<30)
#define maxn 100011
#define g(x) (st[x+maxn])
#define h(x) (hp[x+maxn])
#define q(x) (fp[x+maxn])
using namespace std;
struct edge{
	int r,nxt,w;
}e[maxn<<1];
int head[maxn],esz,size[maxn],num,mn,rt;
int st[maxn<<1],hp[maxn<<1],fp[maxn<<1],ans0,nw0,n;
long long ans=0;
bool vis[maxn];
void addedge(int u,int v,int w){
	e[++esz].r=v;e[esz].nxt=head[u];
	head[u]=esz;e[esz].w=w;
}
void dfs1(int u,int f){
	num++;
	for(int t=head[u];t;t=e[t].nxt)if(!vis[e[t].r]&&e[t].r!=f)
		dfs1(e[t].r,u);
}
void dfs2(int u,int f){
	size[u]=1;
	int mx=0;
	for(int t=head[u];t;t=e[t].nxt)if(!vis[e[t].r]&&e[t].r!=f)
		dfs2(e[t].r,u),mx=max(mx,size[e[t].r]),size[u]+=size[e[t].r];
	mx=max(mx,num-size[u]);
	if(mx<mn)mn=mx,rt=u;
}
void dfs3(int u,int f,int dist){
	if(!dist)ans0++;
	if(!dist&&g(0))nw0++;
	if(dist&&g(dist))ans+=h(-dist);
	else if(dist)ans+=q(-dist);
	g(dist)++;
	for(int t=head[u];t;t=e[t].nxt)if(!vis[e[t].r]&&e[t].r!=f)
		dfs3(e[t].r,u,dist+e[t].w);
	g(dist)--;
}
void dfs4(int u,int f,int dist){
	h(dist)++;
	if(dist&&g(dist))q(dist)++;
	g(dist)++;
	for(int t=head[u];t;t=e[t].nxt)if(!vis[e[t].r]&&e[t].r!=f)
		dfs4(e[t].r,u,dist+e[t].w);
	g(dist)--;
}
void dfs5(int u,int f,int dist){
	h(dist)=q(dist)=0;
	for(int t=head[u];t;t=e[t].nxt)if(!vis[e[t].r]&&e[t].r!=f)
		dfs5(e[t].r,u,dist+e[t].w);
}
void clear(int u){
	ans0=0;
	for(int t=head[u];t;t=e[t].nxt)if(!vis[e[t].r])
		dfs5(e[t].r,u,e[t].w);
}
void getrt(int u){
	mn=1<<30,num=0;
	dfs1(u,0),dfs2(u,0);
}
void work(int u){
	getrt(u);
	vis[rt]=true;
	for(int t=head[rt],p=0;t;t=e[t].nxt)if(!vis[e[t].r]){
		ans0=0,nw0=0,dfs3(e[t].r,rt,e[t].w),dfs4(e[t].r,rt,e[t].w);
		ans+=1ll*ans0*p+nw0,p+=ans0;
	}
	clear(rt);
	for(int t=head[rt];t;t=e[t].nxt)if(!vis[e[t].r])work(e[t].r);
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<n;++i){
		int u,v,w;scanf("%d%d%d",&u,&v,&w);
		addedge(u,v,w>0?1:-1),addedge(v,u,w>0?1:-1);
	}
	work(1);
	printf("%lld",ans);
}

link to mcfx's blog