hihoCoder 1145 幻想乡的日常(树状数组 + 离线处理)

时间:2023-03-09 08:32:04
hihoCoder 1145 幻想乡的日常(树状数组 + 离线处理)

http://hihocoder.com/problemset/problem/1145?sid=1244164

题意:

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

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

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

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

思路:
越来越觉得树状数组在处理区间问题上真的很强。

将询问按右坐标排序,将顶点按从小到大的顺序看它的子节点是否可以插入。

 #include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = +; int tot,n,m;
int head[maxn],c[maxn],ans[maxn]; struct node
{
int v,next;
}e[*maxn]; struct Node
{
int l,r,id;
bool operator< (const Node& rhs) const
{
return r<rhs.r;
}
}query[maxn]; void addEdge(int u, int v)
{
e[tot].v = v;
e[tot].next = head[u];
head[u] = tot++;
} int lowbit(int x)
{
return x&-x;
} int sum(int x)
{
int ret = ;
while(x>)
{
ret += c[x];
x -= lowbit(x);
}
return ret;
} int add(int x)
{
while(x<=n)
{
c[x] += ;
x +=lowbit(x);
}
} int main()
{
//freopen("in.txt","r",stdin);
scanf("%d%d",&n,&m);
tot = ;
memset(head,-,sizeof(head));
for(int i=;i<n-;i++)
{
int u,v;
scanf("%d%d",&u,&v);
addEdge(u,v);
addEdge(v,u);
}
for(int i=;i<m;i++)
{
scanf("%d%d",&query[i].l,&query[i].r);
query[i].id = i;
}
int pos = ;
sort(query,query+m);
for(int u=;u<=n;u++)
{
for(int i=head[u];i!=-;i=e[i].next)
if(e[i].v<u) add(e[i].v);
while(u==query[pos].r && pos<m)
{
ans[query[pos].id] = query[pos].r-query[pos].l+-(sum(u)-sum(query[pos].l-));
pos++;
}
}
for(int i=;i<m;i++)
printf("%d\n",ans[i]);
return ;
}