POJ 1321 棋盘问题

dp[n行][位压缩数][剩下k个棋子]

不用n行滚动数组也可。

更新与之前所有行不冲突的情况以及不放棋子情况，最后对dp[n][all][0]求和。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
int n, k;
const int maxn = 1 << 8 | 1;
int dp[10][maxn][10];
char g[10][10];
int main()
{
    while(scanf("%d%d", &n, &k), n != -1 || k != -1)
    {
        memset(dp, 0, sizeof(dp));
        for(int i = 1; i <= n; i ++)
            scanf("%s", g[i]);
        int M = 1 << n;
        dp[0][0][k] = 1;
        for(int i = 1; i <= n; i ++)
        {
            for(int p = 0; p < M; p ++)
            {
                for(int j = 0; j < n; j ++)
                {
                    if((~p & (1 << j)) && g[i][j] == '#')
                    {
                            for(int q = 1; q <= k; q ++)
                                dp[i][p | (1 << j)][q - 1] += dp[i - 1][p][q];
                    }
                }
                for(int q = 0; q <= k; q ++)
                    dp[i][p][q] += dp[i - 1][p][q];
            }
        }
        int ans = 0;
        for(int i = 0; i < M; i ++)
            ans += dp[n][i][0];
        printf("%d\n", ans);
    }
    return 0;
}