zoukankan      html  css  js  c++  java
  • 树的统计Count (HYSBZ

      一棵树上有n个节点,编号分别为1到n,每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成一些操作:

      I.   CHANGE u t : 把结点u的权值改为t

     II.   QMAX u v: 询问从点u到点v的路径上的节点的最大权值

    III.   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                    Sample Output

    4                          4

    1 2                         1

    2 3                         2

    4 1                         2

    4 2 1 3                       10

    12                          6

    QMAX 3 4                       5

    QMAX 3 3                       6

    QMAX 3 2                       5

    QMAX 2 3                       16

    QSUM 3 4

    QSUM 2 1

    CHANGE 1 5

    QMAX 3 4

    CHANGE 3 6

    QMAX 3 4

    QMAX 2 4

    QSUM 3 4

    2018-08-02


      新学的树链剖分……冉到我了

    解析:

    所谓树链剖分,就是将一颗树剖为链状,实现节点替换、区间求和、区间最大(小)值 的算法

    既然有树,就少不了建图,需要用到先前学的“屠戮”知识:

    int first [MX];

    struct Edge{

      int to,next;

    }edge[MX];

    • 先深搜 dfs_1,更新节点信息: size(子节点到自己的个数)、fa(父节点)、heave(重儿子)…… 

    Attention:走到黑,才回头

    • 再深搜 dfs_2,更新: dfn(便利次序)、tou(头结点)

    Attention:先重后轻,所拆链段不重不漏

    • 照次序(dfn),赋点值(val)—— change(参考线段树)
    • 给区间,求操作 —— Find

    Attention:

    头不同,跳相同;头若同,可操作

    代码如下:

    #include<stdio.h>
    #include<algorithm>
    #define MX 30005
    #define ls x<<1
    #define rs x<<1|1
    using namespace std;
    
    struct Edge{
        int from,to,val,next;
    }edge[MX<<1];
    int n,q,cnt,ss,sf,tr[MX<<2][2],tou[MX],first[MX],dfn[MX],size[MX],fa[MX],heave[MX];
    
    void add(int from,int to)
    {
        edge[++cnt].to=to;
        edge[cnt].next=first[from];
        first[from]=cnt;
    }
    
    void dfs_1(int from)
    {
        size[from]=1;
        for(int i=first[from];i;i=edge[i].next)
        {
            int to=edge[i].to;
            if(to!=fa[from])
            {
                fa[to]=from;
                dfs_1(to);
                size[from]+=size[to];
                if(size[to]>size[heave[from]]) heave[from]=to;
            }
        }
    }
    
    void dfs_2(int from,int to)//(1,1)
    {
        dfn[from]=++ss;
        tou[from]=to;
        if(heave[from]) 
            dfs_2(heave[from],to);
        for(int i=first[from];i;i=edge[i].next)
        {
            int to=edge[i].to;
            if(to!=fa[from] && to!=heave[from]) dfs_2(to,to);
        }
    }
    
    void change(int x,int l,int r,int wz,int val)
    {
        if(l==r) tr[x][0]=tr[x][1]=val;
        else {
            int mid=(l+r)>>1;
            if(wz<=mid) change(ls,l,mid,wz,val);
            else         change(rs,mid+1,r,wz,val);
            tr[x][0]=tr[ls][0]+tr[rs][0];
            tr[x][1]=max(tr[ls][1],tr[rs][1]);
        }
    }
    
    int cl(int a,int b)
    {
        if(sf) return max(a,b);
        else return a+b;
    }
    
    int ask(int x,int l,int r,int ql,int qr)
    {
        if(l>=ql && r<=qr) return tr[x][sf];
        int mid=(l+r)>>1,re=sf?-30001:0;
        if(ql<=mid) re=cl(ask(ls,l,mid,ql,qr),re);
        if(qr>mid)    re=cl(ask(rs,mid+1,r,ql,qr),re);
        return re;
    }
    
    int Find(int from,int to)
    {
        int re=sf?-30001:0;
        while(tou[from]!=tou[to])
        {
            if(dfn[from]<dfn[to]) swap(from,to);
            re=cl(re,ask(1,1,n,dfn[tou[from]],dfn[from]));
            from=fa[tou[from]];
        }
        if(dfn[from]<dfn[to]) swap(from,to);
        re=cl(re,ask(1,1,n,dfn[to],dfn[from]));
        return re;
    }
    
    int main()
    {
        scanf("%d",&n);
        for(int i=1;i<n;++i)
        {
            int from,to;
            scanf("%d%d",&from,&to);
            add(from,to);
            add(to,from);
        }
        dfs_1(1);
        dfs_2(1,1);
        
        for(int i=1;i<=n;++i)
        {
            int val;
            scanf("%d",&val);
            change(1,1,n,dfn[i],val);
        }
        scanf("%d",&q);
        while(q--)
        {
            char con[10];
            int from,to;
            scanf("%s%d%d",con,&from,&to);
            if(con[0]=='C') change(1,1,n,dfn[from],to);
            else sf= con[1]=='M' ,printf("%d
    ",Find(from,to));            
        }
        return 0;
    }
    从0到1很难,但从1到100很容易
  • 相关阅读:
    正则表达式
    字典
    列表 元组
    int 和 str bool 类型 基本数据类型
    循环体的主要内容
    变量与循环体
    第12课.内存控制器与SDRAM
    第11课.串口(UART)的使用
    第10课.掌握ARM芯片时钟体系
    第9课.gcc和arm-linux-gcc和Makefile
  • 原文地址:https://www.cnblogs.com/qseer/p/9407908.html
Copyright © 2011-2022 走看看