uva 10237 Bishops 黑白染色处理

题意：在n*n行棋盘里放置k个棋子，要保证这些棋不在同一斜行，问有多少种放置方式。

方法：黑白染色，黑白格子之间相互不影响。

用dp[i][j]表示第i斜行放置j个棋子，c[i]表示第i斜行有多少棋子。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<vector>
#include<queue>
#include<algorithm>
#include<math.h>
using namespace std;
#pragma comment (linker, "/STACK:1024000000,1024000000")
const int N=33;
const int INF=0x3f3f3f3f;
typedef long long ll;
int n,K;
int c1[N],c2[N];
ll dp1[N][N*N],dp2[N][N*N];
void init(){
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if((i+j)&1)c2[(i+j)>>1]++;
            else c1[(i+j)>>1]++;
        }
    }
}
void get(int m,int c[N],ll dp[N][N*N]){
    for(int i=0;i<=m;i++)dp[i][0]=1;
    for(int i=1;i<=m;i++){
        for(int j=1;j<=K&&j<=c[i];j++){
            dp[i][j]=dp[i-1][j]+dp[i-1][j-1]*(c[i]-j+1);
        }
    }
}
void work(){
    freopen("A_in.txt","r",stdin);
    while(~scanf("%d%d",&n,&K)&&n){
        memset(c1,0,sizeof(c1));
        memset(c2,0,sizeof(c2));
        init();
        sort(c1+1,c1+n+1);
        sort(c2+1,c2+n);
        get(n,c1,dp1);
        get(n-1,c2,dp2);
        long long ans=0;
        for(int i=0;i<=K;i++){
            ans+=dp1[n][i]*dp2[n-1][K-i];
        }
        printf("%lld\n",ans);
    }
}
int main(){
    work();
    return 0;
}