zoukankan      html  css  js  c++  java
  • 【DFS序+单点修改区间求和】POJ 3321 Apple Tree

    poj.org/problem?id=3321

    【题意】

    给一棵n个节点的树,每个节点开始有一个苹果,m次操作
    1.将某个结点的苹果数异或 1
    2.查询一棵子树内的苹果数
      1 #include<iostream>
      2 #include<cstdio>
      3 #include<cstring>
      4 #include<string>
      5 #include<cmath>
      6 #include<algorithm>
      7 using namespace std;
      8 typedef long long ll;
      9 
     10 const int maxn=1e5+2;
     11 const int maxm=2*maxn;
     12 int n;
     13 struct edge
     14 {
     15     int to;
     16     int nxt;
     17 }e[maxm];
     18 int head[maxn];
     19 int tot;
     20 int l[maxn],r[maxn];
     21 int tree[maxm];
     22 bool vis[maxn];
     23 int cid;
     24 int lowbit(int x)
     25 {
     26     return x&(-x);
     27 }
     28 
     29 void add(int k,int x)
     30 {
     31     while(k<=cid)//注意是cid,不是n 
     32     {
     33         tree[k]+=x;
     34         k+=lowbit(k);
     35     }
     36 }
     37 
     38 int query(int k)
     39 {
     40     int ans=0;
     41     while(k)
     42     {
     43         ans+=tree[k];
     44         k-=lowbit(k);
     45     }
     46     return ans;
     47 }
     48 void init()
     49 {
     50     memset(head,-1,sizeof(head));
     51     tot=0;
     52     cid=0;
     53     memset(vis,false,sizeof(vis));
     54 }
     55 
     56 void addedge(int u,int v)
     57 {
     58     e[tot].to=v;
     59     e[tot].nxt=head[u];
     60     head[u]=tot++;
     61 }
     62 
     63 void dfs(int u,int pa)
     64 {
     65     l[u]=++cid;
     66     for(int i=head[u];i!=-1;i=e[i].nxt)
     67     {
     68         int v=e[i].to;
     69         if(v==pa) continue;
     70         dfs(v,u);    
     71     }    
     72     r[u]=++cid;
     73 }
     74 
     75 int main()
     76 {
     77     while(~scanf("%d",&n))
     78     {
     79         init();
     80         int u,v;
     81         for(int i=1;i<=n-1;i++)
     82         {
     83             scanf("%d%d",&u,&v);
     84             addedge(u,v);
     85             addedge(v,u);
     86         }
     87         dfs(1,-1);
     88         for(int i=1;i<=n;i++)
     89         {
     90             add(l[i],1);//单点修改 
     91         }    
     92         int q;
     93         scanf("%d",&q);
     94         char str[2];
     95         int x;
     96         while(q--)
     97         {
     98             scanf("%s%d",str,&x);
     99             if(str[0]=='Q')
    100             {
    101                 int ans=query(r[x])-query(l[x]-1);
    102                 printf("%d
    ",ans);
    103             }
    104             else
    105             {
    106                 if(!vis[x])
    107                 {
    108                     add(l[x],-1);
    109                     vis[x]=true;
    110                 }
    111                 else
    112                 {
    113                     add(l[x],1);
    114                     vis[x]=false;
    115                 }
    116             }
    117         }
    118     }
    119     return 0;
    120  } 
    View Code
  • 相关阅读:
    wode.
    python中迭代器和生成器。
    Embeded linux 之 UBIFS文件系统
    Windows下Git安装和使用
    套接字 之 windows与linux 差异
    Embeded linux之RTL8188EU/RTL8188ETV使用
    嵌入式Linux之“+”版本问题
    Uboot之net
    Embeded linux之reboot
    Embeded linux之cifs文件系统
  • 原文地址:https://www.cnblogs.com/itcsl/p/7418251.html
Copyright © 2011-2022 走看看