[2017CCPC杭州] HDU 6268 点分治+bitset

最新推荐文章于 2020-06-22 21:00:44 发布

SigmaQuadrant

最新推荐文章于 2020-06-22 21:00:44 发布

阅读量122

点赞数

分类专栏：比赛题解

本文链接：https://blog.csdn.net/qq_40859782/article/details/101673630

版权

比赛题解专栏收录该内容

218 篇文章 0 订阅

订阅专栏

给出一个 $n\leq3000$ 个点的树，树上每个结点都有 $w_i\leq1e5$ 的点权。现在要求对于 $\forall i \in [1,m]$ 所有的数，是否存在一个连通子图满足点权和等于 $i$ ， $m\leq1e5$ 。
因为值域范围不会超过 $1 e 5$ ，给每个点开一个 $b i t s e t$ ， $b i t s e t$ 的每一位就代表这个值是否存在，暴力转移 $b i t s e t$ 。
避免退化成链，考虑对重心分治，复杂度为 $O(n^2logn/w)$

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int N=3e3+7;
bitset<100001> f[N];  
bitset<100001> ans;
vector<int> go[N]; 
bool used[N];
int w[N],mx[N],sz[N];
int tot,rt;
void getrt(int u,int fa) {
	mx[u]=0;sz[u]=1;
	for(auto &v:go[u]) {
		if(used[v]) continue;
		if(v==fa) continue;
		getrt(v,u);
		mx[u]=max(mx[u],sz[v]);
		sz[u]+=sz[v];
	}
	mx[u]=max(mx[u],tot-sz[u]);
	if(mx[u]<mx[rt]) rt=u;
}
void dfs(int u,int fa) {
	f[u]<<=w[u];
	sz[u]=1;
	for(auto &v:go[u]) {
		if(used[v]) continue;
		if(v==fa) continue;
		f[v]=f[u];
		dfs(v,u);
		sz[u]+=sz[v];
		f[u]|=f[v];
	}
}
void solve(int u) {
	used[u]=1;
	f[u]=1;
	dfs(u,0);
	ans|=f[u];
	for(auto &v:go[u]) {
		if(!used[v]) {
			tot=sz[v];
			mx[rt=0]=1e9;
			getrt(v,u);
			solve(rt);
		}
	}
}
int main() {
	int T;
	scanf("%d",&T);
	while(T--) {
		memset(used,0,sizeof(used)); 
		ans.reset();
		int n,m;
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++) 
			go[i].clear();
		for(int i=1;i<n;i++) {
			int u,v;
			scanf("%d%d",&u,&v);
			go[u].push_back(v);	
			go[v].push_back(u);	
		}
		for(int i=1;i<=n;i++)
			scanf("%d",&w[i]);
		rt=0;tot=n;mx[rt]=1e9;
		getrt(1,0);
		solve(rt); 
		for(int i=1;i<=m;i++)
			printf("%d",(int)ans[i]); 
		printf("\n"); 
	}
	return 0;
}