杭电 3887 Counting Offspring

根据上篇翻译的文章以及很多个帖子，都讲述了树状数组最基本的功能就是tree[i]保存的是位置i左边小于等于a[i]的数的个数.

这样也就可以解释代码中为什么有f[i]=getsum(sd[i-1])-getsum(st[i]))/2。因为getsum保存的就是左边比i小的数，注意因为序列是通过dfs求出的，因而每个节点都有进入和退出过程，也就是每个节点都出现了2次，比如说对于数4来说，有4个节点，假设3为顶点，边的关系是3-2-1,3-2-4，那么dfs扫描出的序列就是3,2,1,1,4,4,2,3.所以求出的最终结果就需要除以2，因为每个数字都出现了2次。

至于为什么是从n-->1，我也纠结了半天，后来总算是YY出了一点思路【也可能不对】，因为已经将tree初始过了，考虑最原始的BIT，我们是一边遍历原始数组，一边getsum，一边update树，更新时，tree[i]+=1，这里提前将tree update过了，那么只能从后向前走，update(-1)了。假定最后一个点是n，那么在考虑其他节点的时候是不需要考虑这个点的，因为任何一个点都比最后这个点小，所以。。。update的时候是（-1）。

代码中注释的部分是原作者的，为了证实一下我自己YY出的结果是否正确，我测试了一下，果然A掉了。所以我的想法应该是对的，如果从头开始的话，需要从头遍历数组，得结果的时候需要两次update（1），这样的话仅仅需要从1-->就可以了。

代码如下：

#pragma comment(linker,"/STACK:100000000,100000000")#include <iostream>#include <cstdio>#include <algorithm>#include <vector>#include <cstring>using namespace std;const int maxn=100005;vector<int>vt[maxn];int bit[2*maxn];int que[2*maxn];int st[maxn];int sd[maxn];int f[maxn];int n, rt, num;void dfs(int u, int fa){    que[++num]=u;    for(int i=0; i<vt[u].size(); i++)    {        int v=vt[u][i];        if(v==fa) continue;        dfs(v,u);    }    que[++num]=u;}int lowbit(int x){    return x&(-x);}void update(int x, int val){    while(x<=num)    {        bit[x]+=val;        x+=lowbit(x);    }}int getsum(int x){    int ans=0;    while(x>0)    {        ans+=bit[x];        x-=lowbit(x);    }    return ans;}int main(){//int a=4,b=3;//printf("%d",a|b);    while(~scanf("%d%d",&n,&rt),n+rt)    {        for(int i=0; i<=n; i++)            vt[i].clear();        for(int i=1; i<n; i++)        {            int x, y;            scanf("%d%d",&x,&y);            vt[x].push_back(y);            vt[y].push_back(x);        }        fill(st+1,st+1+n,0);        num=0;        dfs(rt,-1);//每个节点开始和结束的位置        for(int i=1; i<=num; i++)        {            if(!st[que[i]])st[que[i]]=i;            else sd[que[i]]=i;        }        memset(bit,0,sizeof(bit));/*        for(int i=1; i<=num; i++)            update(i,1);        for(int i=n; i>=1; i--)        {            f[i]=(getsum(sd[i]-1)-getsum(st[i]))/2;            update(st[i],-1);            update(sd[i],-1);        }*/for(int i=1;i<=n;i++)//这里是测试从1-->n的，注意对比{update(st[i],1);update(sd[i],1);f[i]=(getsum(sd[i]-1)-getsum(st[i]))/2;}        printf("%d",f[1]);        for(int i=2; i<=n; i++)            printf(" %d",f[i]);        puts("");    }    return 0;}