标签：int 题解 Codeforces Evil fa init root mx dis

题目链接：G. Spiders Evil Plan

题目大意：洛谷

题解：首先，\(y\) 条路径可以覆盖 \(2y\) 个叶子，并且容易证明其中一定有一个叶子是直径的一个端点，所以我们以两个直径的端点分别为根建树，然后从大向小取，然后考虑两种调整方案，就可以了。

可以直接拿长链剖分 + 倍增维护。

时间复杂度 \(O(n\log n)\)。

#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int Maxn=100000;
int n,q;
int head[Maxn+5],arrive[Maxn<<1|5],val[Maxn<<1|5],nxt[Maxn<<1|5],tot;
void add_edge(int from,int to,int w){
	arrive[++tot]=to;
	val[tot]=w;
	nxt[tot]=head[from];
	head[from]=tot;
}
struct Tree{
	int mx_dis,root;
	int dis[Maxn+5],max_dis[Maxn+5],dep[Maxn+5];
	int fa[17][Maxn+5],son[Maxn+5],top[Maxn+5],len[Maxn+5];
	vector<int> v[Maxn+5];
	void init_dfs_1(int u){
		dep[u]=dep[fa[0][u]]+1;
		for(int i=1;fa[i-1][fa[i-1][u]];i++){
			fa[i][u]=fa[i-1][fa[i-1][u]];
		}
		max_dis[u]=dis[u];
		if(mx_dis==0||dis[mx_dis]<dis[u]){
			mx_dis=u;
		}
		for(int i=head[u];i;i=nxt[i]){
			int v=arrive[i];
			if(v==fa[0][u]){
				continue;
			}
			fa[0][v]=u;
			dis[v]=dis[u]+val[i];
			init_dfs_1(v);
			max_dis[u]=max(max_dis[u],max_dis[v]);
			if(son[u]==0||max_dis[son[u]]<max_dis[v]){
				son[u]=v;
			}
		}
	}
	void init_dfs_2(int u,int tp){
		top[u]=tp;
		v[tp].push_back(u);
		if(son[u]){
			init_dfs_2(son[u],tp);
		}
		for(int i=head[u];i;i=nxt[i]){
			int v=arrive[i];
			if(v==fa[0][u]||v==son[u]){
				continue;
			}
			init_dfs_2(v,v);
			len[v]=max_dis[v]-dis[u];
		}
	}
	int ans[Maxn+5];
	int col[Maxn+5],s[Maxn+5],c;
	bool cmp(int p,int q){
		return len[p]>len[q];
	}
	int init(int x){
		mx_dis=0;
		dis[x]=0;
		init_dfs_1(x);
		root=mx_dis;
		memset(dis,0,sizeof dis);
		memset(fa,0,sizeof fa);
		memset(dis,0,sizeof dis);
		memset(son,0,sizeof son);
		mx_dis=0;
		init_dfs_1(root);
		init_dfs_2(root,root);
		len[root]=max_dis[root];
		for(int i=1;i<=n;i++){
			if(top[i]==i){
				s[++c]=i;
			}
		}
		sort(s+1,s+1+c,[&](int p,int q){return len[p]>len[q];});
		for(int i=1;i<=c;i++){
			ans[i]=ans[i-1]+len[s[i]];
			for(int u=s[i];u;u=son[u]){
				col[u]=i;
			}
		}
		return root;
	}
	int find_ans(int x,int y,int *a){
		int u=x;
		for(int i=16;i>=0;i--){
			if(col[fa[i][u]]>y){
				u=fa[i][u];
			}
		}
		u=fa[0][u];
		return ans[y]+max_dis[x]-a[u];
	}
	int query(int x,int y){
		y=(y<<1)-1;
		if(y>=c){
			return ans[c];
		}
		if(col[x]<=y){
			return ans[y];
		}
		return max(find_ans(x,y-1,dis),find_ans(x,y,max_dis));
	}
}tree_1,tree_2;
int main(){
	scanf("%d%d",&n,&q);
	for(int i=1;i<n;i++){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add_edge(u,v,w);
		add_edge(v,u,w);
	}
	tree_2.init(tree_1.init(1));
	int ans=0;
	for(int i=1;i<=q;i++){
		int x,y;
		scanf("%d%d",&x,&y);
		x=(x+ans-1)%n+1;
		y=(y+ans-1)%n+1;
		printf("%d\n",(ans=max(tree_1.query(x,y),tree_2.query(x,y))));
	}
	return 0;
}

标签：int,题解,Codeforces,Evil,fa,init,root,mx,dis
来源： https://www.cnblogs.com/withhope/p/13683160.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。