HihoCoder1145幻想乡的日常【树状数组】

描述

幻想乡一共有n处居所，编号从1到n。这些居所被n-1条边连起来，形成了一个树形的结构。

每处居所都居住着一个小精灵。每天小精灵们都会选出一个区间[l,r]，居所编号在这个区间内的小精灵一起来完成一项任务。

特别的，居所相邻的(有边相连的)两个小精灵会自发的组成一队，并且如果a和b相邻b和c相邻，那么a和c也在同一队里面。每天的任务完成之后，队伍就会解散；第二天再根据新的区间组成新的队伍。

给出每天小精灵们选出的区间，你知道每天组成的队伍数量吗？

输入

第一行两个数n和Q(1 <= n, Q <= 100000)，表示居所的数目和组队的天数。

接下来n-1行，每行两个数a和b，表示居所a和b之间有一条边。

接下来Q行，每行两个数l和r，满足1<=l<=r<=n，为该天小精灵选出的区间。

输出

输出Q行，每行一个整数表示该天队伍的数量。

样例输入

3 1
1 2
2 3
1 3

样例输出

1

解题思路：

注意到树的性质：对于询问[l,r]，假设最初所有点都是独立的，那么只要其中有一条边，就会合并两个队伍，所以答案即为区间长度-区间内边数。
对于多个询问，我们离线+树状数组处理即可。

#include<bits/stdc++.h>
using namespace std;

int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}

const int N=100005;
int n,m,bit[N],ans[N];
int tot,first[N],nxt[N<<1],to[N<<1];
struct node{int l,r,id;}q[N];
inline bool cmp(const node &a,const node &b)
{return a.r<b.r;}

void add(int x,int y)
{
    nxt[++tot]=first[x],first[x]=tot,to[tot]=y;
}

void insert(int x,int v)
{
    for(int i=x;i<=n;i+=i&(-i))
        bit[i]+=v;
}

int query(int x)
{
    int res=0;
    for(int i=x;i;i-=i&(-i))res+=bit[i];
    return res;
}

int main()
{
    //freopen("lx.in","r",stdin);
    n=getint(),m=getint();
    for(int i=1;i<n;i++)
    {
        int x=getint(),y=getint();
        add(x,y),add(y,x);
    }
    for(int i=1;i<=m;i++)
        q[i].l=getint(),q[i].r=getint(),q[i].id=i;
    sort(q+1,q+m+1,cmp);
    for(int i=1,j=1;i<=n;i++)
    {
        for(int e=first[i];e;e=nxt[e])
            if(to[e]<i)insert(to[e],1);
        while(q[j].r<=i&&j<=m)
        {
            ans[q[j].id]=q[j].r-q[j].l+1-(query(q[j].r)-query(q[j].l-1));
            j++;
        }
    }
    for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
    return 0;
}