Codechef Union on Tree(点分树,虚树)

6 篇文章 0 订阅
5 篇文章 0 订阅
本文探讨如何使用点分树和虚树解决Codechef上的一个问题,涉及计算每个点最大拓展距离以及如何避免重复计数。通过建立虚树,计算联通块的点数,并减去共同覆盖的点数,最终得出正确答案。同时,介绍了计算中点覆盖点数的方法。
摘要由CSDN通过智能技术生成

在这里插入图片描述
建虚树。
然后在虚树上求出每个点最多可以往外拓展多少距离(意思就是可以由其他点拓展到这个点之后继续拓展)。
那么按照这个定义我们可以得到:
一个点如果被覆盖,那么一定可以覆盖他的点在虚树上是一个联通块。
因为树上连通块满足点数-边数=1。
那么我们把每个点可以覆盖的点数相加,再枚举每条边,减去边的两个点都能覆盖的点数,
那么考虑一个被覆盖的点,能覆盖他的那个联通块会刚好计算了它一次。
那么这样就不重不漏的计算了所有被覆盖的点。
边上两个点共同覆盖的点可以通过计算他们的中点覆盖了多少个点来得到。
计算距离一个点 r r r 的点数可以建出点分树。

A C   C o d e \mathcal AC \ Code AC Code

#include<bits/stdc++.h>
#define maxn 100005
#define lim 17
#define LL long long
#define rep(i,j,k) for(int i=(j),LIM=(k);i<=LIM;i++)
#define per(i,j,k) for(int i=(j),LIM=(k);i>=LIM;i--)
#define ADJ(i,u) for(int i=info[u],v;i;i=Prev[i])
using namespace std;

int n,lg[maxn<<1];
int info[maxn],Prev[maxn<<1],to[maxn<<1],cnt_e;
void Node(int u,int v){ Prev[++cnt_e]=info[u],info[u]=cnt_e,to[cnt_e]=v; }

//DCT
namespace DCT{
	int sz[maxn],vis[maxn],fa[maxn],dep[maxn],dis[lim][maxn];
	
	void dfs(int u,int ff,int tsz,int &mn,int &rt){
		int mx = 0;sz[u] = 1;
		ADJ(i,u) if((v=to[i])^ff && !vis[v])
			dfs(v,u,tsz,mn,rt) , mx = max(mx , sz[v]) , sz[u] += sz[v];
		if((mx = max(mx , tsz - sz[u])) < mn) 
			mn = mx , rt = u;
	}
	
	int Gert(int u,int tsz){
		int mn,rt;
		dfs(u,0,tsz,mn=0x3f3f3f3f,rt);
		return rt;
	}
	
	vector<int>G[maxn],Gx[maxn];
	
	void ser(int u,int ff,vector<int>&t,int d,int dp=-1){
		sz[u] = 1;
		while(t.size() <= d) t.push_back(0);
		assert(d < t.size());
		t[d] += u<=n;
		
		if(~dp) assert(dp >= 0 && dp < lim) , dis[dp][u] = d;
		ADJ(i,u) if((v=to[i])^ff && !vis[v])
			ser(v,u,t,d+1,dp),sz[u] += sz[v];
	}
	
	void prc(vector<int>&t){ rep(i,1,t.size()-1) t[i]+=t[i-1]; }
	
	void Solve(int u){
		vis[u] = 1;
		ser(u,0,G[u],0,dep[u]);
		prc(G[u]);
		ADJ(i,u) if(!vis[v=to[i]]){
			int t = Gert(v,sz[v]);
			ser(v,u,Gx[t],1) , fa[t] = u;
			prc(Gx[t]);
			dep[t] = dep[u] + 1;
			Solve(t);
		}
	}
	
	int qry(vector<int>&t,int d){
		return t.empty() || d < 0 ? 0 : t[min(d,(int)t.size()-1)];
	}
	
	int qry(int u,int d){ 
		int r = 0 , x = u;
		for(int p=0;u;u=fa[p=u]) r += qry(G[u],d-dis[dep[u]][x]) - qry(Gx[p] , d-dis[dep[u]][x]) /*, printf("@%d %d %d %d %d\n",x,u,d,dis[dep[u]][x],r)*/;
		return r;
	}
}

