设
dp[i]
表示
i
所在的子树里面所有的询问点与
虚树就是关键点以及它们两两LCA所形成的树。
我写的常数好大QAQ。
#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<cassert>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<sstream>
#include<climits>
#define X first
#define Y second
#define DB double
#define lc now<<1
#define rc now<<1|1
#define MP make_pair
#define LL long long
#define pb push_back
#define sqr(_) ((_)*(_))
#define int LL
#define INF 0x3f3f3f3f
#define pii pair<int,int>
#define pdd pair<DB,DB>
#define ull unsigned LL
#define DEBUG(...) fprintf(stderr,__VA_ARGS__)
using namespace std;
template<typename T>void Read(T& x)
{
x=0;int flag=0,sgn=1;char c;
while(c=getchar())
{
if(c=='-')sgn=-1;
else if(c>='0'&&c<='9')x*=10,x+=c-'0',flag=1;
else if(flag)break;
}
x*=sgn;
}
const int MAXN=1000010;
int n,m,dep[MAXN],dfn[MAXN],fa[MAXN][21],dfs_clock,g[MAXN][21];
int k,h[MAXN],st[MAXN],top=0,dp[MAXN],pa[MAXN],iskey[MAXN];
vector<int> A;
struct Tree{
int first[MAXN],next[MAXN],e,w[MAXN],to[MAXN];
Tree(){
memset(first,-1,sizeof(first));
e=0;
}
void add(int u,int v,int c)
{
next[e]=first[u];first[u]=e;to[e]=v;w[e]=c;++e;
next[e]=first[v];first[v]=e;to[e]=u;w[e]=c;++e;
}
void init()
{
e=0;
}
void DFS(int u,int f,int c)
{
if(u==1)
dep[u]=1;
else dep[u]=dep[f]+1;
dfn[u]=++dfs_clock;
fa[u][0]=f;
g[u][0]=c;
for(int i=1;i<21;i++)
{
fa[u][i]=fa[fa[u][i-1]][i-1];
g[u][i]=min(g[fa[u][i-1]][i-1],g[u][i-1]);
}
for(int i=first[u];i!=-1;i=next[i])
{
int v=to[i];
if(v==f)continue;
DFS(v,u,w[i]);
}
}
int LCA(int a,int b)
{
if(dep[a]<dep[b])swap(a,b);
for(int i=20;i>=0;i--)
if(dep[fa[a][i]]>=dep[b])
a=fa[a][i];
if(a==b)return a;
for(int i=20;i>=0;i--)
{
if(fa[a][i]&&fa[b][i]&&fa[a][i]!=fa[b][i])
a=fa[a][i],
b=fa[b][i];
}
return fa[a][0];
}
int getMin(int a,int b)
{
int res=INF;
for(int i=20;i>=0;i--)
{
if(fa[a][i]&&dep[fa[a][i]]>=dep[b])
{
res=min(res,g[a][i]);
a=fa[a][i];
}
}
return res;
}
}ori,vit;
bool cmp(const int& a,const int& b)
{
return dfn[a]<dfn[b];
}
void DP(int u,int f)
{
int tmp=0;
if(iskey[u])
return;
for(int i=vit.first[u];i!=-1;i=vit.next[i])
{
if(vit.to[i]==f)continue;
DP(vit.to[i],u);
tmp+=min(dp[vit.to[i]],vit.w[i]);
}
dp[u]=tmp;
}
void build()
{
top=0;
vit.init();
sort(h+1,h+k+1,cmp);
st[++top]=1;
for(int i=1;i<=k;i++)
{
iskey[h[i]]=1;
if(!top)
{
pa[h[i]]=1;
st[++top]=h[i];
continue;
}
int x=ori.LCA(h[i],st[top]);
A.pb(x);
A.pb(h[i]);
for(;dep[st[top]]>dep[x];top--)
if(dep[st[top-1]]<=dep[x])
pa[st[top]]=x;
if(st[top]!=x)
{
pa[x]=st[top];
st[++top]=x;
}
pa[h[i]]=x;
st[++top]=h[i];
}
sort(A.begin(),A.end());
A.erase(unique(A.begin(),A.end()),A.end());
for(int i=0;i<A.size();i++)
if(pa[A[i]])
{
vit.add(A[i],pa[A[i]],ori.getMin(A[i],pa[A[i]]));
//DEBUG("%d %d %d\n",A[i],pa[A[i]],ori.getMin(A[i],pa[A[i]]));
}
DP(1,0);
printf("%lld\n",dp[1]);
for(int i=0;i<A.size();i++)
{
vit.first[A[i]]=-1,pa[A[i]]=0;
dp[A[i]]=INF;
if(iskey[A[i]])
iskey[A[i]]=0;
}
A.clear();
}
main()
{
#ifndef ONLINE_JUDGE
freopen("consume.in","r",stdin);
freopen("consume.out","w",stdout);
#endif
memset(g,INF,sizeof(g));
memset(dp,INF,sizeof(dp));
Read(n);
for(int i=1;i<n;i++)
{
int a,b,c;
Read(a);Read(b);Read(c);
ori.add(a,b,c);
}
ori.DFS(1,0,0);
Read(m);
for(int i=1;i<=m;i++)
{
Read(k);
for(int j=1;j<=k;j++)
Read(h[j]);
build();
}
}