sgu 223 Little Kings 状态压缩dp

  下一行的状态只受前面一行的影响。

  先预处理出在一行中互相无冲突的状态，用数字来表示。比如 一行为3列，那没有冲突的状态就有：(1代表国王，0代表无）

0 0 0 --------> 0

0 1 0  -------->2

0 0 1  --------> 4

1 0 0 --------> 1

1 0 1  --------> 5

....

这样列处理好了，就只用以行为阶段进行dp状态方程转移。

定义dp[ u] [row] [king]  king代表国王个数，row代表行数，u代表此行用的是那个状态（就是上面所示的数

所以 dp[u][row][king] = sum( dp[i][row-1][king - cnt(i)])   

i  是下一行的状态所表示的数字， cnt（i ）是此状态有多少个国王

#include <stdio.h>
#include <iostream>
#include <string.h>
#include <cmath>
#include <vector>
#include <set>
#include <map>
#include <algorithm>
#include <string>
using namespace std;
typedef pair<int,int> pii;
typedef long long ll;
typedef double dl;
typedef vector<pii> vii;
typedef set<pii> sii;
/*sim_operarion*/
#define sfint(x) scanf("%d",&x)
#define sfint2(x,y) scanf("%d%d",&x,&y)
#define sfint3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define sfstr(c) scanf("%s",c)
#define sfdl(x) scanf("%lf",&x)
#define pfint(x) printf("%d\n",x)
#define fr(i,s,n) for(int i=s;i<n;++i)
#define _fr(i,n,s) for(int i=n-1;i>=s;--i)
#define cl(a) memset(a,0,sizeof(a))

int n,k;
int tot;
int canuse[200];

ll dp[200][11][101];

ll dfs(int u,int row,int king){
    if (dp[u][row][king] != -1) return dp[u][row][king];
    int cnt = 0;
    fr(i , 0 , n) if (1<<i &    canuse[u]) ++cnt;
    if (row == 1) {
        if (cnt == king) return dp[u][row][king] = 1;
        else return dp[u][row][king] = 0;
    }

    dp[u][row][king] = 0;
    fr(i , 0 ,tot){
        if (!( (canuse[i]&canuse[u]) || ((canuse[i]<<1)&canuse[u]) || ((canuse[i]>>1)&canuse[u]) )){
            if (king-cnt >=0){
                ll tmp = dfs(i,row-1,king-cnt);
                dp[u][row][king] += tmp;
            }
        }
    }
    return dp[u][row][king];
}


int main() {
    while(cin>>n>>k){
        tot = 0;
        memset(dp,-1,sizeof(dp));
        int lim = 1<<n;
        fr(i , 0 , lim){
            if (!(i<<1&i || 1>>1&i)) canuse[tot++] = i;
        }
        ll ans = dfs(0,n+1,k);
        cout<<ans<<endl;
    }
    return 0;
}