bzoj3697 采药人的路径 （点分治）

bzoj3697 采药人的路径

原题地址：http://www.lydsy.com/JudgeOnline/problem.php?id=3697

题意：
采药人的药田是一个树状结构，每条路径上种植一种药，有0/1两种药。
草药人希望选择一条两种药材数目相等的路径，且选出的路径中有一个可以作为休息站的节点（不包括起点和终点），满足起点到休息站和休息站到终点的路径也是两种药数量相等。
他一共可以选择多少种不同的路径。

数据范围
N ≤ 100,000

题解：
如果把0的药看做-1，那么就是选择一条路径，他的边权和为零，且存在一个点，使分成路径的两侧边权和也为零。
1.休息站在一条从根出发的路径上：
那么该路径上存在一个点，根到他的边权和与根到端点的边权和相等。 他可以与其他子树的边权和为相反数的路径组成合法路径。
2.休息站在根。
两条不在同一子树的边权和为0的路径互相组合。

于是在点分治时，统计满足以上的路径条数。
为了区分子树，用f,g两个数组。
f[i][0/1]表示当前子树路径和为i且这条路径上方是否有过和为i的前缀，
g[i][0/1]分别表示之前遍历的所有子树路径和为i且这条路径上方是否有过和为i的前缀，
初值为g[i][0]=1，要考虑到由根节点出发的路径。
ans+=f[i][1]*(g[-i][0]+g[-i][1])+f[i][0]*g[-i][1]
ans+=f[0][0]*(g[0][0]-1)
清空只清空赋了值的。

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define LL long long