//Tree and Lca
namespace Tree{
	int dep[maxn],mn[lim+1][maxn<<1],fa[maxn],st[maxn],ed[maxn],pos[maxn],f[lim][maxn],tot,tim;
	int ont[maxn],r[maxn];
	void dfs0(int u,int ff){
		mn[0][pos[u] = ++tim] = u , st[u] = ++tot;
		dep[u] = dep[f[0][u] = ff] + 1;
		ADJ(i,u) if((v=to[i])^ff)
			dfs0(v,u) , mn[0][++tim] = u;
		ed[u] = tot;
	}
	int Min(int u,int v){ return dep[u] < dep[v] ? u : v;	}
	int Lca(int u,int v){
		u = pos[u] , v = pos[v];
		if(u > v) swap(u,v);
		int t = lg[v-u+1];
		return Min(mn[t][u] , mn[t][v-(1<<t)+1]);
	}
	void init(){
		dfs0(1,0);
		rep(j,1,lim) rep(i,1,tim-(1<<j)+1) mn[j][i] = min(mn[j-1][i] , mn[j-1][i+(1<<j-1)]);
		rep(j,1,lim-1) rep(i,1,2*n-1) f[j][i] = f[j-1][f[j-1][i]];
		rep(j,2,tim) lg[j] = lg[j >> 1] + 1; 
	}
	bool cmp(const int &u,const int &v){
		return st[u] < st[v];
	} 
	int getk(int u,int k){
		rep(i,0,lim-1) if(k>>i&1)
			u=f[i][u];
		return u;
	}
	
	void Solve(){
		int K;
		scanf("%d",&K);
		static int ar[maxn]={},x,pt[maxn]={},cnt=0,q[maxn]={},tp;
		cnt = 0;
		rep(i,1,K) scanf("%d%d",&ar[i],&x),r[ar[i]]=x<<1,ont[ar[i]]=1;
		sort(ar+1,ar+1+K,cmp);
		rep(i,1,K){
			if(tp){
				int t = Lca(ar[i],q[tp]),p=0;
				for(;tp && dep[q[tp]] > dep[t];p = q[tp--]) if(p) fa[p] = q[tp];
				if(q[tp] ^ t) pt[++cnt] = q[++tp] = t;
				if(p) fa[p] = q[tp];
			}
			pt[++cnt] = q[++tp] = ar[i];
		}
		int p=0;
		for(;tp;p=q[tp--]) if(p) fa[p]=q[tp];
		sort(pt+1,pt+1+cnt,cmp);
		rep(i,1,cnt) if(!ont[pt[i]]) r[pt[i]] = -1;
		per(i,cnt,2) r[fa[pt[i]]] = max(r[fa[pt[i]]] , r[pt[i]] - dep[pt[i]] + dep[fa[pt[i]]]);
		rep(i,2,cnt) r[pt[i]] = max(r[pt[i]] , r[fa[pt[i]]] - dep[pt[i]] + dep[fa[pt[i]]]);
		LL ans = 0;
		rep(i,1,cnt) if(r[pt[i]] >= 0) ans += DCT::qry(pt[i],r[pt[i]]);
		rep(i,2,cnt) if(r[pt[i]] >= 0 && r[fa[pt[i]]] >= 0){
			int z = getk(pt[i] , (dep[pt[i]] - dep[fa[pt[i]]] - r[fa[pt[i]]] + r[pt[i]]) >> 1);
			ans -= DCT::qry(z,r[pt[i]] - dep[pt[i]] + dep[z]);
		}
		rep(i,1,cnt) ont[pt[i]] = 0;
		printf("%lld\n",ans);
	}
}

int main(){
	
	scanf("%d",&n);
	rep(i,1,n-1){
		int u,v;scanf("%d%d",&u,&v);
		Node(u,n+i),Node(n+i,u),Node(v,n+i),Node(n+i,v);
	}
	DCT::Solve(DCT::Gert(1,n*2-1));
	Tree::init();
	int Q;
	scanf("%d",&Q);
	for(;Q--;) Tree::Solve();
}
//1235432143214
/*
	点分树
	虚树
	LCA 
	dfs序
	树上k级祖先 
*/
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。
1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。1、资源项目源码均已通过严格测试验证,保证能够正常运行; 2、项目问题、技术讨论,可以给博主私信或留言,博主看到后会第一时间与您进行沟通; 3、本项目比较适合计算机领域相关的毕业设计课题、课程作业等使用,尤其对于人工智能、计算机科学与技术等相关专业,更为适合; 4、下载使用后,可先查看README.md文件(如有),本项目仅用作交流学习参考,请切勿用于商业用途。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值