求和

求和

https://ac.nowcoder.com/acm/problem/204871

分析

由于我们每次是查询一个子树内的所有节点的权值和
很套路的想法就是将他映射到线性区间上
那么就想到了用dfn序进行维护
由于我们每次有单点修改
那么我们可以使用码量比较小的树状数组直接维护
支持单调修改 + 区间查询
时间复杂度:
期望得分:100分

代码

//Nowcoder 204871
#include <algorithm>
#include <iostream>
#include <cstring>
#include <vector>
#include <cstdio>
#include <cmath>

#define ll long long
#define cl(x, y) memset((x), (y), sizeof(x))
#define rep(i, a, b) for(int i = a; i <= b; i++)
#define per(i, a, b) for(int i = a; i >= b; i--)
#define de(x) cerr << #x << " = " << x << " "
#define inc_mod(x, y) x = ((x - y) % mod + mod) % mod
#define add_mod(x, y) x = (x + y) % mod
#define lowbit(x) (x & (-x))
#define inf 0x3f3f3f3f
#define mod 998244353
#define rson (x << 1 | 1)
#define lson (x << 1)
using namespace std;
const int maxn = 1e6 + 10;

int n, k, m;
int dfn[maxn], dfn_num, R[maxn];
vector<int> G[maxn];
int val[maxn];
ll fro[maxn];

inline void dfs(int u, int fa) {
    dfn[u] = ++dfn_num;
    for(auto v : G[u]) if(v != fa) {
        dfs(v, u);
    }
    R[u] = dfn_num;
}

inline void update(int u, int a) { while(u <= n) fro[u] += a, u += lowbit(u); }
inline ll get(int u) { ll ret = 0; while(u > 0) ret += fro[u], u -= lowbit(u); return ret; }

signed main() {
    cin >> n >> m >> k;
    rep(i, 1, n) scanf("%d", &val[i]);
    int u, v, w;
    rep(i, 2, n) scanf("%d %d", &u, &v) , G[u].push_back(v), G[v].push_back(u);
    dfs(k, k);
    rep(i, 1, n) update(dfn[i], val[i]);
    while(m--) {
        scanf("%d %d", &u, &v);
        if(u == 1) {
            scanf("%d", &w);
            update(dfn[v], w);
        } else printf("%lld\n", get(R[v]) - get(dfn[v] - 1));
    }
    system("pause");
    return 0;
}
全部评论

相关推荐

06-13 15:45
辽宁大学 golang
咱就是说&nbsp;你不主动&nbsp;我也不会主动下一步hhh,急死了
恶龙战士:不建议把这种帖子发到牛客上,建议去小红书发
点赞 评论 收藏
分享
06-13 17:33
门头沟学院 Java
顺序不记了,大致顺序是这样的,有的相同知识点写分开了1.基本数据类型2.基本数据类型和包装类型的区别3.==和equals区别4.ArrayList与LinkedList区别5.hashmap底层原理,put操作时会发生什么6.说出几种树型数据结构7.B树和B+树区别8.jvm加载类机制9.线程池核心参数10.创建线程池的几种方式11.callable与runnable区别12.线程池怎么回收线程13.redis三剑客14.布隆过滤器原理,不要背八股,说说真正使用时遇到了问题没有(我说没有,不知道该怎么回答了)15.堆的内存结构16.自己在写项目时有没有遇见过oom,如何处理,不要背八股,根据真实经验,我说不会17.redis死锁怎么办,watchdog机制如何发现是否锁过期18.如何避免redis红锁19.一个表性别与年龄如何加索引20.自己的项目的QPS怎么测的,有没有真正遇到大数量表21.说一说泛型22.springboot自动装配原理23.springmvc与springboot区别24.aop使用过嘛?动态代理与静态代理区别25.spring循环依赖怎么解决26.你说用过es,es如何分片,怎么存的数据,1000万条数据怎么写入库中27.你说用limit,那么在数据量大之后,如何优化28.rabbitmq如何批次发送,批量读取,答了延迟队列和线程池,都不对29.计网知不知道smtp协议,不知道写了对不对,完全听懵了30.springcloud知道嘛?只是了解反问1.做什么的?短信服务,信息量能到千万级2.对我的建议,基础不错,但是不要只背八股,多去实际开发中理解。面试官人不错,虽然没露脸,但是中间会引导我回答问题,不会的也只是说对我要求没那么高。面完问我在济宁生活有没有困难,最快什么时候到,让人事给我聊薪资了。下午人事打电话,问我27届的会不会跑路,还在想办法如何使我不跑路,不想扣我薪资等。之后我再联系吧,还挺想去的😭,我真不跑路哥😢附一张河科大幽默大专图,科大就是大专罢了
查看30道真题和解析
点赞 评论 收藏
分享
评论
2
2
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务