Hrbust 2065 萌萌哒十五酱的苹果~【树链剖分+线段树】

萌萌哒十五酱的苹果~

Time Limit: 3000 MS Memory Limit: 6000 K Total Submit: 42(7 users) Total Accepted: 5(5 users) Rating:  Special Judge: No

Description

十五家的院子很大诶~     所以可以种树啊~     所以就种了棵苹果树啊（我在废话昂，，，，，，     树上面有n个节点，编号是1~n，每个节点上面有wi个苹果啦><     苹果当然要数一下啦~（诶！！这是为什么。。。     十五自己数起来太累了嘛~所以十五希望你可以帮忙数苹果     总共大概有这些情况（喂喂！！什么叫大概啊！     1.CHANGE u t：树上面标号为u的节点的苹果数改为t     2.QMAX u v:询问点u到v的路径上苹果树最大的苹果数是多少     3.QSUM u v：询问点u到v上总共有多少个苹果~

Input

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

Output

多组数据QAQ     对于十五的每一个“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

十五酱最萌了昂~

Source

HCPC2014校赛训练赛 6

思路：

暴力剖分然后线段树维护一下。

板子题，码农一下。练习一下算法手敲熟练度。

值得注意一个点是，这个题卡了内存，不能开太大方，随之可能改了内存变小之后，吃了屎一样可能会忘记线段树*4的问题。

Wa了好半天= =

Ac代码：

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<vector>
using namespace std;
int n;
vector<int>mp[30500];
int val[30500];
#define lson l,m,rt*2
#define rson m+1,r,rt*2+1
int sum[30500*4];
int maxn[30500*4];
void pushup(int rt)
{
    maxn[rt]=max(maxn[rt*2],maxn[rt*2+1]);
    sum[rt]=sum[rt*2]+sum[rt*2+1];
}
void update(int p,int c,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=c;
        maxn[rt]=c;
        return ;
    }
    pushup(rt);
    int m=(l+r)/2;
    if(p<=m)update(p,c,lson);
    if(p>m)update(p,c,rson);
    pushup(rt);
}
void build(int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=0;
        maxn[rt]=0;
        return ;
    }
    int m=(l+r)/2;
    build(lson);
    build(rson);
    pushup(rt);
}
int query_sum(int L,int R,int l,int r,int rt)
{
    if(l>=L&&r<=R)
    {
        return sum[rt];
    }
    pushup(rt);
    int ans=0;
    int m=(l+r)/2;
    if(L<=m)ans+=query_sum(L,R,lson);
    if(R>m)ans+=query_sum(L,R,rson);
    pushup(rt);
    return ans;
}
int query_maxn(int L,int R,int l,int r,int rt)
{
    if(l>=L&&r<=R)
    {
        return maxn[rt];
    }
    pushup(rt);
    int ans=-0x3f3f3f3f;
    int m=(l+r)/2;
    if(L<=m)ans=max(ans,query_maxn(L,R,lson));
    if(R>m)ans=max(ans,query_maxn(L,R,rson));
    pushup(rt);
    return ans;
}
int cnt;
int size[30500];
int fa[30500];
int son[30500];
int depth[30500];
int Top[30500];
int dfn[30500];
void Dfs(int u,int from,int d)
{
    depth[u]=d;size[u]=1;fa[u]=from;son[u]=-1;
    for(int i=0;i<mp[u].size();i++)
    {
        int v=mp[u][i];
        if(v==from)continue;
        Dfs(v,u,d+1);
        size[u]+=size[v];
        if(son[u]==-1||size[v]>size[son[u]])
        {
            son[u]=v;
        }
    }
}
void Dfs2(int u,int from,int top)
{
    dfn[u]=++cnt;Top[u]=top;
    if(son[u]!=-1)Dfs2(son[u],u,top);
    for(int i=0;i<mp[u].size();i++)
    {
        int v=mp[u][i];
        if(v==from||v==son[u])continue;
        Dfs2(v,u,v);
    }
}
void Slove_sum(int x,int y)
{
    int ans=0;
    int fx=Top[x],fy=Top[y];
    while(fx!=fy)
    {
        if(depth[fx]<depth[fy])
        {
            swap(fx,fy);
            swap(x,y);
        }
        ans+=query_sum(dfn[fx],dfn[x],1,n,1);
        x=fa[fx],fx=Top[x];
    }
    if(depth[x]<depth[y])swap(x,y);
    ans+=query_sum(dfn[y],dfn[x],1,n,1);
    printf("%d\n",ans);
}
void Slove_maxn(int x,int y)
{
    int ans=-0x3f3f3f3f;
    int fx=Top[x],fy=Top[y];
    while(fx!=fy)
    {
        if(depth[fx]<depth[fy])
        {
            swap(fx,fy);
            swap(x,y);
        }
        ans=max(ans,query_maxn(dfn[fx],dfn[x],1,n,1));
        x=fa[fx],fx=Top[x];
    }
    if(depth[x]<depth[y])swap(x,y);
    ans=max(ans,query_maxn(dfn[y],dfn[x],1,n,1));
    printf("%d\n",ans);
}
int main()
{
    while(~scanf("%d",&n))
    {
        memset(maxn,0,sizeof(maxn));
        memset(sum,0,sizeof(sum));
        for(int i=1;i<=n;i++)mp[i].clear();
        for(int i=1;i<=n-1;i++)
        {
            int x,y;scanf("%d%d",&x,&y);
            mp[x].push_back(y);
            mp[y].push_back(x);
        }
        cnt=0;
        Dfs(1,-1,1);
        Dfs2(1,-1,1);
        build(1,n,1);
        for(int i=1;i<=n;i++)scanf("%d",&val[i]),update(dfn[i],val[i],1,n,1);
        int q;scanf("%d",&q);
        while(q--)
        {
            char s[150];
            scanf("%s",s);
            if(s[1]=='S')
            {
                int x,y;scanf("%d%d",&x,&y);
                Slove_sum(x,y);
            }
            if(s[1]=='M')
            {
                int x,y;scanf("%d%d",&x,&y);
                Slove_maxn(x,y);
            }
            if(s[0]=='C')
            {
                int x,y;scanf("%d%d",&x,&y);
                update(dfn[x],y,1,n,1);
            }
        }
    }
}