Description
有\(N\)个节点,标号从\(1\)到\(N\),这\(N\)个节点一开始相互不连通。第$ i\(个节点的初始权值为\)a_i$ ,接下来有如下一些操作:
U x y
加一条边,连接第 \(x\) 个节点和第\(y\) 个节点。
A1 x v
将第 \(x\) 个节点的权值增加 \(v\)。
A2 x v
将第 \(x\) 个节点所在的连通块的所有节点的权值都增加 \(v\)。
A3 v
将所有节点的权值都增加\(v\) 。
F1 x
输出第 \(x\) 个节点当前的权值。
F2 x
输出第 \(x\) 个节点所在的连通块中,权值最大的节点的权值。
F3
输出所有节点中,权值最大的节点的权值。
Solution
离线处理
对原序列进行重新排序,使得每次合并时,两个集合的存在区间恰好相邻
转化为简单的线段树区间修改+区间询问
Code
#include#define ll long long#define max(a,b) ((a)>(b)?(a):(b))#define min(a,b) ((a)>(b)?(b):(a))#define pi std::pair #define reg registerusing namespace std;inline int read(){ int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();} return x*f;}const int MN=6e5+5;int A[MN],Map[MN],fMap[MN];struct Mapper{ int fa[MN],L[MN],R[MN],suf[MN]; void init(int N) { reg int i; for(i=1;i<=N;++i) fa[i]=L[i]=R[i]=i,suf[i]=-1; } int getf(int x){return fa[x]==x?x:fa[x]=getf(fa[x]);} void insert(int x,int y) { x=getf(x);y=getf(y); if(x==y) return; suf[R[x]]=L[y]; L[y]=L[x];fa[x]=y; } bool vis[MN]; void getMap(int N) { memset(vis,0,sizeof vis); reg int i,l,r,cnt=0; for(i=1;i<=N;++i)if(!vis[i]) for(l=L[getf(i)];l>0;l=suf[l]) vis[fMap[++cnt]=l]=true; for(i=1;i<=N;++i) Map[fMap[i]]=i; }}a;struct Operation{int opt,x,y;}O[MN];int readchar(){ static char s[5]; scanf("%s",s+1); if(s[1]=='U') return 1; if(s[1]=='A') return 1+s[2]-'0'; if(s[1]=='F') return 4+s[2]-'0'; }struct Union_Find{ int fa[MN],L[MN],R[MN]; void init(int N) { reg int i; for(i=1;i<=N;++i) fa[i]=i,L[i]=R[i]=Map[i]; } int getf(int x) { return fa[x]==x?x:fa[x]=getf(fa[x]); } void combine(int x,int y) { x=getf(x);y=getf(y);if(x==y) return; fa[x]=y;L[y]=min(L[y],L[x]);R[y]=max(R[x],R[y]); } pi get(int x) { x=getf(x); return make_pair(L[x],R[x]); }}b;struct SegTree{ #define ls x<<1 #define rs x<<1|1 #define mid ((l+r)>>1) int t[MN<<2],lazy[MN<<1]; void up(int x){t[x]=max(t[ls],t[rs]);} void Build(int x,int l,int r) { if(l==r) return (void)(t[x]=A[fMap[l]]); Build(ls,l,mid);Build(rs,mid+1,r);up(x); } void C(int x,int val){t[x]+=val,lazy[x]+=val;} void down(int x){if(lazy[x])C(ls,lazy[x]),C(rs,lazy[x]),lazy[x]=0;} void Modify(int x,int l,int r,int a,int b,int val) { if(l==a&&r==b) return (void)(C(x,val));down(x); if(b<=mid) Modify(ls,l,mid,a,b,val); else if(a>mid) Modify(rs,mid+1,r,a,b,val); else Modify(ls,l,mid,a,mid,val),Modify(rs,mid+1,r,mid+1,b,val); up(x); } int Query(int x,int l,int r,int a,int b) { if(l==a&&r==b) return t[x];down(x); if(b<=mid) return Query(ls,l,mid,a,b); else if(a>mid) return Query(rs,mid+1,r,a,b); else return max(Query(ls,l,mid,a,mid),Query(rs,mid+1,r,mid+1,b)); }}c;int main(){ reg int i,N=read(); a.init(N); for(i=1;i<=N;++i) A[i]=read(); reg int M=read(); for(i=1;i<=M;++i) { O[i].opt=readchar(); if(O[i].opt<7) O[i].x=read(); if(O[i].opt<4) O[i].y=read(); if(O[i].opt==1) a.insert(O[i].x,O[i].y); } a.getMap(N);c.Build(1,1,N);b.init(N);pi xxx; for(i=1;i<=M;++i) { if(O[i].opt==1) b.combine(O[i].x,O[i].y); if(O[i].opt==2) c.Modify(1,1,N,Map[O[i].x],Map[O[i].x],O[i].y); if(O[i].opt==3) xxx=b.get(O[i].x),c.Modify(1,1,N,xxx.first,xxx.second,O[i].y); if(O[i].opt==4) c.Modify(1,1,N,1,N,O[i].x); if(O[i].opt==5) printf("%d\n",c.Query(1,1,N,Map[O[i].x],Map[O[i].x])); if(O[i].opt==6) xxx=b.get(O[i].x),printf("%d\n",c.Query(1,1,N,xxx.first,xxx.second)); if(O[i].opt==7) printf("%d\n",c.Query(1,1,N,1,N)); } return 0;}
Blog来自PaperCloud,未经允许,请勿转载,TKS!