BZOJ 1036: [ZJOI2008]树的统计Count 树链剖分+线段树

1036: [ZJOI2008]树的统计Count

Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 17419 Solved: 7111
[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

一个像板一样的东西。以前打过板这次可以当个整理了23333之前还傻逼地以为要求LCA
傻逼long long（不用开long long） 但求max的初始值要小一点（负数）

//一定要强迫自己不要用st表 #include<cstdio>#include<cstring>#include<algorithm>#define ms(x,y) memset(x,y,sizeof(x))using namespace std;const int N = 100010;const int INF = 0x73f3f3f;int n,m;int a[N];struct node{    int pre,v;}edge[N];int num=0;int head[N];void addedge(int from,int to){    num++;    edge[num].pre=head[from];    edge[num].v=to;    head[from]=num;}int indx=0;int dep[N];int in[N],out[N],seq[N],son[N],siz[N],fa[N];void dfs1(int u,int f,int d){    siz[u]=1,fa[u]=f,dep[u]=d;    for(int i=head[u];i;i=edge[i].pre){        int v=edge[i].v;        if(v==f) continue;        dfs1(v,u,d+1);        siz[u]+=siz[v];        if(son[u]==-1||siz[v]>siz[son[u]]) son[u]=v;    }}int top[N];int al[N];void dfs2(int u,int tp){    indx++;    in[u]=out[u]=indx,seq[indx]=u;    al[in[u]]=a[u];    top[u]=tp;    if(son[u]==-1) return ;    dfs2(son[u],tp);    for(int i=head[u];i;i=edge[i].pre){        int v=edge[i].v;        if(v==son[u]||v==fa[u]) continue;        dfs2(v,v);    }    out[u]=indx;}inline int Max(int a,int b){    return a>b?a:b;}struct Node{    int sum;    int mmax,flag;    int l,r;}t[N];void build(int root,int l,int r){    t[root].l=l,t[root].r=r;    if(l==r){        t[root].mmax=t[root].sum=al[l];        return ;    }    int mid=(l+r)>>1;    build(root<<1,l,mid);    build(root<<1|1,mid+1,r);    t[root].sum=t[root<<1].sum+t[root<<1|1].sum;    t[root].mmax=Max(t[root<<1].mmax,t[root<<1|1].mmax);}/*int getlca(int u,int v){    while(1){        if(top[u]==top[v]){            if(dep[u]<dep[v]) return u;            return v;        }        else if(dep[top[u]]>=dep[top[v]]) u=fa[top[u]];        else v=fa[top[v]];    }}*/void modify(int root,int pos,int delta){    int l=t[root].l,r=t[root].r;    if(l==r&&l==pos){        t[root].sum=t[root].mmax=delta;        return ;    }    int mid=(l+r)>>1;    if(pos<=mid) modify(root<<1,pos,delta);    else modify(root<<1|1,pos,delta);    t[root].sum=t[root<<1].sum+t[root<<1|1].sum;    t[root].mmax=Max(t[root<<1].mmax,t[root<<1|1].mmax);}int getmax(int root,int pos,int val){    int l=t[root].l,r=t[root].r;    if(pos<=l&&val>=r) return t[root].mmax;    int mid=(l+r)>>1;    int ans=-INF;    if(pos<=mid) ans=Max(ans,getmax(root<<1,pos,val));    if(val>mid) ans=Max(ans,getmax(root<<1|1,pos,val));    t[root].mmax=Max(t[root<<1].mmax,t[root<<1|1].mmax);    return ans;}int getmax(int u,int v){    int f1=top[u],f2=top[v];    int mmax=-INF;    while(f1!=f2){        if(dep[f1]<dep[f2]) swap(f1,f2),swap(u,v);        mmax=Max(mmax,getmax(1,in[f1],in[u]));        u=fa[f1];f1=top[u];    }    if(dep[u]<dep[v]) swap(u,v);    mmax=Max(mmax,getmax(1,in[v],in[u]));    return mmax;}int getsum(int root,int pos,int val){    int l=t[root].l,r=t[root].r;    if(pos<=l&&val>=r) return t[root].sum;    int mid=(l+r)>>1;    int ans=0;    if(pos<=mid) ans+=getsum(root<<1,pos,val);    if(val>mid) ans+=getsum(root<<1|1,pos,val);    t[root].sum=t[root<<1].sum+t[root<<1|1].sum;    return ans;}int getsum(int u,int v){    int f1=top[u],f2=top[v];    int sum=0;    while(f1!=f2){        if(dep[f1]<dep[f2]) swap(f1,f2),swap(u,v);        sum+=getsum(1,in[f1],in[u]);        u=fa[f1],f1=top[u];    }    if(dep[u]>dep[v]) swap(u,v);    sum+=getsum(1,in[u],in[v]);    return sum;}int main(){    ms(son,-1);    scanf("%d",&n);    for(register int i=1;i<n;i++){        int u,v;        scanf("%d%d",&u,&v);        addedge(u,v);addedge(v,u);    }    for(register int i=1;i<=n;i++) scanf("%d",&a[i]);       dfs1(1,-1,0);dfs2(1,1);    build(1,1,n);    scanf("%d",&m);    while(m--){        char s[10];        scanf("%s",s);        if(s[0]=='C'){            int u,delta;            scanf("%d%d",&u,&delta);            modify(1,in[u],delta);        }        else if(s[0]=='Q'&&s[1]=='M'){            int u,v;            scanf("%d%d",&u,&v);            printf("%d\n",getmax(u,v));        }        else{            int u,v;            scanf("%d%d",&u,&v);            int ans=getsum(u,v);            printf("%d\n",ans);        }    }    return 0;}