POJ1741 Tree

前言

今天刚刚学了点分治,赶紧把思路写一下,以防止以后忘记

分析

对于一个点,经过它的路径有两种可能,一是在一棵子树里(下图1-2-4),二是在两个子树里(下图2-1-5)。
POJ1741 Tree_第1张图片
那么对于情况二,我们可以把它拆成两条链相加。
对于情况一好像不行,但如果去分类讨论就会让程序更加复杂,但是发现它在它的子树里处理这种情况时就成了情况二,所以可以暂时不用管它,在递归下一层的时候返回来处理就行,不过需要加一个vis数组,防止情况搜多,这样就能够处理这种情况了。
但是有一种路径需要被减去,即合并(1-2-3和1-2-4)这种路径,因为显然不能合并啊,所以要减去。
减的方法是在往下递归的时候,以连向父节点的那条边的权值为起始值处理一遍答案,这种就相当于,模拟了一下从1搜下来的时候的情况,但是只搜这一棵树,就能够求出曾经多算的情况。
还有为了防止不会搜太多层,需要从重心开始搜。

#include
#include
#include
using namespace std;
const int lqs=1e4+10;
struct Edge{
	int to,nxt,val;
}e[lqs<<1];
int h[lqs],idx;
void Ins(int a,int b,int c){
	e[++idx].to=b;e[idx].val=c;
	e[idx].nxt=h[a];h[a]=idx;
}
int n,k,rt,T,siz[lqs],w[lqs];
bool vis[lqs];
void dfs1(int u,int fa){
	siz[u]=1;w[u]=0;
	for(int i=h[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(vis[v]||v==fa)continue;
		dfs1(v,u);
		siz[u]+=siz[v];
		w[u]=max(w[u],siz[v]);
	}
	w[u]=max(w[u],T-siz[u]);
	if(w[u]k)j--;
		if(i>j)break;
		sum+=j-i+1;
	}
	return sum;
}
void solve(int u){
	ans+=calc(u,0);
	vis[u]=1;
	for(int i=h[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(vis[v])continue;
		ans-=calc(v,e[i].val);
		T=siz[v];rt=0;
		dfs1(v,0);
		solve(rt);
	}
}
void init(){
	memset(h,0,sizeof(h));
	ans=rt=idx=0;
	memset(vis,0,sizeof(vis));
}
int main(){
	w[0]=0x7f7f7f7f;
	while(~scanf("%d%d",&n,&k)&&n&&k){
		init();
		for(int i=1;i

你可能感兴趣的:(POJ1741 Tree)