bzoj 2286: [Sdoi2011消耗战 虚树+树形dp

题意

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

分析

第一次写虚树。

显然如果只有一个询问的话只要树形dp即可。
若是多个询问的话我们就可以通过构建虚树来优化复杂度。
对于被选定的点对(x,y),设f=lca(x,y),若x和f之间没有被选定的点,那么x到f的路径就可以变为f->x。那么就可以把这棵树变成一棵只有k*2个节点的树,然后树形dp即可。
现在关键是如何构建虚树。
先把所有选定的点按照dfs序排序,假设y是x和y的lca则x必然可以省去。我们维护一个栈,对于新加入的一个点x,设栈顶元素为y,f=lca(x,y),那么y到f的路径上的点就可以被压缩,然后把f和x入栈即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 250005
#define ll long long
#define inf 0x7fffffff
using namespace std;

int n,m,cnt,last[N],ls[N],fa[N][20],logn,dep[N],dfn[N],tim,a[N],b[N],q[N];
ll f[N],mn[N];
struct edge{int to,next,w;}e[N*2];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void addedge1(int u,int v,int w)
{
    e[++cnt].to=v;e[cnt].w=w;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].w=w;e[cnt].next=last[v];last[v]=cnt;
}

void addedge2(int u,int v)
{
    if (u==v) return;
    e[++cnt].to=v;e[cnt].next=ls[u];ls[u]=cnt;
}

void dfs(int x)
{
    dfn[x]=++tim;
    for (int i=1;i<=logn;i++)
        fa[x][i]=fa[fa[x][i-1]][i-1];
    dep[x]=dep[fa[x][0]]+1;
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa[x][0]) continue;
        mn[e[i].to]=min(mn[x],(ll)e[i].w);
        fa[e[i].to][0]=x;
        dfs(e[i].to);
    }
}

int getlca(int x,int y)
{
    if (dep[x]<dep[y]) swap(x,y);
    for (int i=logn;i>=0;i--)
        if (dep[fa[x][i]]>=dep[y]) x=fa[x][i];
    if (x==y) return x;
    for (int i=logn;i>=0;i--)
        if (fa[x][i]!=fa[y][i])
        {
            x=fa[x][i];y=fa[y][i];
        }
    return fa[x][0];
}

void dp(int x)
{
    f[x]=mn[x];
    ll s=0;
    for (int i=ls[x];i;i=e[i].next)
    {
        dp(e[i].to);
        s+=f[e[i].to];
    }
    if (s<f[x]&&s) f[x]=s;
    ls[x]=0;
}

bool cmp(int a,int b)
{
    return dfn[a]<dfn[b];
}

void solve()
{
    int k=read();
    cnt=0;
    for (int i=1;i<=k;i++)
        a[i]=read();
    sort(a+1,a+k+1,cmp);
    int tot=1;b[tot]=a[1];
    for (int i=2;i<=k;i++)
        if (getlca(a[i],b[tot])!=b[tot])
            b[++tot]=a[i];
    int top=1;
    q[top]=1;
    for (int i=1;i<=tot;i++)
    {
        int now=b[i],f=getlca(now,q[top]);
        while (1)
        {
            if (dep[f]>=dep[q[top-1]])
            {
                addedge2(f,q[top]);top--;
                break;
            }
            addedge2(q[top-1],q[top]);top--;
        }
        if (q[top]!=f) q[++top]=f;
        if (q[top]!=now) q[++top]=now;
    }
    while (top>1)
    {
        addedge2(q[top-1],q[top]);
        top--;
    }
    dp(1);
    printf("%lld\n",f[1]);
}

int main()
{
    n=read();
    logn=log(n)/log(2);
    for (int i=1;i<n;i++)
    {
        int x=read(),y=read(),z=read();
        addedge1(x,y,z);
    }
    mn[1]=1e17;
    dfs(1);
    m=read();
    for (int i=1;i<=m;i++)
        solve();
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值