【题目描述】
IcePrincess_1968 和 IcePrince_1968 长大了,他们开始协助国王 IceKing_1968 管理国内事物。
IcePrincess_1968 和 IcePrince_1968 住在一个宁静悠远的王国:IceKingdom —— 飘雪圣域。飘雪圣域有 nn 个城镇,编号 1,2,3…n1,2,3…n。有些城镇之间有道路,且满足任意两点之间有且仅有一条路径。飘雪圣域风景优美,但气候并不是太好。根据 IcePrince_1968 的气候探测仪,将来会发生 qq 场暴风雪。每场暴风雪可以用两个整数 li,rili,ri刻画,表示这场暴风雪之后,只有编号属于[li,ri][li,ri]的城市没有受到暴风雪的影响。
在暴风雪的影响下迅速确定王国的农业生产方案是非常重要的事情。IceKing_1968 认为,一个农业生产地域应该是一个极大连通块,满足每个节点都没有被暴风雪影响。这里极大连通块的定义是:不存在一个不属于该点集的未被暴风雪影响的点与该连通块连通。
IcePrincess_1968 要负责算出每次暴风雪后,王国能拥有多少个农业生产地域。注意这里每次暴风雪是独立的,即每次暴风雪过后,直到每个城镇重新焕发生机,下一次暴风雪才会到来。
正如上文所述,IcePrincess_1968 擅长文学但不擅长计算机,于是请你帮忙。
【输入】
第一行包含两个正整数 n,qn,q,表示 IceKingdom 的城镇个数和暴风雪次数。
第 22 至第 nn 行,每行两个正整数 x,yx,y,表示城镇 xx 和城镇 yy 之间有一条道路。
第 n+1n+1 至第 n+qn+q 行,每行两个正整数 li,rili,ri,描述一场暴风雪,含义如题面所述。
【输出】
输出文件共有 qq 行,第 ii 行表示在第 ii 场暴风雪之后农业生产地域的个数。
【样例输入】
4 3
1 2
2 3
2 4
1 2
1 3
3 4
【样例输出 】
1
1
2
【思路】
我们发现:每增加一条好边,联通块个数一定减一。(由树的性质可得)
因此我们可以考虑维护每一次询问的条件下的好边数量。我们定义一条边为好边,当且仅当其端点都在[li,ri]之间。鉴于其树的状态一定,我们可以考虑离线算法。我们规定u为边e的小编号端点,v为大编号端点。我们按v从大到小对每条边排序,按询问的右端点大小从大到小排序,一条边必须满足u>=l且v<=r才是好边。由此,我们可以用树状数组维护左端点在li->ri间的边的数量,依次回答每个询问,并逐次删去右端点不合法的点,就可以解决问题了。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<queue>
#define re register
using namespace std;
int n,m,a,b,s,t;
inline int low(int x){return x&(-x);}
int c[200001];
struct node1{
int u,v;
}e[200001];
inline void change(int k,int v)
{
while(k<=n)
{
c[k]+=v;
k+=low(k);
}
}
inline int ask(int k)
{
int ret=0;
while(k>0)
{
ret+=c[k];
k-=low(k);
}
return ret;
}
struct node2{
int id;
int l,r;
int ans;
}q[200001];
int now;
inline bool cmp1(node1 a,node1 b){return a.v>b.v;}
inline bool cmp2(node2 a,node2 b){return a.r>b.r;}
inline bool cmp3(node2 a,node2 b){return a.id<b.id;}
int main()
{
scanf("%d%d",&n,&m);
for(int re i=1;i<n;i++)
{
scanf("%d%d",&e[i].u,&e[i].v);
if(e[i].u>e[i].v)swap(e[i].u,e[i].v);
change(e[i].u,1);
}
sort(e+1,e+n,cmp1);
now=1;
for(int re i=1;i<=m;i++)
{
scanf("%d%d",&q[i].l,&q[i].r);
q[i].id=i;
}
sort(q+1,q+m+1,cmp2);
for(int re i=1;i<=m;i++)
{
while(e[now].v>q[i].r)change(e[now++].u,-1);
q[i].ans=q[i].r-q[i].l+1-(ask(q[i].r)-ask(q[i].l-1));
}
sort(q+1,q+m+1,cmp3);
for(int re i=1;i<=m;i++)
{
printf("%d\n",q[i].ans);
}
return 0;
}
当然,采用更高级的数据结构主席树可以做到在线回答。