BZOJ4057: [Cerc2012]Kingdoms

https://lydsy.com/JudgeOnline/problem.php?id=4057

分析:

• 沙茶狀壓
• 預處理出來\(num_{i,j}\)表示\(j\)狀態下\(i\)是否可能破產。
• 然後轉移即可。

程式碼:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
#define N 22
#define M (1<<20)
int n,d[N][N];
int num[N][M],p[M];
bool f[M];
int main() {
    int T;
    scanf("%d",&T);
    while(T--) {
        scanf("%d",&n);
        int i,j;
        for(i=0;i<n;i++) p[1<<i]=i+1;
        for(i=1;i<=n;i++) for(j=1;j<=n;j++) scanf("%d",&d[i][j]);
        int mask=(1<<n)-1;
        for(i=0;i<=mask;i++) f[i]=0;
        f[mask]=1;
        for(i=1;i<=n;i++) {
            for(j=1;j<=mask;j++) {
                int k=j&(-j);
                num[i][j]=num[i][j-k]+d[i][p[k]];
            }
        }
        for(i=mask;i;i--) if(f[i]) {
            for(j=1;j<=n;j++) if(i&(1<<(j-1))) {
                if(num[j][i]>0) {
                    f[i-(1<<(j-1))]=1;
                }
            }
        }
        int flg=0;
        for(i=1;i<=n;i++) if(f[1<<(i-1)]) {
            if(flg) printf(" %d",i);
            else flg=printf("%d",i);
        }
        if(!flg) printf("0");
        puts("");
    }   
}