第五届“传智杯”练习题--树的变迁

最新推荐文章于 2023-04-09 19:18:37 发布

南安@

最新推荐文章于 2023-04-09 19:18:37 发布

阅读量422

点赞数 1

文章标签： c语言算法 c++

本文链接：https://blog.csdn.net/weixin_63984309/article/details/127938986

版权

题目描述

给定一棵具有 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复制

输出 #1复制

说明/提示

所有测试数据数据满足 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;
}