loj6009「网络流 24 题」软件补丁(状态压缩+spfa)

首先n很小,我们可以压缩状态来表示每个问题的修复情况,0表示未修复,1表示修好了,则我们要求的就是从0到bin[n]-1的最短路。则我们一共有O(2^n)个点,然后相当于m个转移,求一下最短路就好啦。复杂度 O(2nm)

#include 
using namespace std;
#define ll long long
#define N 1100000
#define inf 0x3f3f3f3f
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,bin[25],b1[110],b2[110],f1[110],f2[110],val[110],dis[N];
char s[30];
bool inq[N];
inline void spfa(){
    queue<int>q;memset(dis,inf,sizeof(dis));
    q.push(0);dis[0]=0;inq[0]=1;
    while(!q.empty()){
        int x=q.front();q.pop();inq[x]=0;
        for(int i=1;i<=m;++i){
            if((x|b1[i])!=bin[n]-1||x&b2[i]) continue;
            int y=x|f1[i];y&=f2[i];
            if(dis[x]+val[i]if(!inq[y]) inq[y]=1,q.push(y);
            }
        }
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();bin[0]=1;
    for(int i=1;i<=n;++i) bin[i]=bin[i-1]<<1;
    for(int i=1;i<=m;++i){
        val[i]=read();scanf("%s",s+1);
        for(int j=1;j<=n;++j){
            if(s[j]!='-') b1[i]|=bin[j-1];
            if(s[j]=='+') b2[i]|=bin[j-1];
        }scanf("%s",s+1);
        for(int j=1;j<=n;++j){
            if(s[j]=='-') f1[i]|=bin[j-1];
            if(s[j]!='+') f2[i]|=bin[j-1];
        }
    }spfa();
    if(dis[bin[n]-1]==inf) puts("0");
    else printf("%d\n",dis[bin[n]-1]);
    return 0;
}

你可能感兴趣的:(其他oj,状压dp,最短路)