1036: [ZJOI2008]树的统计Count

1036: [ZJOI2008]树的统计Count

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 13969  Solved: 5644
[Submit][Status][Discuss]

Description

　　一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成
一些操作： I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I
II. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意：从点u到点v的路径上的节点包括u和v本身

Input

　　输入的第一行为一个整数n，表示节点的个数。接下来n – 1行，每行2个整数a和b，表示节点a和节点b之间有
一条边相连。接下来n行，每行一个整数，第i行的整数wi表示节点i的权值。接下来1行，为一个整数q，表示操作
的总数。接下来q行，每行一个操作，以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。 
对于100％的数据，保证1<=n<=30000，0<=q<=200000；中途操作中保证每个节点的权值w在-30000到30000之间。

Output

　　对于每个“QMAX”或者“QSUM”的操作，每行输出一个整数表示要求输出的结果。

Sample Input

4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4

Sample Output

4
1
2
2
10
6
5
6
5
16

HINT

题目大意：

如题。

思路：

今天下午看到题目，感觉挺简单的，不就是个树链剖分吗？！噼里啪啦一阵子敲完。

WA 。。。

难道是我线段树写的有问题吗。。

换个姿势重新写一发。。。

wa。。。

我把成段更新换成单点更新。

wa。。。

又把懒惰标记 去掉，

wa。。。

0.0   我有点方。。。

重新写了一发 查询那里。。

继续 wa 。。

后来发现权值有负数。。。。  MDZZ

其实树链剖分之后  在查询和求和的时候都是 在重链上进行的。剖分之后就是线段树的事情了。     

一开始我还在想，不可能是我线段树写错了吧。亏我还写了三遍。。。居然是题意没看清楚。。。

AC代码：

#include <cstdio>#include <cstring>#include <vector>#include <algorithm>#include <iostream>using namespace std;#define Del(a,b) memset(a,b,sizeof(a))const int N = 30010;int n,m,q;int dep[N],siz[N],fa[N],id[N],son[N],val[N],top[N]; //top 最近的重链父节点int rankk[N];int num;int coss[N];vector<int> v[N];struct tree{    int x,y,val;    void read()    {        scanf("%d%d",&x,&y);    }};tree e[N];void dfs1(int u, int f, int d)///当前节点  father  dep   一开始  u = root{    dep[u] = d;    siz[u] = 1;    son[u] = 0;    fa[u] = f;    for (int i = 0; i < v[u].size(); i++)    {        int ff = v[u][i];        if (ff == f)            continue;        dfs1(ff, u, d + 1);        siz[u] += siz[ff];        if (siz[son[u]] < siz[ff])            son[u] = ff;    }}void dfs2(int u, int tp){    top[u] = tp;    id[u] = ++num;    rankk[id[u]]=u;    if (son[u])        dfs2(son[u], tp);    for (int i = 0; i < v[u].size(); i++)    {        int ff = v[u][i];        if (ff == fa[u] || ff == son[u])            continue;        dfs2(ff, ff);    }}///线段树部分！#define lson l , m , rt << 1#define rson m + 1 , r , rt << 1 | 1struct Tree{    int l,r,sum,maxx;};Tree tree[4*N];void PushUp(int rt){    tree[rt].sum =tree[rt<<1].sum + tree[rt<<1|1].sum;    tree[rt].maxx = max(tree[rt<<1].maxx, tree[rt<<1|1].maxx);}void build(int l,int r,int rt){    tree[rt].l=l;    tree[rt].r=r;    if (l == r)    {        tree[rt].sum =tree[rt].maxx= coss[rankk[l]];        return ;    }    int m = (l + r) >> 1;    build(lson);    build(rson);    PushUp(rt);}void update(int p,int sc,int l,int r,int rt){    if (l == r)    {        tree[rt].maxx = sc;        tree[rt].sum=sc;        return ;    }    int m = (l + r) >> 1;    if (p <= m)        update(p, sc, lson);    else        update(p, sc, rson);    PushUp(rt);}int querysum(int L,int R,int l,int r,int rt){    if (L <= l && r <= R)    {        return tree[rt].sum;    }    int m = (l + r) >> 1;    int ret = 0;    if (L <= m)        ret+= querysum(L, R, lson);    if (m < R)        ret += querysum(L, R, rson);    return ret;}int Qsum(int u,int v){    int sum=0;    int tp1 = top[u], tp2 = top[v];    while (tp1 != tp2)    {        if (dep[tp1] < dep[tp2])        {            swap(tp1, tp2);            swap(u, v);        }        sum+=querysum(id[tp1], id[u],1,n,1);        u = fa[tp1];        tp1 = top[u];    }    if (dep[u] > dep[v])        swap(u, v);    sum+=querysum(id[u], id[v],1,n,1);    return sum;}int querymax(int L,int R,int l,int r,int rt){    if (L <= l && r <= R)    {        return tree[rt].maxx;    }    int m = (l + r) >> 1;    int ret = -100000000;    if (L <= m)        ret = max(ret, querymax(L, R, lson));    if (R > m)        ret = max(ret, querymax(L, R, rson));    return ret;}int Qmax(int u,int v){    int sum=-100000000;    int tp1 = top[u], tp2 = top[v];    while (tp1 != tp2)    {        if (dep[tp1] < dep[tp2])        {            swap(tp1, tp2);            swap(u, v);        }        sum=max(sum,querymax(id[tp1], id[u],1,n,1));        u = fa[tp1];        tp1 = top[u];    }    if (dep[u] > dep[v])        swap(u, v);    sum=max(sum,querymax(id[u], id[v],1,n,1));    return sum;}void Clear(int n){    for(int i=1; i<=n; i++)        v[i].clear();}int main(){    scanf("%d",&n);    Clear(n);    for(int i=1; i<n; i++)    {        e[i].read();        v[e[i].x].push_back(e[i].y);        v[e[i].y].push_back(e[i].x);    }    for(int i=1; i<=n; i++)    {        scanf("%d",&coss[i]);    }    num=0;    dfs1(1,0,1);    dfs2(1,1);    build(1,n,1);    char s[200];    scanf("%d",&q);    while(q--)    {        scanf("%s",s);        int x,y,t;        if(s[0]=='Q'&&s[1]=='S')        {            scanf("%d%d",&x,&y);            printf("%d\n",Qsum(x,y));        }        else if(s[0]=='Q'&&s[1]=='M')        {            scanf("%d%d",&x,&y);            printf("%d\n",Qmax(x,y));        }        else if(s[0]=='C')        {            scanf("%d%d",&x,&t);            update(id[x],t,1,n,1);        }    }    return 0;}