将树压缩成若干个有用的点
for(int i=0;i<k;++i){
int f=lca(stk[r],h[i]);
while(r&&id[f]<id[stk[r-1]]){
g1.add(stk[r-1],stk[r],get(stk[r],stk[r-1]));
--r;
}
if(f^stk[r])g1.add(f,stk[r],get(stk[r],f)),--r;
if(f^stk[r])stk[++r]=f;
stk[++r]=h[i];
]
P2495 [SDOI2011]消耗战
在一场战争中,战场由n个岛屿和n-1个桥梁组成,保证每两个岛屿间有且仅有一条路径可达。现在,我军已经侦查到敌军的总部在编号为1的岛屿,而且他们已经没有足够多的能源维系战斗,我军胜利在望。已知在其他k个岛屿上有丰富能源,为了防止敌军获取能源,我军的任务是炸毁一些桥梁,使得敌军不能到达任何能源丰富的岛屿。由于不同桥梁的材质和结构不同,所以炸毁不同的桥梁有不同的代价,我军希望在满足目标的同时使得总代价最小。
侦查部门还发现,敌军有一台神秘机器。即使我军切断所有能源之后,他们也可以用那台机器。机器产生的效果不仅仅会修复所有我军炸毁的桥梁,而且会重新随机资源分布(但可以保证的是,资源不会分布到1号岛屿上)。不过侦查部门还发现了这台机器只能够使用m次,所以我们只需要把每次任务完成即可。
输入格式
第一行一个整数n,代表岛屿数量。
接下来n-1行,每行三个整数u,v,w,代表u号岛屿和v号岛屿由一条代价为c的桥梁直接相连,保证1<=u,v<=n且1<=c<=100000。
第n+1行,一个整数m,代表敌方机器能使用的次数。
接下来m行,每行一个整数ki,代表第i次后,有ki个岛屿资源丰富,接下来k个整数h1,h2,…hk,表示资源丰富岛屿的编号。
输入
10
1 5 13
1 9 6
2 1 19
2 4 8
2 3 91
5 6 8
7 5 4
7 8 31
10 7 9
3
2 10 6
4 5 7 8 3
3 9 4 6
输出
12
32
22
#include<bits/stdc++.h>
using namespace std;
char buf[1<<20],*_=buf,*__=buf;
#define gc() (_==__&&(__=(_=buf)+fread(buf,1,1<<20,stdin),_==__)?EOF:*_++)
#define TT template<class T>inline
TT bool read(T &x){
x=0;char c=gc();bool f=0;
while(c<48||c>57){if(c==EOF)return 0;f^=(c=='-'),c=gc();}
while(47<c&&c<58)x=(x<<3)+(x<<1)+(c^48),c=gc();
if(f)x=-x;return 1;
}
TT bool read(T&a,T&b){return read(a)&&read(b);}
TT bool read(T&a,T&b,T&c){return read(a)&&read(b)&&read(c);}
typedef long long ll;
const ll MAXN=25e5+8,mod=1e9+7,inf=0x3f3f3f3f;
struct G{
struct E{int y,v,nt;}e[MAXN*2];
int head[MAXN],cnt;
inline void add(int x,int y,int v){//x->y
e[++cnt].y=y;e[cnt].v=v;
e[cnt].nt=head[x];head[x]=cnt;
}
}g0,g1;
int fa[MAXN][19],dep[MAXN],cost[MAXN][19],id[MAXN],idcnt;
void predfs(int x,int f){
id[x]=++idcnt;
fa[x][0]=f,dep[x]=dep[f]+1;
for(int i=1;i<=18;++i){
fa[x][i]=fa[fa[x][i-1]][i-1];
cost[x][i]=min(cost[x][i-1],cost[fa[x][i-1]][i-1]);
}
for(int i=g0.head[x],y;i;i=g0.e[i].nt){
y=g0.e[i].y;
if(dep[y])continue;
cost[y][0]=g0.e[i].v;
predfs(y,x);
}
}
int lca(int x,int y){
if(dep[x]<dep[y])swap(x,y);
for(int d=dep[x]-dep[y],i=18;d;--i)
if(d>=1<<i)d-=1<<i,x=fa[x][i];
if(x==y)return x;
for(int i=18;i>=0;--i)
if(fa[x][i]^fa[y][i])x=fa[x][i],y=fa[y][i];
return fa[x][0];
}
int get(int x,int f){
int res=inf;
for(int d=dep[x]-dep[f],i=18;d;--i)
if(d>=1<<i)d-=1<<i,res=min(res,cost[x][i]),x=fa[x][i];
return res;
}
int n,m,k;
bool cmp(const int a,const int b){return id[a]<id[b];}
int h[MAXN];
int stk[MAXN],r;
bool ish[MAXN];
ll solve(int x){
ll res=0,tmp;
for(int i=g1.head[x],y;i;i=g1.e[i].nt){
y=g1.e[i].y;
if(ish[y])res+=g1.e[i].v;
else tmp=solve(y),res+=min(tmp,(ll)g1.e[i].v);
}
return res;
}
int head[MAXN],rr;
int main() {
read(n);
for(int i=1,x,y,v;i<n;++i){
read(x,y,v);
g0.add(x,y,v);
g0.add(y,x,v);
}
predfs(1,0);
read(m);
while(m--){
read(k);
for(int i=0;i<k;++i)read(h[i]),ish[h[i]]=1;
sort(h,h+k,cmp);
stk[r=0]=1;
head[rr=0]=1;
for(int i=0;i<k;++i){
int f=lca(stk[r],h[i]);
while(r&&id[f]<id[stk[r-1]]){
g1.add(stk[r-1],stk[r],get(stk[r],stk[r-1]));
head[++rr]=stk[--r];
}
if(f^stk[r])g1.add(f,stk[r],get(stk[r],f)),head[++rr]=stk[--r];
if(f^stk[r])stk[++r]=f;
stk[++r]=h[i];
}
while(r>0){
int x=stk[r],f=stk[r-1],v=get(x,f);
g1.add(f,x,v);
head[++rr]=stk[--r];
}
printf("%lld\n",solve(1));
for(int i=0;i<k;++i)ish[h[i]]=0;
for(int i=0;i<=rr;++i)g1.head[head[i]]=0;
}
return 0;
}