UVa 861 Little Bishops(棋盘放棋)

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#define mst(a,b) memset(a,b,sizeof(a))
typedef long long LL;
using namespace std;
const int N = 9;
int Cb
,Cw
,Rb
[65],Rw
[65];
//C[]-旋转45°后黑白棋每行格子数(白色多一行)、R[i][j]-前i行放j个棋子的方法数

void Init(int n){
mst(Cb,0);
mst(Cw,0);
Cw[1]=1;
for(int i=1;i<=n;i++){
if(i<=n/2){
Cw[i]=2*i-1;
Cb[i]=2*i;
}
else{
if(!Cw[n+1-i])
Cw[i]=Cw[i-1]+2;
else Cw[i]=Cw[n+1-i];
Cb[i]=Cb[n-i];
}
//cout<<"第"<<i<<"行白棋："<<Cw[i]<<endl;
//cout<<"第"<<i<<"行黑棋："<<Cb[i]<<endl;
}
}

void init(int n){
mst(Cb,0);
mst(Cw,0);
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
if((i+j)&1) Cb[(i+j)>>1]++;
else Cw[(i+j)>>1]++;
}

}
/*for(int i=1;i<=n;i++){
cout<<"第"<<i<<"行白棋："<<Cw[i]<<endl;
cout<<"第"<<i<<"行黑棋："<<Cb[i]<<endl;
}*/
}

void Bishop(int n,int k,int C[],int R[][65]){
for(int i=0;i<=n;i++)   R[i][0]=1;
for(int i=1;i<=k;i++)   R[0][i]=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=C[i];j++)
R[i][j]=R[i-1][j]+R[i-1][j-1]*(C[i]-j+1);
}

int main()
{
int n,k;
while(scanf("%d%d",&n,&k)&&(n||k)){
Init(n);            //两个init作用相同，初始化Cb、Cw数组
sort(Cb+1,Cb+n);
sort(Cw+1,Cw+n+1);

Bishop(n-1,k,Cb,Rb);
Bishop(n,k,Cw,Rw);
LL ans=0;
for(int i=0;i<=k;i++)
ans+=Rb[n-1][i]*Rw
[k-i];
printf("%lld\n",ans);

}

return 0;
}