zoukankan      html  css  js  c++  java
  • HDU 3887 Counting Offspring (DFS序+树状数组)

    <题目链接>

    题目大意:

    给定一个$n$个节点的无向树,并且给出这棵树的根,然后问你,对于每个节点,输出该节点的子树中节点序号小于它的节点个数。

    解题分析:
    用DFS序将整棵树拍平,然后用树状数组维护一下这个DFS序,$in[x]$到$out[x]$之间的节点表示的就是以这个点为根的子树中的节点。按照序号从小到大遍历,然后每次遍历,先询问$(in[x],out[x])$之间已经被点亮的点的个数,询问后将自己点亮即可。

    #include <bits/stdc++.h>
    using namespace std;
    
    template<typename T>
    inline void read(T&x){
        x=0;int f=1;char c=getchar();
        while(c<'0'||c>'9'){ if(c=='-')f=-1;c=getchar(); }
        while(c>='0' && c<='9'){ x=x*10+c-'0'; c=getchar(); }
        x*=f;
    }
    #define REP(i,s,t) for(int i=s;i<=t;i++)
    #define clr(a,b) memset(a,b,sizeof(a))
    const int N = 1e5+5;
    
    struct Edge{ int to,nxt; }e[N<<1];
    int n,m,cnt,cur,tot;
    int head[N],tr[N<<1],in[N],out[N];
    
    inline void init(){
        cnt=tot=0;
        clr(head,-1);clr(tr,0);
        cur=n<<1;
    }
    inline void add(int u,int v){
        e[++cnt]=(Edge){ v,head[u] };head[u]=cnt;
    }
    void dfs(int u,int pre){        //求出DFS序 
        in[u]=++tot;
        for(int i=head[u];~i;i=e[i].nxt){
            int v=e[i].to;
            if(v==pre)continue;
            dfs(v,u);
        }
        out[u]=++tot;
    }
    inline int lowbit(int x){ return x&(-x); }
    
    inline void update(int i){
        while(i<=cur){
            tr[i]+=1;
            i+=lowbit(i);
        }
    }
    inline int sum(int i){
        int ans=0;
        while(i>=1){
            ans+=tr[i];
            i-=lowbit(i);
        }
        return ans;
    }
    int main(){
        while(~scanf("%d%d",&n,&m),n||m){
            init();
            REP(i,1,n-1){
                int u,v;read(u);read(v);
                add(u,v);add(v,u);
            }
            dfs(m,-1);
            REP(i,1,n){
                int now=sum(out[i]-1)-sum(in[i]);    //因为要严格小于
                i==n?printf("%d
    ",now):printf("%d ",now);
                update(in[i]);
            }
        }
    }
  • 相关阅读:
    9本Java程序员必读的书
    最短路径问题:dijkstar
    RSA加密算法
    BFC 浅谈
    纯css3配合vue实现微信语音播放效果
    Vue内置组件keep-alive的使用
    vim常用命令
    Java实体映射工具MapStruct的使用
    hexo文章编写部分语法总结以及hexo使用
    高级进程间通信
  • 原文地址:https://www.cnblogs.com/00isok/p/10892389.html
Copyright © 2011-2022 走看看