标签：amp 正整数 ret style algorithm struct content mil 个数
输出 q 行，每行对应一个询问，代表询问的答案。
题解：如果确定了a和b，那么c的个数就是a和b的公共子树大小-1，所以我们考虑所有b的贡献。
先考虑b不在a子树中的情况，那么b只能在a到根的路径上，这种情况显然可以直接计算。
那么考虑b在a子树中的情况，相当于b的深度和DFS序都要在一个范围内，这可以看成一个二维数点问题，用离线+树状数组轻松解决。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=300010;
typedef long long ll;
int n,m,cnt;
int to[maxn<<1],next[maxn<<1],head[maxn],dep[maxn],siz[maxn],fa[maxn],p1[maxn],p2[maxn],p[maxn];
ll s[maxn],ans[maxn];
struct node
{
int x,k,org;
}q[maxn];
inline int rd()
{
int ret=0,f=1; char gc=getchar();
while(gc<‘0‘||gc>‘9‘) {if(gc==‘-‘) f=-f; gc=getchar();}
while(gc>=‘0‘&&gc<=‘9‘) ret=ret*10+(gc^‘0‘),gc=getchar();
return ret*f;
}
inline void add(int a,int b)
{
to[cnt]=b,next[cnt]=head[a],head[a]=cnt++;
}
void dfs(int x)
{
siz[x]=1,p1[x]=++p2[0];
for(int i=head[x];i!=-1;i=next[i]) if(to[i]!=fa[x])
fa[to[i]]=x,dep[to[i]]=dep[x]+1,dfs(to[i]),siz[x]+=siz[to[i]];
p2[x]=p2[0];
}
bool cmpq(const node &a,const node &b)
{
return a.k<b.k;
}
bool cmpp(const int &a,const int &b)
{
return dep[a]<dep[b];
}
inline void updata(int x,int v)
{
for(int i=x;i<=n;i+=i&-i) s[i]+=v;
}
inline ll query(int x)
{
ll ret=0;
for(int i=x;i;i-=i&-i) ret+=s[i];
return ret;
}
int main()
{
n=rd(),m=rd();
int i,j,a,b;
memset(head,-1,sizeof(head));
for(i=1;i<n;i++) a=rd(),b=rd(),add(a,b),add(b,a);
dep[1]=1,dfs(1);
for(i=1;i<=m;i++)
q[i].x=rd(),q[i].k=rd(),ans[i]=(ll)(min(dep[q[i].x]-1,q[i].k)-1)*(siz[q[i].x]-1),q[i].k+=dep[q[i].x],q[i].org=i;
sort(q+1,q+m+1,cmpq);
for(i=1;i<=n;i++) p[i]=i;
sort(p+1,p+n+1,cmpp);
for(i=j=1;i<=m;i++)
{
for(;dep[p[j]]<=q[i].k&&j<=n;j++) updata(p1[p[j]],siz[p[j]]-1);
ans[q[i].org]+=query(p2[q[i].x])-query(p1[q[i].x]-1);
}
for(i=1;i<=m;i++) printf("%lld\n",ans[i]);
return 0;
}
标签：amp 正整数 ret style algorithm struct content mil 个数
原文地址：http://www.cnblogs.com/CQzhangyu/p/7787087.html
