首页 > 代码库 > 消耗战(bzoj 2286)

消耗战(bzoj 2286)

Description

在一场战争中,战场由n个岛屿和n-1个桥梁组成,保证每两个岛屿间有且仅有一条路径可达。现在,我军已经侦查到敌军的总部在编号为1的岛屿,而且他们已经没有足够多的能源维系战斗,我军胜利在望。已知在其他k个岛屿上有丰富能源,为了防止敌军获取能源,我军的任务是炸毁一些桥梁,使得敌军不能到达任何能源丰富的岛屿。由于不同桥梁的材质和结构不同,所以炸毁不同的桥梁有不同的代价,我军希望在满足目标的同时使得总代价最小。
侦查部门还发现,敌军有一台神秘机器。即使我军切断所有能源之后,他们也可以用那台机器。机器产生的效果不仅仅会修复所有我军炸毁的桥梁,而且会重新随机资源分布(但可以保证的是,资源不会分布到1号岛屿上)。不过侦查部门还发现了这台机器只能够使用m次,所以我们只需要把每次任务完成即可。

Input

第一行一个整数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,表示资源丰富岛屿的编号。

 

Output

输出有m行,分别代表每次任务的最小代价。

 

 

Sample Input

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

Sample Output

12
32
22

HINT

 

 对于100%的数据,2<=n<=250000,m>=1,sigma(ki)<=500000,1<=ki<=n-1

/*
    虚树+树形DP
    设f[i]表示除掉以i为根的子树中所有关键点的最小花费,g[i]表示i是否是关键点。
    f[i]=Σmin(g[e[i].v]?inf:f[e[i].v],e[i].w)
    但是DP的复杂度是O(n)的,我们考虑每次的关键点是很少的,
    我们可以只把这些关键点和对答案有用的点(lca)提出来,建一棵虚树,在虚树上DP。
    如何建立虚树呢?用一个单调栈。
    将关键点按照dfs序排序,栈中的元素形成一条由根节点出发的链,初始栈中只有根节点。
    每次加入一个节点,求出节点与栈顶的LCA,将栈中所有深度大于LCA的节点全都弹掉。 
    然后将LCA和该节点入栈,注意有些重复的情况要考虑。 
    在这个模拟的DFS过程中顺便把DP做了即可。 
*/
#include<cstdio>
#include<iostream>
#include<algorithm>
#define N 250010
#define lon long long
#define inf 1000000000
using namespace std;
int head[N],a[N],n,m,tot,cnt;
int pos[N],dep[N],fa[N][20],dis[N][20];
int g[N],stack[N];
lon f[N];
struct node{int v,w,pre;}e[N*2];
void add(int x,int y,int z){
    e[++tot].v=y;
    e[tot].w=z;
    e[tot].pre=head[x];
    head[x]=tot;
}
void dfs(int x){
    pos[x]=++cnt;
    dep[x]=dep[fa[x][0]]+1;
    for(int i=head[x];i;i=e[i].pre)
        if(e[i].v!=fa[x][0]){
            fa[e[i].v][0]=x;
            dis[e[i].v][0]=e[i].w;
            dfs(e[i].v);
        }
}
bool cmp(int x,int y){
    return pos[x]<pos[y];
}
int LCA(int x,int y){
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=19;i>=0;i--)
        if(dep[fa[x][i]]>=dep[y])
            x=fa[x][i];
    if(x==y) return x;
    for(int i=19;i>=0;i--)
        if(fa[x][i]!=fa[y][i])
            x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int calc(int x,int y){
    int re=inf;
    for(int i=19;i>=0;i--)
        if(dep[fa[x][i]]>=dep[y])
            re=min(re,dis[x][i]),x=fa[x][i];
    return re;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
        add(y,x,z);
    }
    dfs(1);
    for(int j=1;j<=19;j++)
        for(int i=1;i<=n;i++){
            fa[i][j]=fa[fa[i][j-1]][j-1];
            dis[i][j]=min(dis[i][j-1],dis[fa[i][j-1]][j-1]);
        }
    int top=0;
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        int k;scanf("%d",&k);
        for(int j=1;j<=k;j++) scanf("%d",&a[j]);
        sort(a+1,a+k+1,cmp);
        stack[++top]=1;
        f[1]=0;g[1]=0;
        for(int j=1;j<=k;j++){
            int lca=LCA(stack[top],a[j]);
            while(dep[stack[top]]>dep[lca]){
                if(dep[stack[top-1]]<=dep[lca]){
                    int tmp=min(g[top]?inf:f[top],(lon)calc(stack[top],lca));
                    stack[top--]=0;
                    if(lca!=stack[top]){
                        stack[++top]=lca;
                        f[top]=0;g[top]=0;
                    }
                    f[top]+=tmp;
                    break;
                }
                else {
                    f[top-1]+=min(g[top]?inf:f[top],(lon)calc(stack[top],stack[top-1]));
                    stack[top--]=0;
                }
            }
            if(stack[top]!=a[j]){
                stack[++top]=a[j];
                f[top]=0;
            }
            g[top]=1;
        }
        while(top>1){
            f[top-1]+=min(g[top]?inf:f[top],(lon)calc(stack[top],stack[top-1]));
            stack[top--]=0;
        }
        printf("%lld\n",f[top--]);
    }
    return 0;
}

 

消耗战(bzoj 2286)