BZOJ 2286 消耗战
//Twenty #include<cstdio> #include<cstdlib> #include<iostream> #include<algorithm> #include<cmath> #include<cstring> #include<queue> #include<vector> const int maxn=500000+299; typedef long long LL; using namespace std; int k,kk,ecnt,n,R[maxn],fir[maxn],nxt[maxn],to[maxn],val[maxn],vis[maxn],f[maxn][21]; int fr[maxn],nx[maxn],tt[maxn]; int dfs_cnt,ls[maxn],le[maxn],a[maxn],sta[maxn],mi[maxn],time_now; bool cmp(int a,int b){ return ls[a]<ls[b]; } void add1(int u,int v,int w){ nx[++ecnt]=fr[u]; fr[u]=ecnt; tt[ecnt]=v; val[ecnt]=w; } void add(int u,int v,int w){ nxt[++ecnt]=fir[u]; fir[u]=ecnt; to[ecnt]=v; val[ecnt]=w; } void dfs(int x,int fa){ /*(if(x==67) { int debug=1; }*/ f[x][0]=fa; R[x]=R[fa]+1; ls[x]=++dfs_cnt; for(int i=1;i<=20;i++) f[x][i]=f[f[x][i-1]][i-1]; //在dfs之前不是之后QAQ for(int i=fr[x];i;i=nx[i]) if(tt[i]!=fa) { if(x==1) mi[tt[i]]=val[i]; else mi[tt[i]]=min(mi[x],val[i]); dfs(tt[i],x); } le[x]=dfs_cnt; } int lca(int x,int y){ if(R[x]<R[y]) swap(x,y); for(int i=20;i>=0;i--){ if(R[x]==R[y]) break; if(R[x]-(1<<i)>=R[y]) x=f[x][i]; } if(x==y) return x; for(int i=20;i>=0;i--) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i]; return f[x][0]; } bool in(int fa,int x){ return ls[fa]<=ls[x]&&le[fa]>=le[x]; } LL tree_dp(int x){ LL res=0; for(int &i=fir[x];i;i=nxt[i]){ res+=tree_dp(to[i]); } if(vis[x]==time_now) return (LL)mi[x]; if(x==1) return res; else return min(res,(LL)mi[x]); } int main() { scanf("%d",&n); for(int i=1;i<n;i++){ int x,y,w; scanf("%d%d%d",&x,&y,&w); add1(x,y,w); add1(y,x,w); } dfs(1,0); int now; scanf("%d",&now); for(time_now=1;time_now<=now;time_now++){ scanf("%d",&k); kk=k; for(int i=1;i<=k;i++){ scanf("%d",&a[i]); vis[a[i]]=time_now; } sort(a+1,a+k+1,cmp); //k=unique(a+1,a+k+1)-(a+1); for(int i=2;i<=k;i++) { a[++kk]=lca(a[i],a[i-1]); } a[++kk]=1; sort(a+1,a+kk+1,cmp); kk=unique(a+1,a+kk+1)-(a+1); int top=0; ecnt=0; for(int i=1;i<=kk;i++){ while(top&&!in(sta[top],a[i])){ top--; } if(top){ add(sta[top],a[i],1); } sta[++top]=a[i]; } //mi[1]=1e9+7; 傻了吧 printf("%lld\n",tree_dp(1)); } return 0; }