hdu 3966 Aragorn's Story

来源：互联网发布：中世纪2优化9拜占庭编辑：程序博客网时间：2024/05/16 06:58

树链剖分的入门题。

题目大意：n个结点的树，每个结点有一个值。两种操作：1、将结点a-b路径上的结点值都加上或减去一个值。 2、询问结点a的值。

建树之后，将树状数组初始化为0，假设每个结点的值也都为0。然后进行更新。

设结点i建树之后的序号为id[i]，初始值为A[i]，则在id[i]位置增加A[i]，在id[i]+1的位置减少A[i]，这样在询问结点的值时，只需用树状数组的求一下前id[i]项和即为序号为id[i]的结点的值。

对于更新操作，对于两个结点u、v。找到其所在链的顶端结点top[u]，top[v]，比较其深度，若dep[top[u]]>dep[top[v]，则将结点序号为id[top[u]]到id[u]的值更新，再令u=fa[top[u]。直到top[u]=top[v]，此时若u!=v，则再更新u-v路径上结点的值。

注意：为防止爆栈，得加上这句#pragma comment(linker,"/STACK:102400000,102400000")

#include <cstdio>#include <cstring>#include <vector>#include <algorithm>#pragma comment(linker,"/STACK:102400000,102400000")using namespace std;#define maxn 50005int A[maxn],dep[maxn], fa[maxn], son[maxn], num[maxn], top[maxn], id[maxn],  C[maxn],tot;vector<int> g[maxn];int lowbit(int x){return x&(-x);}void dfs1(int u){    num[u] = 1,son[u] = 0;    for (int i = 0; i < g[u].size(); i++)        if (g[u][i] != fa[u])        {            fa[g[u][i]]=u;            dep[g[u][i]]=dep[u]+1;            dfs1(g[u][i]);            num[u] += num[g[u][i]];            if (num[son[u]] < num[g[u][i]])            son[u] = g[u][i];        }}void dfs2(int u, int tp){    id[u] = ++tot;    top[u] = tp;    if (son[u]) dfs2(son[u], tp);    for (int i = 0; i < g[u].size(); i++)    {        int v = g[u][i];        if (v == fa[u] || v == son[u]) continue;        dfs2(v, v);    }}void add(int x, int v){    while (x < maxn)    {        C[x] += v;        x += lowbit(x);    }}void add(int l, int r, int v){    add(l, v);    add(r + 1, -v);}void update(int u, int v, int w){    int f1 = top[u], f2 = top[v];    while (f1 != f2)    {        if (dep[f1] < dep[f2])        {            swap(f1, f2);            swap(u, v);        }        add(id[f1], id[u], w);        u = fa[f1];        f1 = top[u];    }    if (dep[u] > dep[v]) swap(u, v);    add(id[u], id[v], w);}int query(int x){    int ans = 0;    while (x)    {        ans += C[x];        x -= lowbit(x);    }    return ans;}int main(){    int n,m,p;    while (~scanf("%d%d%d", &n, &m, &p))    {        memset(C, 0, sizeof(C));        memset(num,0,sizeof(num));        dep[1]=fa[1]=tot=0;        for (int i = 1; i <= n; i++)        {            scanf("%d", &A[i]);            g[i].clear();        }        int u, v;        while (m--)        {            scanf("%d%d", &u, &v);            g[u].push_back(v);            g[v].push_back(u);        }        dfs1(1);        dfs2(1, 1);        for (int i = 1; i <= n; i++) add(id[i], id[i], A[i]);        char q[2];        int a, b, c;        while (p--)        {            scanf("%s", q);            if (q[0] == 'I' || q[0] == 'D')            {                scanf("%d%d%d", &a, &b, &c);                if (q[0] == 'D') c = -c;                update(a, b, c);            }            else            {                scanf("%d", &a);                printf("%d\n", query(id[a]));            }        }    }    return 0;}

0 0