题目描述
给定一棵具有 nn 个节点的树,每个节点有一个初始权值 a_iai。
一共需要进行 mm 次操作,每次操作包括:
-
1 e
编号为 ee 的边突然消失,使得它所在的那棵树变成了两棵树。 -
2 u val
编号为 uu 的节点的权值变成了 valval。 -
3 u
进行了一次查询,查询 uu 所在的那棵树的权值之和。
现在你需要来模拟上述事件,以了解树的变迁。
输入格式
第一行为 n, mn,m,如上所述。
第二行有 nn 个数,为 nn 个结点的初始权值,在 10^3103 以内。
下面 n-1n−1 行,每行一组 u, vu,v,表示一条边。(保证初始为一棵树)
下面 mm 行有 mm 个操作:
先读入一个 \text{opt}opt,表示操作类型。
-
\text{opt}=1opt=1 时,读入 ee,表示删掉读入的第 ee 条边。(保证第 ee 条边存在)
-
\text{opt}=2opt=2 时,读入 u,valu,val,表示把结点 uu 的权值改成 valval(val \le 1000val≤1000)。
-
\text{opt}=3opt=3 时,读入 uu,表示查询 uu 所在的那棵树的结点权值和。
输出格式
对于每个查询操作,输出一行一个数表示答案。
输入输出样例
输入 #1复制
2 3 1 1 1 2 2 2 4 1 1 3 2
输出 #1复制
4
说明/提示
所有测试数据数据满足 1 \leq n,m \leq {10}^51≤n,m≤105,1 \leq a_i,val \leq 10001≤ai,val≤1000。
代码如下(示例):
#include<bits/stdc++.h>
using namespace std;
struct node{
int u,v;
}e[100005];
int num[100005];
int sum[100005];
int ans[100005];
int vis[100005];
vector<int> val[100005];
int f[100005];
struct cz{
int opt;
int u;
}cc[100005];
int opt,cnt;
int find(int x)
{
return x==f[x]?x:f[x] = find(f[x]);
}
void merge(int x,int y)
{
int xx = find(x),yy = find(y);
if(xx!=yy)
{
f[xx] = yy;
sum[yy]+=sum[xx];
sum[xx] = 0;
}
}
int n,m;
int main()
{
cin>>n>>m;
for(int i = 1;i<=n;++i)
cin>>num[i];
for(int i = 1;i<n;++i)
cin>>e[i].u>>e[i].v;
for(int i = 1;i<=m;++i)
{
cin>>cc[i].opt;
if(cc[i].opt==1)
{
cin>>cc[i].u;
vis[cc[i].u] = 1;
}
if(cc[i].opt==2)
{
cin>>cc[i].u;
int v;
cin>>v;
val[cc[i].u].push_back(num[cc[i].u]);
num[cc[i].u] = v;
}
if(cc[i].opt==3)
cin>>cc[i].u;
}
for(int i = 1;i<=n;++i)
sum[i] = num[i],f[i] = i;
for(int i = 1;i<n;++i)
{
if(!vis[i])
merge(e[i].u,e[i].v);
}
for(int i = m;i>0;--i)
{
if(cc[i].opt==1)
merge(e[cc[i].u].u,e[cc[i].u].v);
if(cc[i].opt==2)
{
int v = cc[i].u;
int now = val[v][val[v].size()-1];
val[v].pop_back();
sum[find(v)]+=now-num[v];
num[v] = now;
}
if(cc[i].opt==3)
{
ans[++cnt] = sum[find(cc[i].u)];
}
}
for(int i = cnt;i>0;--i)
cout<<ans[i]<<endl;
return 0;
}