bzoj2599

第一道点分治。

给一棵树,每条边有权.求一条路径,权值和等于K,且边的数量最小.

 

开一个100W的数组t,t[i]表示权值为i的路径最少边数

找到重心分成若干子树后， 得出一棵子树的所有点到根的权值和x，到根a条边，用t[k-x]+a更新答案，全部查询完后

然后再用所有a更新t[x]

这样可以保证不出现点分治中的不合法情况

把一棵树的所有子树搞完后再遍历所有子树恢复T数组，如果用memset应该会比较慢

 

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdlib>
using namespace std;
const int N=200005;
const int inf=0x3f3f3f3f;
inline int read()
{
	int ans,f=1;char ch;
	while ((ch=getchar())<'0'||ch>'9') if (ch=='-') f=-1;ans=ch-'0';
	while ((ch=getchar())>='0'&&ch<='9') ans=ans*10+ch-'0';
	return ans*f;
}

int n,k;

int head[N],tot;
struct aa
{
	int dis,to,pre;
}edge[N*2];
void addedge(int x,int y,int z)
{
	edge[++tot].to=y;edge[tot].dis=z;edge[tot].pre=head[x];head[x]=tot;
}
int f[1000009],ans;
int rt,sum,mx,size[N];//ÖØÐÄÊý×é 
bool vis[N];

void getroot(int u,int fa)
{
	size[u]=1;
	int tmp=0;
	for (int i=head[u];i;i=edge[i].pre)
	if (fa!=edge[i].to&&!vis[edge[i].to])
	{
		getroot(edge[i].to,u);
		size[u]+=size[edge[i].to];
		tmp=max(tmp,size[edge[i].to]);
	}
	tmp=max(tmp,sum-size[u]);
	if (tmp<mx) mx=tmp,rt=u;
}
void cal(int u,int fa,int d,int dis)//dis: the distance to the root;d: the number of edge to the root;   
{
	if (dis<=k) ans=min(ans,f[k-dis]+d);
	for (int i=head[u];i;i=edge[i].pre)
	if (fa!=edge[i].to&&!vis[edge[i].to]) 
	cal(edge[i].to,u,d+1,dis+edge[i].dis);
}
void add(int u,int fa,int flag,int d,int dis)
{
	if (dis<=k) 
	{
		if (flag) f[dis]=min(f[dis],d);
		else f[dis]=inf;
	}
	for (int i=head[u];i;i=edge[i].pre)
	if (edge[i].to!=fa&&!vis[edge[i].to]) 
	add(edge[i].to,u,flag,d+1,dis+edge[i].dis);
}
void work(int u)
{
	vis[u]=true;f[0]=0;
	for (int i=head[u];i;i=edge[i].pre)
	if (!vis[edge[i].to])
	{
		cal(edge[i].to,0,1,edge[i].dis);
		add(edge[i].to,0,1,1,edge[i].dis);
	}
	for (int i=head[u];i;i=edge[i].pre)
	if (!vis[edge[i].to]) 
		add(edge[i].to,0,0,1,edge[i].dis);
	
	for (int i=head[u];i;i=edge[i].pre)
	if (!vis[edge[i].to]) 
	{
		rt=0,sum=size[edge[i].to],mx=inf;
	 	getroot(edge[i].to,0);
		work(rt);
	}
}
int main()
{
	n=read(),k=read();
	
	int x,y,z;
	for (int i=1;i<n;i++)
	{
		x=read()+1,y=read()+1,z=read();
		addedge(x,y,z);
		addedge(y,x,z);
	}
	
	ans=n;
	sum=n,mx=inf;
	memset(f,inf,sizeof(f));
	getroot(1,0);
	work(rt);
	if (ans!=n) printf("%d",ans);
	else printf("-1");
	return 0;
}

总结

1：

所谓点分治 对于一条树路径 只有经过或不经过一个点的情况

对于不经过的情况 把一棵树按这个点拆成好几棵分治就行了

一般是用于树上路径问题当中的。

2：其中关于重心查找以及dfs的一般形式是作为模板出现的，基本思路也差不多。