BZOJ 4057 状压DP

Posted 2020-09-07 SiriusRen

思路：

状压一下 就完了...

f[i]表示选了的集合为i

转移的时候判一判就好了..

//By SiriusRen
#include <cstdio>
#include <cstring>
using namespace std;
int cases,n,a[22][22],f[1024*1024],F;
int main(){
    scanf("%d",&cases);
    while(cases--){
        scanf("%d",&n);
        for(int i=0;i<(1<<n);i++)f[i]=0;
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                scanf("%d",&a[i][j]);
        f[0]=1,F=0;
        for(int i=0;i<(1<<n);i++)if(f[i])
            for(int j=0;j<n;j++)
                if(!(i&(1<<j))&&!f[i|(1<<j)]){
                    int res=0;
                    for(int k=0;k<n;k++)
                        if(!(i&(1<<k)))res-=a[j][k];
                    if(res<0)f[i|(1<<j)]=1;
                }
        for(int i=0;i<n;i++)if(f[((1<<n)-1)^(1<<i)])F=i+1;
        if(!F)puts("0");
        else for(int i=0;i<n;i++)if(f[((1<<n)-1)^(1<<i)])
                printf("%d%c",i+1,i==F-1?‘\n‘:‘ ‘);
    }
}