题目描述
给出n个结点以及每个点初始时对应的权值wi。起始时点与点之间没有连边。有3类操作:
1、bridge A B:询问结点A与结点B是否连通。
如果是则输出“no”。否则输出“yes”,并且在结点A和结点B之间连一条无向边。
2、penguins A X:将结点A对应的权值wA修改为X。
3、excursion A B:如果结点A和结点B不连通,则输出“impossible”。
否则输出结点A到结点B的路径上的点对应的权值的和。
给出q个操作,要求在线处理所有操作。
数据范围:1<=n<=30000, 1<=q<=300000, 0<=wi<=1000。
输入
第一行包含一个整数n(1<=n<=30000),表示节点的数目。
第二行包含n个整数,第i个整数表示第i个节点初始时对应的权值。
第三行包含一个整数q(1<=n<=300000),表示操作的数目。
以下q行,每行包含一个操作,操作的类别见题目描述。
任意时刻每个节点对应的权值都是1到1000的整数。
输出
输出所有bridge操作和excursion操作对应的输出,每个一行。
样例输入
5
4 2 4 5 6
10
excursion 1 1
excursion 1 2
bridge 1 2
excursion 1 2
bridge 3 4
bridge 3 5
excursion 4 5
bridge 1 3
excursion 2 4
excursion 2 5
4 2 4 5 6
10
excursion 1 1
excursion 1 2
bridge 1 2
excursion 1 2
bridge 3 4
bridge 3 5
excursion 4 5
bridge 1 3
excursion 2 4
excursion 2 5
样例输出
4
impossible
yes
6
yes
yes
15
yes
15
16
impossible
yes
6
yes
yes
15
yes
15
16
LCT模板题,splay维护子树和,注意操作1连通输出no,不连通输出yes。
#include<set> #include<map> #include<cmath> #include<queue> #include<stack> #include<vector> #include<cstdio> #include<bitset> #include<cstring> #include<iostream> #include<algorithm> using namespace std; int n,m; int x,y; char ch[10]; int s[30010][2]; int f[30010]; int r[30010]; int st[30010]; int v[30010]; int sum[30010]; int get(int rt) { return rt==s[f[rt]][1]; } void pushup(int rt) { sum[rt]=sum[s[rt][0]]+sum[s[rt][1]]+v[rt]; } void pushdown(int rt) { if(r[rt]) { r[s[rt][0]]^=1; r[s[rt][1]]^=1; r[rt]^=1; swap(s[rt][0],s[rt][1]); } } int is_root(int rt) { return rt!=s[f[rt]][0]&&rt!=s[f[rt]][1]; } void rotate(int rt) { int fa=f[rt]; int anc=f[fa]; int k=get(rt); if(!is_root(fa)) { s[anc][get(fa)]=rt; } s[fa][k]=s[rt][k^1]; f[s[fa][k]]=fa; s[rt][k^1]=fa; f[fa]=rt; f[rt]=anc; pushup(fa); pushup(rt); } void splay(int rt) { int top=0; st[++top]=rt; for(int i=rt;!is_root(i);i=f[i]) { st[++top]=f[i]; } for(int i=top;i>=1;i--) { pushdown(st[i]); } for(int fa;!is_root(rt);rotate(rt)) { if(!is_root(fa=f[rt])) { rotate(get(fa)==get(rt)?fa:rt); } } } void access(int rt) { for(int x=0;rt;x=rt,rt=f[rt]) { splay(rt); s[rt][1]=x; pushup(rt); } } void reverse(int rt) { access(rt); splay(rt); r[rt]^=1; } int find(int rt) { access(rt); splay(rt); while(s[rt][0]) { rt=s[rt][0]; } return rt; } void link(int x,int y) { reverse(x); f[x]=y; } void change(int rt,int x) { v[rt]=x; access(rt); splay(rt); } int query(int x,int y) { reverse(x); access(y); splay(y); return sum[y]; } int main() { scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%d",&v[i]); } scanf("%d",&m); while(m--) { scanf("%s",ch); scanf("%d%d",&x,&y); if(ch[0]=='e') { if(find(x)!=find(y)) { printf("impossible "); } else { printf("%d ",query(x,y)); } } else if(ch[0]=='b') { if(find(x)!=find(y)) { printf("yes "); link(x,y); } else { printf("no "); } } else { change(x,y); } } }