bzoj 3611(洛谷 4103) [Heoi2014]大工程——虚树

时间:2023-03-09 01:54:53
bzoj 3611(洛谷 4103) [Heoi2014]大工程——虚树

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=3611

   https://www.luogu.org/problemnew/show/P4103

虚树的边权就是两端点的 dep 差。

求最短边可以像求最长边一样记录一个 d2[ cr ] 表示 cr 到其子树里最近的关键点的距离。

可以用 vis[ ] 记录哪些点是关键点。在 DP 的 dfs 里清空数组真是舒服。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
int rdn()
{
int ret=;bool fx=;char ch=getchar();
while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
return fx?ret:-ret;
}
int Mx(int a,int b){return a>b?a:b;}
int Mn(int a,int b){return a<b?a:b;}
const int N=1e6+,K=;
int n,hd[N],xnt,to[N<<],nxt[N<<];
int dep[N],dfn[N],tim,bin[K],pre[N][K];
bool cmp(int x,int y){return dfn[x]<dfn[y];}
void add(int x,int y){to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;}
void dfs(int cr,int fa)
{
dfn[cr]=++tim;
dep[cr]=dep[fa]+; pre[cr][]=fa;
for(int t=;bin[t]<=dep[cr];t++)
pre[cr][t]=pre[pre[cr][t-]][t-];
for(int i=hd[cr],v;i;i=nxt[i])
if((v=to[i])!=fa) dfs(v,cr);
}
int get_lca(int x,int y)
{
if(dep[x]<dep[y])swap(x,y);
int d=dep[x]-dep[y];
for(int t=;bin[t]<=d;t++)
if(d&bin[t])x=pre[x][t];
if(x==y)return x;
for(int t=;t>=;t--)
if(pre[x][t]!=pre[y][t])
x=pre[x][t],y=pre[y][t];
return pre[x][];
}
namespace Tr{
int hd[N],xnt,to[N],nxt[N],len,mn; ll sm;
int p[N],tot,sta[N],top,siz[N],dis[N],d2[N];
bool vis[N];
void add(int x,int y){to[++xnt]=y;nxt[xnt]=hd[x];hd[x]=xnt;}
void dfs(int cr)
{
dis[cr]=; siz[cr]=vis[cr]; d2[cr]=(vis[cr]?:N);
for(int i=hd[cr],v;i;i=nxt[i])
{
dfs(v=to[i]); int w=dep[v]-dep[cr];
sm+=(ll)w*siz[v]*(tot-siz[v]);
len=Mx(len,dis[cr]+dis[v]+w); dis[cr]=Mx(dis[cr],dis[v]+w);
mn=Mn(mn,d2[cr]+d2[v]+w); d2[cr]=Mn(d2[cr],d2[v]+w);
siz[cr]+=siz[v];
}
vis[cr]=; hd[cr]=;
}
void solve()
{
tot=rdn(); xnt=;//xnt
for(int i=;i<=tot;i++)p[i]=rdn(),vis[p[i]]=;
sort(p+,p+tot+,cmp); sta[top=]=p[];//cmp!!!
for(int i=;i<=tot;i++)
{
int u=p[i], lca=get_lca(u,sta[top]);
while(top&&dfn[lca]<dfn[sta[top]])
{
if(dfn[sta[top-]]<dfn[lca])
add(lca,sta[top]);
else add(sta[top-],sta[top]);
top--;
}
if(sta[top]!=lca)sta[++top]=lca;
sta[++top]=u;
}
for(int i=;i<top;i++)add(sta[i],sta[i+]);
sm=; mn=N; len=;
dfs(sta[]); printf("%lld %d %d\n",sm,mn,len);
}
}
int main()
{
bin[]=;for(int i=;i<=;i++)bin[i]=bin[i-]<<;
n=rdn();
for(int i=,u,v;i<n;i++)
u=rdn(),v=rdn(),add(u,v),add(v,u);
dfs(,); int Q=rdn();
while(Q--)Tr::solve();
return ;
}