LightOJ 1005 - Rooks （组合数学dp模拟）

题目链接：http://lightoj.com/volume_showproblem.php?problem=1005

题意：n*n的棋盘上放m个车，问有多少种放车的方式使得每个车都不会互相攻击到（即每个车都占有他所在的一行一竖两条直线）

思路：这道dp模拟的是组合数学的算法（ n行中选出m行，C(n,m)，再在n列中选出m列随便放A(n,m)，答案为C(n,m)*A(n,m) ）

代码：

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <string>
#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
#include <deque>
#include <queue>
#include <stack>
#define ll long long

using namespace std;

ll dp[35][35];

void solve()
{
	int i,j;
	memset(dp,0,sizeof(dp));//初始化排除k>n的时候的情况
	for(i=1;i<31;i++){
		dp[i][0]=1;
		dp[i][1]=i*i;
	}
	for(i=2;i<31;i++){
		for(j=2;j<=i;j++)
			dp[i][j]=dp[i][1]*dp[i-1][j-1]/j;//模拟C(n,m)*A(n,m) 
	}
}

int main()
{
	int n,k,t=1,T;
	solve();
	scanf("%d",&T);
	while(T--){
		scanf("%d%d",&n,&k);
		printf("Case %d: %lld\n",t++,dp[n][k]);
	}
	return 0;
}