【图论】点分治

时间复杂度:
【图论】点分治_第1张图片
洛谷P3806(100分加强数据版)

#include
#define ll long long
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define per(i,n,a) for(int i=n;i>=a;i--)
#define endl '\n'
#define mem(a,b) memset(a,b,sizeof(a))
#define IO ios::sync_with_stdio(false);cin.tie(0);
using namespace std;
const int INF=0x3f3f3f3f;
const ll inf=0x3f3f3f3f3f3f3f3f;
const int mod=1e9+7;
const int maxn=1e5+5;
const int maxk=1e7+5;
const int limit=1e7;
int tot,head[maxn];
struct E{
	int to,next,w;
}edge[maxn<<1];
void add(int u,int v,int w){
	edge[tot].to=v;
	edge[tot].w=w;
	edge[tot].next=head[u];
	head[u]=tot++;
}
int n,m,rt,sum,cnt,q[maxn];
int tmp[maxn],siz[maxn],dis[maxn],maxp[maxn];
bool judge[maxk],ans[maxn],vis[maxn];
void getrt(int u,int f){
	siz[u]=1,maxp[u]=0;
	for(int i=head[u];i!=-1;i=edge[i].next){
		int v=edge[i].to;
		if(v==f||vis[v]) continue;
		getrt(v,u);
		siz[u]+=siz[v];
		if(siz[v]>maxp[u]) maxp[u]=siz[v];
	}
	maxp[u]=max(maxp[u],sum-siz[u]);
	if(maxp[u]<maxp[rt]) rt=u;
}
void getdis(int u,int f){
	if(dis[u]<=limit) tmp[cnt++]=dis[u];
	for(int i=head[u];i!=-1;i=edge[i].next){
		int v=edge[i].to;
		if(v==f||vis[v]) continue;
		dis[v]=dis[u]+edge[i].w;
		getdis(v,u);
	}
}
void solve(int u){
	queue<int> que;
	for(int i=head[u];i!=-1;i=edge[i].next){
		int v=edge[i].to;
		if(vis[v]) continue;
		cnt=0;
		dis[v]=edge[i].w;
		getdis(v,u);
		for(int j=0;j<cnt;j++)
			for(int k=0;k<m;k++)
				if(q[k]>=tmp[j])
					ans[k]|=judge[q[k]-tmp[j]];
		for(int j=0;j<cnt;j++){
			que.push(tmp[j]);
			judge[tmp[j]]=true;
		}
	}
	while(!que.empty()){
		judge[que.front()]=false;
		que.pop();
	}
}
void divide(int u){
	vis[u]=judge[0]=true;
	solve(u);
	for(int i=head[u];i!=-1;i=edge[i].next){
		int v=edge[i].to;
		if(vis[v]) continue;
		maxp[rt=0]=sum=siz[v];
		getrt(v,0);
		getrt(rt,0);
		divide(rt);
	}
}
int main(){
	scanf("%d%d",&n,&m);mem(head,-1);
	for(int i=1;i<n;i++){
		int u,v,w;scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);add(v,u,w);
	}
	for(int i=0;i<m;i++) scanf("%d",&q[i]);
	maxp[0]=sum=n;
	getrt(1,0);
	getrt(rt,0);
	divide(rt);
	for(int i=0;i<m;i++){
		if(ans[i]) puts("AYE");
		else puts("NAY");
	}
}


你可能感兴趣的:(图论)