题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=3572
算是细节题了吧。。
构出虚树,考虑z正反DP两次求出虚树中每个点最近的议事处是哪一个点,然后枚举虚树中的每一条边${X->Y}$,对于两点间在原树中的路径,显然存在一个分界点使得自分界点之上的所有点归最靠近$X$的议事处管辖,之下的点归最靠经$Y$的议事处管辖,还有一些没有考虑过的点,另外统计一下即可。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstdlib>
#include<cmath>
#include<cstring>
using namespace std;
#define maxn 310000
#define llg long long
#define yyj(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
llg n,m,f[maxn][],deep[maxn],dfn[maxn],cnt,size[maxn],s[maxn],top;
llg a[maxn],b[maxn],bel[maxn],c[maxn],rem[maxn],ans[maxn]; vector<llg>g[maxn]; bool cmp(llg x,llg y) {return dfn[x]<dfn[y];} void link(llg x,llg y){g[x].push_back(y);} void dfs(llg x,llg fa)
{
dfn[x]=++cnt;
llg w=g[x].size(),v;
size[x]=;
for (llg i=;i<w;i++)
{
v=g[x][i];
if (v==fa) continue;
deep[v]=deep[x]+;
f[v][]=x;
dfs(v,x);
size[x]+=size[v];
}
} void make_f()
{
for (llg j=;j<=;j++)
for (llg i=;i<=n;i++)
f[i][j]=f[f[i][j-]][j-];
} llg lca(llg x,llg y)
{
if (deep[x]<deep[y]) swap(x,y);
for (llg i=;i>=;i--)
if (deep[f[x][i]]>=deep[y])
x=f[x][i];
if (x==y) return x;
for (llg i=;i>=;i--)
if (f[x][i]!=f[y][i])
{
x=f[x][i];
y=f[y][i];
}
return f[x][];
} llg dis(llg x,llg y) {return deep[x]+deep[y]-*deep[lca(x,y)];} void init()
{
cin>>n;
llg x,y;
for (llg i=;i<n;i++)
{
scanf("%lld%lld",&x,&y);
g[x].push_back(y),g[y].push_back(x);
}
deep[]=;
dfs(,-);
make_f();
} void build_tree()
{
cnt=top=;
sort(a+,a+m+,cmp);
if (bel[]!=) s[++top]=;
for (llg i=;i<=m;i++)
{
llg x=a[i],fa=;
while (top)
{
fa=lca(s[top],x);
if (top> && deep[fa]<deep[s[top-]])
{
link(s[top-],s[top]);
top--;
}
else
if (deep[fa]<deep[s[top]]){ link(fa,s[top--]); break;}
else
break; }
if (s[top]!=fa) s[++top]=fa;
s[++top]=x;
}
while (--top) link(s[top],s[top+]);
} void dfs1(llg x,llg fa)
{
c[++cnt]=x;
rem[x]=size[x];
llg w=g[x].size(),v;
for (llg i=;i<w;i++)
{
v=g[x][i];
dfs1(v,x);
if (v==fa) continue;
if (!bel[v]) continue;//why
llg d1=dis(bel[v],x),d2=dis(bel[x],x);
if ((d1==d2 && bel[v]<bel[x]) || (d1<d2) || !bel[x]) bel[x]=bel[v];
}
} void dfs2(llg x,llg fa)
{
llg w=g[x].size(),v;
for (llg i=;i<w;i++)
{
v=g[x][i];
if (v==fa) continue;
llg d1=dis(bel[x],v),d2=dis(bel[v],v);
if ((d1==d2 && bel[v]>bel[x]) || (d1<d2) || !bel[v])
bel[v]=bel[x];
dfs2(v,x);
}
} void work(llg x,llg y)
{
llg son=y,mid=y;
for (llg i=;i>=;i--)
if (deep[f[son][i]]>deep[x]) son=f[son][i];
rem[x]-=size[son];
if (bel[x]==bel[y]) {ans[bel[x]]+=size[son]-size[y]; return;}
for (llg i=;i>=;i--)
{
llg ne=f[mid][i];
if (deep[ne]<=deep[x]) continue;
llg d1=dis(bel[x],ne),d2=dis(bel[y],ne);
if (d1>d2 || (d1==d2 && bel[y]<bel[x])) mid=ne;
}
ans[bel[x]]+=size[son]-size[mid];
ans[bel[y]]+=size[mid]-size[y];
} int main()
{
yyj("tree");
init();
llg T; cin>>T;
for (llg i=;i<=n;i++)
{
while (!g[i].empty()) g[i].pop_back();
//std::vector(c).swap(vec);
}
while (T--)
{
scanf("%lld",&m);
for (llg i=;i<=m;i++)
{
scanf("%lld",&a[i]);
b[i]=a[i]; bel[a[i]]=a[i];
}
build_tree();
dfs1(,-); dfs2(,-);
for (llg i=;i<=cnt;i++)
{
llg w=g[c[i]].size();
llg la=-;
for (llg j=;j<w;j++)
{
if (g[c[i]][j]==la) continue;
la=g[c[i]][j];
work(c[i],g[c[i]][j]);
}
}
for (llg i=;i<=cnt;i++) ans[bel[c[i]]]+=rem[c[i]];
for (llg i=;i<=m;i++) printf("%lld ",ans[b[i]]);
printf("\n");
for (llg i=;i<=cnt;i++)
{
ans[c[i]]=bel[c[i]]=rem[c[i]]=;
while (!g[c[i]].empty()) g[c[i]].pop_back();
}
}
return ;
}