【poj1741】tree 点分治

【题目大意】

给一颗n个节点的树,每条边上有一个距离v(v<=1000)。

定义d(u,v)为u到v的最小距离;

给定k值,求有多少点对(u,v)使u到v的距离小于等于k。

其中n<=10000,k<2^31

【题解】

蒟蒻发现很久以前学的点分治已经忘得差不多了,于是恶补一发

所谓点分治,就是枚举每一个点为根的情况,计算当前根对答案的贡献。

具体可以参考http://www.cnblogs.com/chty/p/5912360.html(这是我以前的博客)

#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define FILE "read"
#define MAXN 10010
#define INF 1000000000
#define up(i,j,n) for(int i=j;i<=n;++i)
#define dn(i,j,n) for(int i=j;i>=n;--i) 
namespace INIT{
	char buf[1<<15],*fs,*ft;
	inline char getc(){return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;}
	inline int read(){
		int x=0,f=1;  char ch=getc();
		while(!isdigit(ch))  {if(ch=='-')  f=-1;  ch=getc();}
		while(isdigit(ch))  {x=x*10+ch-'0';  ch=getc();}
		return x*f;
	}
}using namespace INIT;
struct node{int y,next,v;}e[MAXN*2];
int n,k,root,sum,len,ans,Link[MAXN],f[MAXN],dis[MAXN],deep[MAXN],vis[MAXN],size[MAXN];
void insert(int x,int y,int v){e[++len].next=Link[x];Link[x]=len;e[len].y=y;e[len].v=v;}
void getroot(int x,int fa){
	size[x]=1; f[x]=0;
	for(int i=Link[x];i;i=e[i].next){
		if(e[i].y==fa||vis[e[i].y])  continue;
		getroot(e[i].y,x);  size[x]+=size[e[i].y];
		f[x]=max(f[x],size[e[i].y]);
	}
	f[x]=max(f[x],sum-size[x]);
	if(f[x]



你可能感兴趣的:(poj,点分治)