bzoj1036: [ZJOI2008]树的统计Count

来源:互联网 发布:电脑学打字软件 编辑:程序博客网 时间:2024/03/28 23:07

树链剖分+线段树(入门)

/**************************************************************
    Problem: 1036
    User: 96655
    Language: C++
    Result: Accepted
    Time:2472 ms
    Memory:5536 kb
****************************************************************/
 
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#include<iostream>
#include<cstdlib>
#include<queue>
#include<map>
#include<set>
#include<vector>
#include<cmath>
#include<stack>
#include<utility>
usingnamespace std;
constint maxn=30005;
structEdge
{
    intv,next;
} edge[maxn*2];
intva[maxn],p,head[maxn],clk,n;
voidinit()
{
    memset(head,-1,sizeof(head));
    p=clk=0;
}
voidaddedge(intu,intv)
{
    edge[p].v=v;
    edge[p].next=head[u];
    head[u]=p++;
}
intid[maxn],sz[maxn],dep[maxn],fa[maxn],son[maxn],top[maxn];
intxx[maxn];
voiddfs1(intu,intf,intd)
{
    dep[u]=d;
    fa[u]=f;
    sz[u]=1;
    son[u]=-1;
    for(inti=head[u]; ~i; i=edge[i].next)
    {
        intv=edge[i].v;
        if(v==f)continue;
        dfs1(v,u,d+1);
        sz[u]+=sz[v];
        if(son[u]==-1||sz[v]>sz[son[u]])
            son[u]=v;
    }
}
voiddfs2(intu,inttp)
{
    id[u]=++clk;
    xx[id[u]]=va[u];
    top[u]=tp;
    if(son[u]!=-1)dfs2(son[u],tp);
    for(inti=head[u]; ~i; i=edge[i].next)
    {
        intv=edge[i].v;
        if(v==fa[u]||v==son[u])continue;
        dfs2(v,v);
    }
}
intsum[maxn*4],maxv[maxn*4];
voidpushup(intrt)
{
    sum[rt]=sum[rt*2]+sum[rt*2+1];
    maxv[rt]=max(maxv[rt*2],maxv[rt*2+1]);
}
voidbuild(intrt,intl,intr)
{
    if(l==r)
    {
        sum[rt]=maxv[rt]=xx[l];
        return;
    }
    intm=(l+r)>>1;
    build(rt*2,l,m);
    build(rt*2+1,m+1,r);
    pushup(rt);
}
voidupdate(intrt,intl,intr,intpos,intc)
{
    if(l==r)
    {
        maxv[rt]=sum[rt]=c;
        return;
    }
    intm=(l+r)>>1;
    if(pos<=m)update(rt*2,l,m,pos,c);
    elseupdate(rt*2+1,m+1,r,pos,c);
    pushup(rt);
}
intquery1(intrt,intl,intr,intx,inty)
{
    if(x<=l&&r<=y)
    {
        returnsum[rt];
    }
    intans=0;
    intm=(l+r)>>1;
    if(x<=m)ans+=query1(rt*2,l,m,x,y);
    if(y>m)ans+=query1(rt*2+1,m+1,r,x,y);
    returnans;
}
intquery2(intrt,intl,intr,intx,inty)
{
    if(x<=l&&r<=y)
    {
        returnmaxv[rt];
    }
    intans=-99999;
    intm=(l+r)>>1;
    if(x<=m)ans=max(ans,query2(rt*2,l,m,x,y));
    if(y>m)ans=max(ans,query2(rt*2+1,m+1,r,x,y));
    returnans;
}
intgetsum(intu,intv)
{
    intans=0;
    while(top[u]!=top[v])
    {
        if(dep[top[u]]<dep[top[v]])
            swap(u,v);
        ans+=query1(1,1,n,id[top[u]],id[u]);
        u=fa[top[u]];
    }
    if(dep[u]>dep[v])swap(u,v);
    ans+=query1(1,1,n,id[u],id[v]);
    returnans;
}
intgetmax(intu,intv)
{
    intans=-99999;
    while(top[u]!=top[v])
    {
        if(dep[top[u]]<dep[top[v]])
            swap(u,v);
        ans=max(ans,query2(1,1,n,id[top[u]],id[u]));
        u=fa[top[u]];
    }
    if(dep[u]>dep[v])swap(u,v);
    ans=max(ans,query2(1,1,n,id[u],id[v]));
    returnans;
}
intmain()
{
    while(~scanf("%d",&n))
    {
        init();
        for(inti=1; i<n; ++i)
        {
            intu,v;
            scanf("%d%d",&u,&v);
            addedge(u,v);
            addedge(v,u);
        }
        for(inti=1; i<=n; i++)
            scanf("%d",&va[i]);
        dfs1(1,1,0);
        dfs2(1,1);
        build(1,1,n);
        intq;
        scanf("%d",&q);
        while(q--)
        {
            chars[20];
            intx,y;
            scanf("%s%d%d",s,&x,&y);
            if(s[0]=='Q')
            {
                if(s[1]=='M')printf("%d\n",getmax(x,y));
                elseprintf("%d\n",getsum(x,y));
            }
            elseupdate(1,1,n,id[x],y);
        }
    }
    return0;
}

0 0
原创粉丝点击