zoukankan      html  css  js  c++  java
  • POJ3321 Apple tree

    /*
        很容易发现这题用线段树来做
    树形结构稍微处理一下就可以变为线性结构,dfs跑一遍将树中的结点重新编号,对于每个节点,当访问完子树之后,将子树的最大节点也保存,这样就能得到一个区间段,于是乎,每棵子树的编号就都变成连续的(读入时的标号与线段树中实际标号不一样,所以先用一个pre数组来记录原来的标号),剩下的就可以套用线段树或者是树状数组的模板了,当查询某个结点时查询所对应的连续区间就可以了
    */
    
    #include<iostream>
    #include<cstdio>
    #include<algorithm> 
    #include<cstring> 
    using namespace std; 
    #define maxn 100010
    struct node{ 
        int lef, rig, id;
    }tree[maxn];
    int root[maxn];
    int n, q, v; 
    struct E{
        int v, next; 
    }edge[maxn]; 
    int pre[maxn], apple[maxn];
    struct SEG{ 
        int lef, rig, mid, ans; 
    }seg[maxn*4];  
    void dfs(int pp)
    {
        tree[pp].lef = v; 
        for(int i= pre[pp]; i!= -1; i= edge[i].next)
          dfs(edge[i].v);
        root[pp] = v; tree[pp].rig = v++; 
    } 
    void maketree(int num, int lef, int rig)
    { 
        seg[num].lef = lef; 
        seg[num].rig = rig; 
        seg[num].mid = (lef + rig) >> 1; 
        seg[num].ans = rig - lef + 1; 
        if( lef != rig)
        { 
            maketree(num*2, lef, seg[num].mid); 
            maketree(num*2+1, seg[num].mid+1, rig); 
        } 
    } 
    int cal( int num, int lef, int rig)
    { 
        if( seg[num].lef == lef && seg[num].rig == rig)
          return seg[num].ans; 
          if( rig <= seg[num].mid)
            return cal( num*2, lef, rig); 
          else if( lef > seg[num].mid)
            return cal( num*2+1 , lef, rig);
          else return cal( num*2, lef , seg[num].mid) + cal(num*2+1, seg[num].mid+1, rig); 
    } 
    void insert( int num, int lef, int val)
    { 
        seg[num].ans += val; 
        if( seg[num].lef == seg[num].rig) 
          return; 
        if( lef <= seg[num].mid) 
          insert( num*2, lef, val); 
        else if( lef > seg[num].mid) 
          insert( num*2+1, lef, val); 
    }
    void init() 
    { 
        int aa, bb, value = 0; 
        memset( pre, -1, sizeof(pre)); 
        for(int i= 1; i<n; i++) 
        { 
            scanf("%d%d", &aa, &bb); 
            edge[value].v = bb; 
            edge[value].next = pre[aa];  
            pre[aa] = value++; 
        } 
        for(int i= 1; i<= n; i++) 
          apple[i] = 1; 
    }
    int main() 
    { 
        int aa, bb; 
        char tmp[10]; 
        scanf("%d", &n); 
        init(); 
        v = 1; 
        dfs(1); 
        maketree(1, 1, n); 
        scanf("%d", &q); 
        for(int i= 0; i<q; i++) 
        { 
            scanf("%s%d", tmp, &aa); 
            if( tmp[0] == 'Q') 
              printf("%d
    ", cal(1, tree[aa].lef, tree[aa].rig)); 
            else 
            { 
                if( apple[root[aa]] == 1) 
                bb = -1; 
                else bb = 1; 
                insert( 1, root[aa], bb); 
                apple[root[aa]] = 1 - apple[root[aa]]; 
            } 
        }
        return 0; 
    }
  • 相关阅读:
    JavaScript对原始数据类型的拆装箱操作
    Javascript继承(原始写法,非es6 class)
    动态作用域与词法作用域
    自行车的保养
    探索JS引擎工作原理 (转)
    C语言提高 (7) 第七天 回调函数 预处理函数DEBUG 动态链接库
    C语言提高 (6) 第六天 文件(续) 链表的操作
    C语言提高 (5) 第五天 结构体,结构体对齐 文件
    C语言提高 (4) 第四天 数组与数组作为参数时的数组指针
    C语言提高 (3) 第三天 二级指针的三种模型 栈上指针数组、栈上二维数组、堆上开辟空间
  • 原文地址:https://www.cnblogs.com/xiaoningmeng/p/5765143.html
Copyright © 2011-2022 走看看