POJ1741 Tree dp+树的点分治(楼天成推荐男人必做八题之一)

题目大意就是给一个无根树，让你找出这个书中两个点之间距离小于K的点对有多少对。。。。典型的树的点分治，无奈编码能力太弱，变了两天都没写好，太弱了，最后听说是楼教推荐的男人八题，我艹，男人不好当啊。。。。。。。刚开始搞树的分治，所以自己写的丑代码就不贴出来了，在网上找了一份精简的分享一下：具体树分治的知识参见漆子超09年的论文。。。。

将无根树转化成有根树进行观察。满足条件的点对有两种情况：两个点的路径横跨树根，两个点位于同一颗子树中。

如果我们已经知道了此时所有点到根的距离a[i]，a[x] + a[y] <= k的(x, y)对数就是结果，这个可以通过排序之后O(n)的复杂度求出。然后根据分治的思想，分别对所有的儿子求一遍即可，但是这会出现重复的——当前情况下两个点位于一颗子树中，那么应该将其减掉（显然这两个点是满足题意的，为什么减掉呢？因为在对子树进行求解的时候，会重新计算）。

在进行分治时，为了避免树退化成一条链而导致时间复杂度变为O(N^2)，每次都找树的重心，这样，所有的子树规模就会变的很小了。时间复杂度O(Nlog^2N)。

树的重心的算法可以线性求解。

 

 

 

/**********************
* author:crazy_石头
* Pro:POJ 1987
* algorithm:Tree dp+点分治
* Time:31ms
* Judge Status:Accepted
***********************/
#include 
#include 
#include 
#include 
#include 
#include 
#include 

using namespace std;

#define rep(i,h,n) for(int i=(h);i<=(n);i++)
#define ms(a,b) memset((a),(b),sizeof(a))
#define eps 1e-6
#define INF 1<<29
#define LL __int64
const int N=20000+5;
const int maxm=200+10;

struct node
{
	int v, l;
	node() {};
	node(int _v, int _l): v(_v), l(_l) {};
};

vector g[N];
int n, k, size, s[N], f[N], root, d[N], K, ans;

vector dep;
bool done[N];

void getroot(int now, int fa)//找树的重心
{
	int v;
	s[now] = 1; f[now] = 0;
	for (int i=0; i

 

转自blog：http://blog.csdn.net/yang_7_46/article/details/9966455