[bzoj2286] [Sdoi2011消耗战

时间:2023-03-10 00:12:41
[bzoj2286] [Sdoi2011消耗战

  还是虚树恩。。模板都能打挂QAQ

  先在原树上预处理出mndis[i],表示根节点到节点i 路径上边权的最小值(就是断开i与根的联系的最小花费)

  建完虚树在虚树上跑树形DP。。f[i]表示断开  i 所在子树内所有有资源的节点  与根节点的联系的最小花费。

  若i 节点没资源:f[i]=min( mndis[i] , sigma(f[j]) ),(j是i的儿子,且j所在子树内有有资源的节点)。

  若i 节点有资源:f[i]=mndis[i]。。。

  链剖大法好。。

 #include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn=;
const int inf=;
struct zs{
int too,pre,dis;
}e[maxn<<];
struct zs1{
int too,pre;
}e1[maxn<<];
int last[maxn],tot,last1[maxn],tot1;
int mndis[maxn],sz[maxn],st[maxn],top;
int dfn[maxn],dep[maxn],bel[maxn],fa[maxn],size[maxn],tim;
int have[maxn],poi[maxn];
int i,j,k,K,n,m,a,b,c;
ll f[maxn]; int ra;char rx;
inline int read(){
rx=getchar(),ra=;
while(rx<''||rx>'')rx=getchar();
while(rx>=''&&rx<='')ra*=,ra+=rx-,rx=getchar();return ra;
} inline void insert(int a,int b,int c){
e[++tot].too=b,e[tot].dis=c,e[tot].pre=last[a],last[a]=tot;
e[++tot].too=a,e[tot].dis=c,e[tot].pre=last[b],last[b]=tot;
}
inline void ins(int a,int b){
// printf(" %d-->%d\n",a,b);
e1[++tot1].too=b,e1[tot1].pre=last1[a],last1[a]=tot1;
} void dfs(int x){
dep[x]=dep[fa[x]]+,size[x]=;
for(int i=last[x];i;i=e[i].pre)
if(e[i].too!=fa[x])
fa[e[i].too]=x,
mndis[e[i].too]=mndis[x]<e[i].dis?mndis[x]:e[i].dis,
dfs(e[i].too),
size[x]+=size[e[i].too];
}
void dfs2(int x,int chain){
int i,mxpos=;bel[x]=chain,dfn[x]=++tim;
for(i=last[x];i;i=e[i].pre)if(e[i].too!=fa[x]&&size[e[i].too]>=size[mxpos])mxpos=e[i].too;
if(!mxpos)return;
dfs2(mxpos,chain);
for(i=last[x];i;i=e[i].pre)if(e[i].too!=fa[x]&&e[i].too!=mxpos)dfs2(e[i].too,e[i].too);
}
inline int getlca(int a,int b){
if(dep[bel[a]]<dep[bel[a]])swap(a,b);
while(bel[a]!=bel[b]){
a=fa[bel[a]];
if(dep[bel[a]]<dep[bel[b]])swap(a,b);
}
return dep[a]<dep[b]?a:b;
} inline void build(){
register int i,lca;
st[top=]=;
for(i=;i<=K;i++){
lca=getlca(poi[i],st[top]);
while(dfn[st[top]]>dfn[lca])
if(dfn[st[--top]]<=dfn[lca]){
ins(lca,st[top+]);
if(st[top]!=lca)st[++top]=lca;
}else ins(st[top],st[top+]);//,puts("");
st[++top]=poi[i];
}
while(top>)ins(st[top-],st[top]),top--;
}
void query(int x){
int i,to;
f[x]=sz[x]=;
if(have[x]==m){f[x]=mndis[x],sz[x]=;return;}
for(to=e1[i=last1[x]].too;i;to=e1[i=e1[i].pre].too){
query(to);
if(sz[to])f[x]+=f[to],sz[x]+=sz[to];
}
if(x!=&&mndis[x]<f[x])f[x]=mndis[x];
} bool cmp(int a,int b){return dfn[a]<dfn[b];}
int main(){
n=read();
for(i=;i<n;i++)a=read(),b=read(),c=read(),insert(a,b,c);
mndis[]=inf,dfs(),dfs2(,);
// for(i=1;i<=n;i++)printf(" %d\n",mndis[i]); for(m=read();m;m--){
K=read();for(i=;i<=K;i++)have[poi[i]=read()]=m; sort(poi+,poi++K,cmp),
build(),
query(),
printf("%lld\n",f[]); if(m>){for(i=tot1;i;i--)last1[e1[i].too]=;last1[]=tot1=;}
}
return ;
}