P2495 [SDOI2011]消耗战

最新推荐文章于 2025-07-09 20:32:09 发布

weixin_30407613

最新推荐文章于 2025-07-09 20:32:09 发布

阅读量57

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/xzz_233/p/8654200.html

P2495 [SDOI2011]消耗战

虚树入门题目？？

先链接一波：https://www.cnblogs.com/zzqsblog/p/5560645.html

可以每次做一个树形dp，复杂度\(O(n^2)\)

但是发现每次的dp有很多一样，所以造成很多无用的转移

所以说要尝试搞出所有有用的点来

有用的点也就是给的点+两两之间的lca（在lca需要合并信息了）

这个要算出来并不是平方级别的

直接把给出的点按照dfs序从小到大sort一遍去重即可，证明见下面

然后现在知道了虚树上的点，如何求出虚树？？？

只需要维护一个栈，按dfs序从小到大加点，加点时若栈顶不是该点的祖先就弹栈，然后连接栈顶和这个点就行了。

虚树的边权根据题目而定。比如本题，鸽（注：没错别字）一条边就相当于鸽掉一条链，所以链的权值等于链上边权取min。

然后愉快的dp就星了。

~~倍增好麻烦。。~~

记得开longlong.

// It is made by XZZ
#include<cstdio>
#include<algorithm>
#include<set>
#define il inline
#define rg register
#define vd void
#define sta static
typedef long long ll;
using namespace std;
il int gi(){
    rg int x=0,f=1;rg char ch=getchar();
    while(ch<'0'||ch>'9')f=ch=='-'?-1:f,ch=getchar();
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return x*f;
}
const int maxn=250010;
int fir[maxn],nxt[maxn<<1],dis[maxn<<1],w[maxn<<1],id;
il vd link(int a,int b,int c){nxt[++id]=fir[a],fir[a]=id,dis[id]=b,w[id]=c;}
int dfn[maxn],dep[maxn],f[18][maxn],g[18][maxn];
il vd dfs(int x){
    dfn[x]=++dfn[0];
    for(int i=fir[x];i;i=nxt[i]){
        if(f[0][x]==dis[i])continue;
        dep[dis[i]]=dep[x]+1;
        f[0][dis[i]]=x;
        g[0][dis[i]]=w[i];
        dfs(dis[i]);
    }
}
il int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    for(rg int i=17;~i;--i)if(dep[f[i][x]]>=dep[y])x=f[i][x];
    for(rg int i=17;~i;--i)if(f[i][x]!=f[i][y])x=f[i][x],y=f[i][y];
    if(x!=y)x=f[0][x];
    return x;
}
il pair<int,int> jump(int x,int k){
    int ret=1e9;
    for(rg int i=17;~i;--i)if(k&(1<<i))ret=min(ret,g[i][x]),x=f[i][x];
    return make_pair(x,ret);
}
int s[maxn<<1],stk[maxn],top;
il int cmp(const int&a,const int&b){return dfn[a]<dfn[b];}
int fir_[maxn],nxt_[maxn],dis_[maxn],w_[maxn],id_;
ll F[maxn];
il vd link_(int a,int b,int c){nxt_[++id_]=fir_[a],fir_[a]=id_,dis_[id_]=b,w_[id_]=c;}
int fafa[maxn];
il vd dp(int x,ll lst=1e18){
    F[x]=lst;ll sum=0;
    if(fafa[s[x]]==fafa[0])return;
    for(int i=fir_[x];i;i=nxt_[i]){
        int y=dis_[i],z=w_[i];
        dp(y,z);
        sum+=F[y];
    }
    if(sum<F[x])F[x]=sum;
}
main(){
#ifdef xzz
    freopen("2495.in","r",stdin);
    freopen("2495.out","w",stdout);
#endif
    int n=gi(),a,b,c;
    for(rg int i=1;i<n;++i)a=gi(),b=gi(),c=gi(),link(a,b,c),link(b,a,c);
    dep[1]=1;dfs(1);
    for(rg int i=1;i<18;++i)
        for(rg int j=1;j<=n;++j){
            f[i][j]=f[i-1][f[i-1][j]];
            g[i][j]=min(g[i-1][j],g[i-1][f[i-1][j]]);
        }
    int m,q=gi();
    for(rg int yyb=1;yyb<=q;++yyb){
        m=gi();
        for(rg int i=1;i<=m;++i)s[i]=gi(),fafa[s[i]]=yyb;
        fafa[0]=yyb;
        sort(s+1,s+m+1,cmp);
        for(rg int i=1;i<m;++i)s[i+m]=lca(s[i],s[i+1]);
        s[m+m]=1;
        m+=m;
        sort(s+1,s+m+1,cmp);
        m=unique(s+1,s+m+1)-s-1;
        top=0;
        id_=0;
        for(rg int i=1;i<=m;++i)fir_[i]=0;
        for(rg int i=1;i<=m;++i){
            while(top&&(dep[s[stk[top]]]>=dep[s[i]]||jump(s[i],dep[s[i]]-dep[s[stk[top]]]).first!=s[stk[top]]))--top;
            if(top)link_(stk[top],i,jump(s[i],dep[s[i]]-dep[s[stk[top]]]).second);
            stk[++top]=i;
        }
        dp(1);
        printf("%lld\n",F[1]);
    }
    return 0;
}

证明：