码迷,mamicode.com
首页 > 其他好文 > 详细

bzoj3531

时间:2017-12-06 21:41:01      阅读:105      评论:0      收藏:0      [点我收藏+]

标签:一个   struct   int   ext   end   main   printf   线段   数组   

题解:

这一道题目和模板的差别在于这一刀题目还有一个信仰

如果没有信仰,那么就是模板

然后考虑对于每一个信仰,我们都建立一颗线段树

注意要动态开点

所以要数组多开大一些

代码:

#include<bits/stdc++.h>
using namespace std;
const int N=200005;
int n,m,cnt,tot,al,roots[N],pre[N],c[N],w[N],top[N];
int x,y,z,fa[N],son[N],siz[N],dep[N],L[N],first[N];
char ch;
struct edge
{
    int u,v,next;
}e[N];
struct node
{
    int maxn,sum,lson,rson;
}T[N*10];
void pushup(int now)
{
    T[now].maxn=max(T[T[now].lson].maxn,T[T[now].rson].maxn);
    T[now].sum=T[T[now].lson].sum+T[T[now].rson].sum;
}
void add(int x,int y)
{
    e[++tot].u=x;
    e[tot].v=y;
    e[tot].next=first[x];
    first[x]=tot;
}
void dfs1(int now)
{
    siz[now]=1;
    for (int i=first[now];i;i=e[i].next)
    if (e[i].v!=fa[now])
     {
        fa[e[i].v]=now;
        dep[e[i].v]=dep[now]+1;
        dfs1(e[i].v);
        if (siz[e[i].v]>siz[son[now]]) son[now]=e[i].v;
        siz[now]+=siz[e[i].v];
     }
}
void dfs2(int now,int tp)
{
    top[now]=tp;pre[++cnt]=now;L[now]=cnt;
    if (son[now]) dfs2(son[now],tp);
    for (int i=first[now];i;i=e[i].next)
    if (e[i].v!=fa[now]&&e[i].v!=son[now]) dfs2(e[i].v,e[i].v);
}
void update(int &now,int begin,int end,int pos,int num)
{
    if (!now) now=++al;
    if (begin==end) {T[now].maxn=T[now].sum=num;return;}
    int mid=(begin+end)>>1;
    if (pos<=mid) update(T[now].lson,begin,mid,pos,num);
    else update(T[now].rson,mid+1,end,pos,num);
    pushup(now);
}
int get_sum(int now,int begin,int end,int l,int r)
{
    if (l<=begin&&end<=r) return T[now].sum;
    int mid=(begin+end)>>1,ans=0;
    if (mid>=l) ans+=get_sum(T[now].lson,begin,mid,l,r);
    if (mid<r) ans+=get_sum(T[now].rson,mid+1,end,l,r);
    return ans;
}
int get_max(int now,int begin,int end,int l,int r)
{
    if (l<=begin&&end<=r) return T[now].maxn;
    int mid=(begin+end)>>1,ans=0;
    if (mid>=l) ans=max(ans,get_max(T[now].lson,begin,mid,l,r));
    if (mid<r) ans=max(ans,get_max(T[now].rson,mid+1,end,l,r));
    return ans;
}
int find(int color,int l,int r)
{
    int f1=top[l],f2=top[r],ans=0;
    while (f1!=f2)
     {
        if (dep[f1]<dep[f2]) swap(f1,f2),swap(l,r);
        if (ch==S) ans+=get_sum(roots[color],1,cnt,L[f1],L[l]);
        else ans=max(ans,get_max(roots[color],1,cnt,L[f1],L[l]));
        l=fa[f1];f1=top[l];
     }
    if (dep[l]>dep[r]) swap(l,r);
    if (ch==S) ans+=get_sum(roots[color],1,cnt,L[l],L[r]);
    else ans=max(ans,get_max(roots[color],1,cnt,L[l],L[r]));
    return ans;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++)scanf("%d%d",&w[i],&c[i]);
    for (int i=1;i<n;i++)scanf("%d%d",&x,&y),add(x,y),add(y,x);
    dfs1(1);dfs2(1,1);
    for (int i=1;i<=n;i++)update(roots[c[i]],1,cnt,L[i],w[i]);
    while (m--)
     {
        ch=getchar();
        while (ch!=C&&ch!=Q) ch=getchar();
        if (ch==C)
         {
            ch=getchar();
            scanf("%d%d",&x,&y);
            if (ch==C)
             {
                 update(roots[c[x]],1,n,L[x],0);
                c[x]=y;
                update(roots[c[x]],1,n,L[x],w[x]);
             }
            else update(roots[c[x]],1,n,L[x],y),w[x]=y;
         }
        else
         {
            ch=getchar();
            scanf("%d%d",&x,&y);
            printf("%d\n",find(c[x],x,y));
         }
     }
}

 

bzoj3531

标签:一个   struct   int   ext   end   main   printf   线段   数组   

原文地址:http://www.cnblogs.com/xuanyiming/p/7994468.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!