bzoj3058 四叶草魔杖(状压dp+最小生成树)

正解是最小生成树+状压dp。暴力的求出每个集合的和,若为0,则对这个集合做最小生成树,它的mst就是使得这个集合的点都合法的代价。然后如果s1,s2两个集合不重复,且都可以合法,则s1与s2的并集也可以由他俩的代价和实现合法,状压转移一下取最小值即可。(我本以为这样的转移会T,但是想一想常数应该极小。)

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 20
inline 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;
}
int n,m,a[N],bin[N],fa[N],sum[100000],dp[100000];//dp[s]-s状态均合法的最小代价 
struct edge{
    int x,y,val;
}e[N*N>>1];
vector<int>b;
inline bool cmp(edge x,edge y){return x.val<y.val;}
inline int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
inline int Kruskal(int s){
    int tot=0,res=0;
    for(int i=0;i<n;++i)
        if(s&bin[i]) fa[i]=i,tot++;
    for(int i=1;i<=m;++i){
        if(s&bin[e[i].x]&&s&bin[e[i].y]){
            int xx=find(e[i].x),yy=find(e[i].y);
            if(xx!=yy) fa[xx]=yy,res+=e[i].val,tot--;
        }
    }if(tot!=1) return inf;return res;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();bin[0]=1;memset(dp,inf,sizeof(dp));
    for(int i=0;i<n;++i) a[i]=read(),bin[i+1]=bin[i]<<1;
    for(int i=1;i<=m;++i)
        e[i].x=read(),e[i].y=read(),e[i].val=read();
    sort(e+1,e+m+1,cmp);
    for(int s=1;s<=bin[n]-1;++s)
        for(int i=0;i<n;++i) if(s&bin[i]) sum[s]+=a[i];
    for(int s=1;s<=bin[n]-1;++s)
        if(!sum[s]) dp[s]=Kruskal(s);//和为0,算代价 
    for(int s1=1;s1<=bin[n]-1;++s1){
        if(dp[s1]==inf) continue;
        for(int s2=s1+1;s2<=bin[n]-1;++s2){
            if(dp[s2]==inf||s2&s1) continue;
            dp[s2|s1]=min(dp[s2|s1],dp[s1]+dp[s2]);
        }
    }if(dp[bin[n]-1]==inf) puts("Impossible");
    else printf("%d\n",dp[bin[n]-1]);
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值