虚树模板


vector<pair<int,int>>imp;

vector<int>G[MAX];

int s[MAX],s_top;

void build()
{
	sort(imp.begin(),imp.end());
	//为了方便处理,为我们把树根固定当做关键点 
	s[s_top=1]=1;
	//按照dfs序遍历每个点
	for (int i=0;i<imp.size();i++){
	  int v=imp[i].second;
	  int t=lca(v,s[s_top]);
	  //t的dfs序肯定小于栈顶元素(因为他是v和栈顶元素的LCA) 
	  while (dfn[t]<dfn[s[s_top]]){
	  	//直到栈顶的元素的前一个元素的dfs序小于当前的LCA,停止循环,如果当前的LCA不在栈中,那么入栈
		  if(dfn[t]>=dfs[s[s_top-1]]){
		  	G[t].push_back(s[s_top]);
		  	if (s[--s_top]!=t) s[++s_top]=t;
		  	break;
		  } 
		  
		  //进入循环以为着栈顶元素dfs小于当前的LCA,并且沃恩是按照dfs序遍历的,那么栈顶的元素的子树肯定已经构建完毕,需要出栈并对当前的LCA建边
		  G[s[s_top-1]].push_back(s[s_top]);
		  s_top--; 
	  } 
	  s[++s_top]=v;
	} 
	while (s_top--) G[s[s_top]].push_back(s[s_top+1]); 
 } 
全部评论

相关推荐

比亚迪汽车新技术研究院 硬件工程师 总包21左右 硕士
点赞 评论 收藏
分享
10-06 12:46
门头沟学院 Java
跨考小白:定时任务启动
点赞 评论 收藏
分享
11-14 16:13
已编辑
重庆科技大学 测试工程师
Amazarashi66:不进帖子我都知道🐮❤️网什么含金量
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务