虚树

将树压缩成若干个有用的点

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
using namespace std;
char buf[1<<20],*_=buf,*__=buf;
#define gc() (_==__&&(__=(_=buf)+fread(buf,1,1<<20,stdin),_==__)?EOF:*_++)
#define TT templateinline
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;
}

你可能感兴趣的:(模板,图论)