POJ - 1741 树的点分治

第一次接触树分治。打算今晚学的，先看了一蛤树的重心怎么求，之后知道了重心后再来看这道题，这种题是不是启发式合并也可以？一遍一遍加边地合并上去，然后枚举小的块的每个点，二分大的块，然而这还是要遍历一遍小的块再遍历一遍大的块才能预处理出道链接的两个点的距离，就不满足启发式合并的nlogn了，因为要对大的块进行遍历，燃热这个n<1e4，不是很大，说不定能水过去，然而T了。。。果然是楼教主的男人8题，极限数据很强。于是还是老老实实写树分治。

理解了一蛤以后，树的点分治就是求出跟这个点有关的所有情况，然后把所有点的与该点有关的情况加在一起，得到最后的答案，在这道题中，就是枚举必须经过当前点的路径有多少<=k的。由于枚举子树中所有dis[u]+dis[v]<=k的情况可以O(n)扫描得到，这个总数为，所以先加上这个总数，再减掉那些从同一个子树中到达根节点又回到同一个子树的路径对刚才总数的贡献，就得到了从一个子树经过该点到另外一个子树中对答案的所有贡献。(还是有点没想清楚，过一点时间再来看看)

网上代码求树的重心有问题，然而还是A了。然而网上代码的树的点分治写法十分优秀，直接定义了一个结构体centertree，然后再内部定局部变量，方便每次找子树后找到重心再变成一棵新树。

2020.1.12 upd

终于懂那里为什么要减去每一棵相同子树的贡献了，因为长度相加得到<=j只能是经过当前u点且去往不同子树的路径，而无视所有路径组合，相同子树到root的距离加起来不是他们之间的距离，所以算完以后要减去每一棵子树中自己的路径组合。

 

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define maxl 10010
#define inf 2000000001

using namespace std;

int n,k,cnt,len,size,ans;
int dis[maxl],ehead[maxl],data[maxl];
struct ed
{
	int to,nxt,l;
}e[maxl<<1];
bool vis[maxl];

inline int max(int a,int b)
{
	if(a>b)
		return a;
	else
		return b;
}

struct cntertree
{
	int n,ans,size;
	int son[maxl];
	void dfs(int u,int fa)
	{
		son[u]=1;
		int res=0;
		for(int i=ehead[u];i;i=e[i].nxt)
		{
			int v=e[i].to;
			if(v==fa || vis[v]) continue;
			dfs(v,u);
			son[u]+=son[v];
			res=max(res,son[v]); //why son[v]-1? can ac 
		}
		res=max(res,n-son[u]);
		if(res<size)
		{
			ans=u;
			size=res;
		}
	}
	int getcenter(int x)
	{
		ans=0;size=inf;
		dfs(x,0);
		return ans;
	}
}tree;

inline void add(int u,int v,int l)
{
	e[++cnt].to=v;e[cnt].l=l;e[cnt].nxt=ehead[u];ehead[u]=cnt;
}

inline void prework()
{
	int u,v,l;
	cnt=0;
	memset(ehead,0,sizeof(ehead));
	memset(vis,false,sizeof(vis));
	for(int i=1;i<=n-1;i++)
	{
		scanf("%d%d%d",&u,&v,&l);
		add(u,v,l);add(v,u,l);
	}
}

void getdata(int u,int fa)
{
	data[++len]=dis[u];
	int v;
	for(int i=ehead[u];i;i=e[i].nxt)
	{
		v=e[i].to;
		if(v==fa || vis[v]) continue;
		dis[v]=dis[u]+e[i].l;
		getdata(v,u);
	}
}

inline int calc(int u,int w)
{
	dis[u]=w;len=0;
	getdata(u,0);
	sort(data+1,data+len+1);
	int sum=0,l=1,r=len;
	while(l<r)
	{
		if(data[r]+data[l]<=k)
			sum+=(r-l),l++;
		else
			r--;
	}
	return sum;
}

void solve(int u)
{
	ans+=calc(u,0);
	vis[u]=true;
	int v;
	for(int i=ehead[u];i;i=e[i].nxt)
	{
		v=e[i].to;
		if(vis[v]) continue;
		ans-=calc(v,e[i].l);	//减掉同一棵子树中的答案 (why?)
		tree.n=tree.son[v];
		int rt=tree.getcenter(v);
		solve(rt);
	}
}

inline void mainwork()
{
	ans=0;
	tree.n=n;
	int rt=tree.getcenter(1);
	solve(rt);
}

inline void print()
{
	printf("%d\n",ans);
}

int main()
{
	while(~scanf("%d%d",&n,&k))
	{
		if(n==0 && k==0)
			break;
		prework();
		mainwork();
		print();
	}
	return 0;
}