【比较函数坑点】D. Li Hua and Tree

题目

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define fi first
#define se second
#define lson p << 1
#define rson p << 1 | 1
const int maxn = 1e5 + 5, inf = 1e18, maxm = 4e4 + 5;
const int mod = 1e9 + 7;
// const int mod = 998244353;
const int N = 1e6;
// int a[505][5005];
// bool vis[505][505];
// char s[505][505];
int a[maxn], b[maxn];
bool vis[maxn];
string s;
int n, m;

int siz[maxn];
struct Node{
    // int val, id;
    // bool operator<(const Node &u)const{
    //     return val < u.val;
    // }
    int siz, id;
    bool operator<(const Node &u)const{
        // if(siz[id] != siz[u.id])
        //     return siz[id] > siz[u.id];
        // return id < u.id;不能用利用全局数组比较,因为siz数组也在更新,可能出现set排序不及时的情况
        if(siz != u.siz){
            return siz > u.siz;
        }
        return id < u.id;
    }
}c[maxn];

int ans[maxn];
// priority_queue<Node> G[maxn];
set<Node> G[maxn];
int f[maxn], fa[maxn];
vector<int> g[maxn];

void dfs(int u, int p){
    fa[u] = p;
    f[u] = a[u];
    siz[u] = 1;
    for(auto v : g[u]){
        // int v = z.id;
        if(v == p) continue;
        dfs(v, u);
        G[u].insert({siz[v], v});
        // G[u].insert({v});
        siz[u] += siz[v];
        f[u] += f[v];
    }
}

void solve(){
    int res = 0;
    int q, k;
    cin >> n >> m;
    for(int i = 1; i <= n; i++){
        cin >> a[i];
    }
    for(int i = 1; i < n; i++){
        int u, v;
        cin >> u >> v;
        g[u].pb(v);
        g[v].pb(u);
    }
    dfs(1, 1);
    for(int tt = 1; tt <= m; tt++){
        int op, x;
        cin >> op >> x;
        if(op == 1){
            cout << f[x] << '\n';
        }
        else{
            if(G[x].size() == 0) continue;
            int par = fa[x];
            int son = G[x].begin() -> id;
            int tmp;
            G[x].erase(Node{siz[son], son});
            G[par].erase(Node{siz[x], x});
            tmp = siz[x];
            siz[x] -= siz[son];
            siz[son] += tmp - siz[son];
            G[son].insert(Node{siz[x], x});
            G[par].insert(Node{siz[son], son});
            tmp = f[x];
            f[x] -= f[son];
            f[son] += tmp - f[son];
            fa[son] = par;
            fa[x] = son;
        }  
    }
}
    
signed main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    int T = 1;
    // cin >> T;
    while (T--)
    {
        solve();
    }
    
    return 0;
}
相关推荐
小码农<^_^>14 分钟前
优选算法精品课--滑动窗口算法(一)
算法
羊小猪~~16 分钟前
神经网络基础--什么是正向传播??什么是方向传播??
人工智能·pytorch·python·深度学习·神经网络·算法·机器学习
软工菜鸡41 分钟前
预训练语言模型BERT——PaddleNLP中的预训练模型
大数据·人工智能·深度学习·算法·语言模型·自然语言处理·bert
南宫生44 分钟前
贪心算法习题其三【力扣】【算法学习day.20】
java·数据结构·学习·算法·leetcode·贪心算法
AI视觉网奇1 小时前
sklearn 安装使用笔记
人工智能·算法·sklearn
JingHongB1 小时前
代码随想录算法训练营Day55 | 图论理论基础、深度优先搜索理论基础、卡玛网 98.所有可达路径、797. 所有可能的路径、广度优先搜索理论基础
算法·深度优先·图论
weixin_432702262 小时前
代码随想录算法训练营第五十五天|图论理论基础
数据结构·python·算法·深度优先·图论
小冉在学习2 小时前
day52 图论章节刷题Part04(110.字符串接龙、105.有向图的完全可达性、106.岛屿的周长 )
算法·深度优先·图论
Repeat7152 小时前
图论基础--孤岛系列
算法·深度优先·广度优先·图论基础
小冉在学习2 小时前
day53 图论章节刷题Part05(并查集理论基础、寻找存在的路径)
java·算法·图论