题目描述
比特山是比特镇的飙车圣地。在比特山上一共有$n$个广场,编号依次为$1$到$n$,这些广场之间通过$n−1$条双向车道直接或间接地连接在一起,形成了一棵树的结构。
因为每条车道的修建时间以及建筑材料都不尽相同,所以可以用两个数字$l_i,r_i$量化地表示一条车道的承受区间,只有当汽车以不小于$l_i$且不大于$r_i$的速度经过这条车道时,才不会对路面造成伤害。
$Byteasar$最近新买了一辆跑车,他想在比特山飙一次车。$Byteasar$计划选择两个不同的点$S,T$,然后在它们树上的最短路径上行驶,且不对上面任意一条车道造成伤害。
$Byteasar$不喜欢改变速度,所以他会告诉你他的车速。为了挑选出最合适的车速,$Byteasar$一共会向你询问$m$次。请帮助他找到一条合法的道路,使得路径上经过的车道数尽可能多。
输入格式
第一行包含两个正整数$n,m$,表示广场的总数和询问的总数。
接下来$n−1$行,每行四个正整数$u_i,v_i,l_i,r_i$,表示一条连接$u_i$和$v_i$的双向车道,且承受区间为$[l_i,r_i]$。
接下来$m$行,每行一个正整数$q_i$,分别表示每个询问的车速。
输出格式
输出$m$行,每行一个整数,其中第$i$行输出车速为$q_i$时的最长路径的长度,如果找不到合法的路径则输出$0$。
样例
我们将速度$l,r$看成区间,那么我们就是要找速度是$q_i$时所能经过多少连续区间。
现将问题离线。
那么可以用线段树维护边,对于线段树上区间$l,r$,我们将边的这条边压进去。
然后我们在线段树上往上合并每一条边,用并查集维护是否已经连到了一起,合并时一共分为$6$种情况;注意并查集不能路径压缩,要按秩合并,因为我们在合并的同时还要维护信息。
对于合并时求距离,还是用$LCA$,但是倍增$LCA$常数较大,建议用树链剖分。
时间复杂度:$\Theta(n\log^2n)$。
期望得分:$100$分。
实际得分:$100$分。
代码时刻
#include<bits/stdc++.h> #define L(x) x<<1 #define R(x) x<<1|1 using namespace std; struct rec{int nxt,to;}e[150000]; struct node{int x,y,v;pair<int,int> p;}sta[70001]; int head[70001],cnt; int n,m,t; int f[70001],dep[70001]; pair<int,int> st[70001]; int fa[70001],depth[70001],top[70001],size[70001],son[70001]; vector<pair<int,int> > pos[500000]; int ans[70001]; void add(int x,int y) { e[++cnt].nxt=head[x]; e[cnt].to=y; head[x]=cnt; } int find(int x){return x==f[x]?x:find(f[x]);} void dfs1(int x) { size[x]=1; for(int i=head[x];i;i=e[i].nxt) if(e[i].to!=fa[x]) { fa[e[i].to]=x; depth[e[i].to]=depth[x]+1; dfs1(e[i].to); size[x]+=size[e[i].to]; if(size[e[i].to]>size[son[x]])son[x]=e[i].to; } } void dfs2(int x,int tp) { top[x]=tp; if(!son[x])return; dfs2(son[x],tp); for(int i=head[x];i;i=e[i].nxt) if(!top[e[i].to])dfs2(e[i].to,e[i].to); } int LCA(int x,int y) { while(top[x]!=top[y]) { if(depth[top[x]]<depth[top[y]])swap(x,y); x=fa[top[x]]; } return depth[x]<depth[y]?x:y; } int dis(int x,int y){return depth[x]+depth[y]-(depth[LCA(x,y)]<<1);} void build(int x,int l,int r,int L,int R,int u,int v) { if(r<L||R<l)return; if(L<=l&&r<=R){pos[x].push_back(make_pair(u,v));return;} int mid=(l+r)>>1; build(L(x),l,mid,L,R,u,v); build(R(x),mid+1,r,L,R,u,v); } void merge(int x,int y,int &flag) { x=find(x); y=find(y); int res=0,d; pair<int,int> mzz; if(res<(d=dis(st[x].first,st[x].second)))res=d,mzz=make_pair(st[x].first,st[x].second); if(res<(d=dis(st[x].first,st[y].first)))res=d,mzz=make_pair(st[x].first,st[y].first); if(res<(d=dis(st[x].first,st[y].second)))res=d,mzz=make_pair(st[x].first,st[y].second); if(res<(d=dis(st[x].second,st[y].first)))res=d,mzz=make_pair(st[x].second,st[y].first); if(res<(d=dis(st[x].second,st[y].second)))res=d,mzz=make_pair(st[x].second,st[y].second); if(res<(d=dis(st[y].first,st[y].second)))res=d,mzz=make_pair(st[y].first,st[y].second); flag=max(flag,res); if(dep[x]<dep[y])swap(x,y); sta[++t]=(node){x,y,0,st[x]}; if(dep[x]==dep[y]) { dep[x]++; sta[t].v=1; } f[y]=x; st[x]=mzz; } void del(int x) { while(x<t) { dep[sta[t].x]-=sta[t].v; f[sta[t].y]=sta[t].y; st[sta[t].x]=sta[t].p; t--; } } void solve(int x,int l,int r,int res) { int now=t; for(int i=0;i<pos[x].size();i++) merge(pos[x][i].first,pos[x][i].second,res); if(l==r){ans[l]=res;del(now);return;} int mid=(l+r)>>1; solve(L(x),l,mid,res); solve(R(x),mid+1,r,res); del(now); } int main() { scanf("%d%d",&n,&m); for(int i=1;i<n;i++) { int u,v,l,r; scanf("%d%d%d%d",&u,&v,&l,&r); add(u,v);add(v,u); build(1,1,n,l,r,u,v); } dfs1(1); dfs2(1,1); for(int i=1;i<=n;i++) { f[i]=i; st[i]=make_pair(i,i); } solve(1,1,n,0); while(m--) { int x;scanf("%d",&x); printf("%d\n",ans[x]); } return 0; }
rp++
原文地址:https://www.cnblogs.com/wzc521/p/11623020.html