【每日一题】dfs序专题,求和

求和

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

Solution





#include <bits/stdc++.h>
using namespace std;
#define js ios::sync_with_stdio(false);cin.tie(0); cout.tie(0)
#define all(__vv__) (__vv__).begin(), (__vv__).end()
#define endl "\n"
#define pai pair<int, int>
#define ms(__x__,__val__) memset(__x__, __val__, sizeof(__x__))
typedef long long ll; typedef unsigned long long ull; typedef long double ld;
inline ll read() { ll s = 0, w = 1; char ch = getchar(); for (; !isdigit(ch); ch = getchar()) if (ch == '-') w = -1; for (; isdigit(ch); ch = getchar())    s = (s << 1) + (s << 3) + (ch ^ 48); return s * w; }
inline void print(ll x, int op = 10) { if (!x) { putchar('0'); if (op)    putchar(op); return; }    char F[40]; ll tmp = x > 0 ? x : -x;    if (x < 0)putchar('-');    int cnt = 0;    while (tmp > 0) { F[cnt++] = tmp % 10 + '0';        tmp /= 10; }    while (cnt > 0)putchar(F[--cnt]);    if (op)    putchar(op); }
inline ll gcd(ll x, ll y) { return y ? gcd(y, x % y) : x; }
ll qpow(ll a, ll b) { ll ans = 1;    while (b) { if (b & 1)    ans *= a;        b >>= 1;        a *= a; }    return ans; }    ll qpow(ll a, ll b, ll mod) { ll ans = 1; while (b) { if (b & 1)(ans *= a) %= mod; b >>= 1; (a *= a) %= mod; }return ans % mod; }
const int dir[][2] = { {0,1},{1,0},{0,-1},{-1,0},{1,1},{1,-1},{-1,1},{-1,-1} };
const int MOD = 1e9 + 7;
const int INF = 0x3f3f3f3f;
const int N = 1e6 + 7;

vector<int> edge[N];
int tot, val[N], l[N], r[N];
ll sum[N];

void dfs(int u, int tmp) {
    l[u] = ++tot;
    for (auto& v : edge[u]) {
        if (v == tmp)    continue;
        dfs(v, u);
    }
    r[u] = tot;
}

int lowbit(int x) { return x & (-x); }
void update(int x, int y) {
    for (; x < N; x += lowbit(x))
        sum[x] += y;
}
ll query(int x) {
    ll res = 0;
    for (; x; x -= lowbit(x))
        res += sum[x];
    return res;
}

int main() {
    int T = 1;
    //T = read();
    while (T--) {
        int n = read(), m = read(), s = read();
        for (int i = 1; i <= n; ++i)    val[i] = read();
        for (int i = 1; i < n; ++i) {
            int u = read(), v = read();
            edge[u].push_back(v);
            edge[v].push_back(u);
        }
        dfs(s, 0);
        for (int i = 1; i <= n; ++i)
            update(l[i], val[i]);
        while (m--) {
            int op = read();
            if (op & 1) {
                int u = read(), x = read();
                update(l[u], x); //单点修改
            }
            else {
                int u = read();
                print(query(r[u]) - query(l[u] - 1)); // 区间查询
            }
        }
    }
    return 0;
}
每日一题 文章被收录于专栏

每日一题

全部评论

相关推荐

鬼迹人途:你去投一投尚游游戏,服务器一面,第一个图算法,做完了给你一个策略题,你给出方案他就提出低概率问题,答不上当场给你挂
点赞 评论 收藏
分享
05-19 19:57
蚌埠学院 Python
2237:Gpa70不算高,建议只写排名,个人技能不在多而在精,缩到8条以内。项目留一个含金量高的,减少间距弄到一页,硕士简历也就一页,本科不要写很多
实习,投递多份简历没人回...
点赞 评论 收藏
分享
码农索隆:单休一个月少休息4天,一年就是48天,平时节假日,别人3天假期,单休的两天
点赞 评论 收藏
分享
评论
3
收藏
分享

创作者周榜

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