题目
有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: 输出所有节点中,权值最大的节点的权值
输入格式
输入的第一行是一个整数N,代表节点个数。
接下来一行输入N个整数,a[1], a[2], …, a[N],代表N个节点的初始权值。
再下一行输入一个整数Q,代表接下来的操作数。
最后输入Q行,每行的格式如题目描述所示。
输出格式
对于操作F1, F2, F3,输出对应的结果,每个结果占一行。
输入样例
3
0 0 0
8
A1 3 -20
A1 2 20
U 1 3
A2 1 10
F1 3
F2 3
A3 -10
F3
输出样例
-10
10
10
提示
对于30%的数据,保证 N<=100,Q<=10000
对于80%的数据,保证 N<=100000,Q<=100000
对于100%的数据,保证 N<=300000,Q<=300000
对于所有的数据,保证输入合法,并且 -1000<=v, a[1], a[2], …, a[N]<=1000
题解
据说此题很多人堆套堆,怎么这么难写我那么弱当然是用线段树啦
我觉得线段树的确好写到不知哪里去
对于所有操作,似乎在线段树上都很好实现,唯一的难点就在于点的编号
那么问题就转化成了,给定一种编号方法,使任意时刻同一个联通块内的所有点编号连续
只需要分两种情况想就很容易实现了:
我们想象,一开始所有点相互独立,没什么关系
①当两个独立的点相连时,它们的编号一定是连续的,否则此时就不满足所需性质
那我们就先用链表将它们连起来,表示编号连续
②当两个联通块相连时,由我们维护的性质得:两个联通块内部的点编号一定是连续的,现在我们需要两个联通块编号连续,我们只需要将它们的编号衔接起来就好了,那么我们把其中一个联通块所对应的链 接到另一个联通块对应的链末尾就好了
可以发现,这样子操作之后,我们就会得出若干个链,表示链上的点编号必须按链上的顺序
所以我们按链的顺序标号,就能保证所有时刻联通块内部点的编号连续
取链头链尾用并查集实现
我们在询问的时候,也要用上并查集,并且链接顺序与标号的时候相同,保证每个联通块目前的代表元一定是标号时编号最小的点,所以我们再维护并查集的大小就可以轻松求出每次操作的区间啦~
数据结构部分就只用实现一个简单的线段树比堆套堆不知道要好写到哪里去
丑丑的代码
#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long int
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
#define BUG(s,n) for (int i = 1; i <= (n); i++) cout<<s[i]<<' '; puts("");
#define ls (u << 1)
#define rs (u << 1 | 1)
using namespace std;
const int maxn = 300005,maxm = 100005,INF = 1000000000;
inline int read(){int out = 0,flag = 1; char c = getchar();while (c < 48 || c > 57) {if (c == '-') flag = -1; c = getchar();}while (c >= 48 && c <= 57) {out = (out << 3) + (out << 1) + c - '0'; c = getchar();}return out * flag;
}
struct Query{int opt,a,b,c;}q[maxn];
int n,m,pre[maxn],post[maxn],id[maxn],Hash[maxn],val[maxn],cnt;
int nxt[maxn],siz[maxn];
int mx[4 * maxn],tag[4 * maxn];
char opt[10];
int findu(int u){return u == pre[u] ? u : pre[u] = findu(pre[u]);}
int findd(int u){return u == post[u] ? u : post[u] = findd(post[u]);}
void build(int u,int l,int r){if (l == r) {mx[u] = val[Hash[l]]; return;}int mid = l + r >> 1;build(ls,l,mid);build(rs,mid + 1,r);mx[u] = max(mx[ls],mx[rs]);
}
void pd(int u){if (tag[u]){mx[ls] += tag[u]; tag[ls] += tag[u];mx[rs] += tag[u]; tag[rs] += tag[u];tag[u] = 0;}
}
void modify(int u,int l,int r,int L,int R,int v){if (l >= L && r <= R){mx[u] += v; tag[u] += v; return;}pd(u);int mid = l + r >> 1;if (mid >= L) modify(ls,l,mid,L,R,v);if (mid < R) modify(rs,mid + 1,r,L,R,v);mx[u] = max(mx[ls],mx[rs]);
}
int query(int u,int l,int r,int L,int R){if (l >= L && r <= R) return mx[u];pd(u);int mid = l + r >> 1;if (mid >= R) return query(ls,l,mid,L,R);else if (mid < L) return query(rs,mid + 1,r,L,R);else return max(query(ls,l,mid,L,R),query(rs,mid + 1,r,L,R));
}
int main(){n = read();for (int i = 1; i <= n; i++) val[i] = read(),pre[i] = post[i] = i;m = read();int fa,fb,sa,sb;for (int i = 1; i <= m; i++){scanf("%s",opt);if (opt[0] == 'U'){q[i].opt = 0,q[i].a = read(),q[i].b = read();fa = findu(q[i].a); fb = findu(q[i].b);if (fa == fb) continue;sa = findd(q[i].a); sb = findd(q[i].b);nxt[sa] = fb;pre[fb] = fa;post[sa] = sb;}else if (opt[0] == 'A'){q[i].a = read();if (opt[1] == '1') q[i].opt = 1,q[i].b = read();else if (opt[1] == '2') q[i].opt = 2,q[i].b = read();else q[i].opt = 3;}else {if (opt[1] == '1') q[i].opt = 4,q[i].a = read();else if (opt[1] == '2') q[i].opt = 5,q[i].a = read();else q[i].opt = 6;}}for (int i = 1; i <= n; i++){if (id[i]) continue;int u = findu(i);while (u) id[u] = ++cnt,Hash[cnt] = u,u = nxt[u];}build(1,1,n);for (int i = 1; i <= n; i++) pre[i] = i,siz[i] = 1;for (int i = 1; i <= m; i++){switch(q[i].opt){case 0:fa = findu(q[i].a); fb = findu(q[i].b);if (fa != fb){siz[fa] += siz[fb];pre[fb] = fa;}break;case 1:modify(1,1,n,id[q[i].a],id[q[i].a],q[i].b);break;case 2:fa = findu(q[i].a);modify(1,1,n,id[fa],id[fa] + siz[fa] - 1,q[i].b);break;case 3:modify(1,1,n,1,n,q[i].a);break;case 4:printf("%d\n",query(1,1,n,id[q[i].a],id[q[i].a]));break;case 5:fa = findu(q[i].a);printf("%d\n",query(1,1,n,id[fa],id[fa] + siz[fa] - 1));break;case 6:printf("%d\n",mx[1]);break;}}return 0;
}