杭电 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;

}