建虚树。
然后在虚树上求出每个点最多可以往外拓展多少距离(意思就是可以由其他点拓展到这个点之后继续拓展)。
那么按照这个定义我们可以得到:
一个点如果被覆盖,那么一定可以覆盖他的点在虚树上是一个联通块。
因为树上连通块满足点数-边数=1。
那么我们把每个点可以覆盖的点数相加,再枚举每条边,减去边的两个点都能覆盖的点数,
那么考虑一个被覆盖的点,能覆盖他的那个联通块会刚好计算了它一次。
那么这样就不重不漏的计算了所有被覆盖的点。
边上两个点共同覆盖的点可以通过计算他们的中点覆盖了多少个点来得到。
计算距离一个点
r
r
r 的点数可以建出点分树。
A C C o d e \mathcal AC \ Code AC Code
#include<bits/stdc++.h>
#define maxn 100005
#define lim 17
#define LL long long
#define rep(i,j,k) for(int i=(j),LIM=(k);i<=LIM;i++)
#define per(i,j,k) for(int i=(j),LIM=(k);i>=LIM;i--)
#define ADJ(i,u) for(int i=info[u],v;i;i=Prev[i])
using namespace std;
int n,lg[maxn<<1];
int info[maxn],Prev[maxn<<1],to[maxn<<1],cnt_e;
void Node(int u,int v){ Prev[++cnt_e]=info[u],info[u]=cnt_e,to[cnt_e]=v; }
//DCT
namespace DCT{
int sz[maxn],vis[maxn],fa[maxn],dep[maxn],dis[lim][maxn];
void dfs(int u,int ff,int tsz,int &mn,int &rt){
int mx = 0;sz[u] = 1;
ADJ(i,u) if((v=to[i])^ff && !vis[v])
dfs(v,u,tsz,mn,rt) , mx = max(mx , sz[v]) , sz[u] += sz[v];
if((mx = max(mx , tsz - sz[u])) < mn)
mn = mx , rt = u;
}
int Gert(int u,int tsz){
int mn,rt;
dfs(u,0,tsz,mn=0x3f3f3f3f,rt);
return rt;
}
vector<int>G[maxn],Gx[maxn];
void ser(int u,int ff,vector<int>&t,int d,int dp=-1){
sz[u] = 1;
while(t.size() <= d) t.push_back(0);
assert(d < t.size());
t[d] += u<=n;
if(~dp) assert(dp >= 0 && dp < lim) , dis[dp][u] = d;
ADJ(i,u) if((v=to[i])^ff && !vis[v])
ser(v,u,t,d+1,dp),sz[u] += sz[v];
}
void prc(vector<int>&t){ rep(i,1,t.size()-1) t[i]+=t[i-1]; }
void Solve(int u){
vis[u] = 1;
ser(u,0,G[u],0,dep[u]);
prc(G[u]);
ADJ(i,u) if(!vis[v=to[i]]){
int t = Gert(v,sz[v]);
ser(v,u,Gx[t],1) , fa[t] = u;
prc(Gx[t]);
dep[t] = dep[u] + 1;
Solve(t);
}
}
int qry(vector<int>&t,int d){
return t.empty() || d < 0 ? 0 : t[min(d,(int)t.size()-1)];
}
int qry(int u,int d){
int r = 0 , x = u;
for(int p=0;u;u=fa[p=u]) r += qry(G[u],d-dis[dep[u]][x]) - qry(Gx[p] , d-dis[dep[u]][x]) /*, printf("@%d %d %d %d %d\n",x,u,d,dis[dep[u]][x],r)*/;
return r;
}
}
//Tree and Lca
namespace Tree{
int dep[maxn],mn[lim+1][maxn<<1],fa[maxn],st[maxn],ed[maxn],pos[maxn],f[lim][maxn],tot,tim;
int ont[maxn],r[maxn];
void dfs0(int u,int ff){
mn[0][pos[u] = ++tim] = u , st[u] = ++tot;
dep[u] = dep[f[0][u] = ff] + 1;
ADJ(i,u) if((v=to[i])^ff)
dfs0(v,u) , mn[0][++tim] = u;
ed[u] = tot;
}
int Min(int u,int v){ return dep[u] < dep[v] ? u : v; }
int Lca(int u,int v){
u = pos[u] , v = pos[v];
if(u > v) swap(u,v);
int t = lg[v-u+1];
return Min(mn[t][u] , mn[t][v-(1<<t)+1]);
}
void init(){
dfs0(1,0);
rep(j,1,lim) rep(i,1,tim-(1<<j)+1) mn[j][i] = min(mn[j-1][i] , mn[j-1][i+(1<<j-1)]);
rep(j,1,lim-1) rep(i,1,2*n-1) f[j][i] = f[j-1][f[j-1][i]];
rep(j,2,tim) lg[j] = lg[j >> 1] + 1;
}
bool cmp(const int &u,const int &v){
return st[u] < st[v];
}
int getk(int u,int k){
rep(i,0,lim-1) if(k>>i&1)
u=f[i][u];
return u;
}
void Solve(){
int K;
scanf("%d",&K);
static int ar[maxn]={},x,pt[maxn]={},cnt=0,q[maxn]={},tp;
cnt = 0;
rep(i,1,K) scanf("%d%d",&ar[i],&x),r[ar[i]]=x<<1,ont[ar[i]]=1;
sort(ar+1,ar+1+K,cmp);
rep(i,1,K){
if(tp){
int t = Lca(ar[i],q[tp]),p=0;
for(;tp && dep[q[tp]] > dep[t];p = q[tp--]) if(p) fa[p] = q[tp];
if(q[tp] ^ t) pt[++cnt] = q[++tp] = t;
if(p) fa[p] = q[tp];
}
pt[++cnt] = q[++tp] = ar[i];
}
int p=0;
for(;tp;p=q[tp--]) if(p) fa[p]=q[tp];
sort(pt+1,pt+1+cnt,cmp);
rep(i,1,cnt) if(!ont[pt[i]]) r[pt[i]] = -1;
per(i,cnt,2) r[fa[pt[i]]] = max(r[fa[pt[i]]] , r[pt[i]] - dep[pt[i]] + dep[fa[pt[i]]]);
rep(i,2,cnt) r[pt[i]] = max(r[pt[i]] , r[fa[pt[i]]] - dep[pt[i]] + dep[fa[pt[i]]]);
LL ans = 0;
rep(i,1,cnt) if(r[pt[i]] >= 0) ans += DCT::qry(pt[i],r[pt[i]]);
rep(i,2,cnt) if(r[pt[i]] >= 0 && r[fa[pt[i]]] >= 0){
int z = getk(pt[i] , (dep[pt[i]] - dep[fa[pt[i]]] - r[fa[pt[i]]] + r[pt[i]]) >> 1);
ans -= DCT::qry(z,r[pt[i]] - dep[pt[i]] + dep[z]);
}
rep(i,1,cnt) ont[pt[i]] = 0;
printf("%lld\n",ans);
}
}
int main(){
scanf("%d",&n);
rep(i,1,n-1){
int u,v;scanf("%d%d",&u,&v);
Node(u,n+i),Node(n+i,u),Node(v,n+i),Node(n+i,v);
}
DCT::Solve(DCT::Gert(1,n*2-1));
Tree::init();
int Q;
scanf("%d",&Q);
for(;Q--;) Tree::Solve();
}
//1235432143214
/*
点分树
虚树
LCA
dfs序
树上k级祖先
*/