CodeForces 1228E Another Filling the Grid (组合数 + dp)

题目: 传送门
思路:
      dp[ i ] [ j ] 表示到 第 i 行时 已经有 j 列有 1 的方案数.
      设该行新增的有 1 的列数为 x,组合数位C,则 对于 x 的 取值有两种转移方式.
      如果 x == 0 :
      因为每行必须有一个 1,那么所有的 1 都在 已经 存在的 j 列里面方案数为 (k^j - (k-1)^j ) * (k-1)^n-j
      如果 x != 0:
      那已经存在的 j 列里面有无1已经无所谓了(因为 n-j 列里面一定有1) 方案数为 (k^j) * (C_(x,n-j)) * (k-1)^n-j-k.

      预处理出所有我们可能使用的组合数和次幂, 由此我们可以得到O(n³)的复杂度下转移.

Ac_Code

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <stack>
#include <bitset>
#include <vector>
#include <map>
#include <string>
#include <cstring>
#define fir first
#define sec second
using namespace std;

const long long mod = 1e9+7;

int n,l;
long long dp[255][255];
long long cur1[250];
long long cur2[250];
long long C[255][255];

int main() {
	scanf("%d%d",&n,&l);
	cur1[0] = cur2[0] = 1;
	for(int i=1;i<=n;i++) {
		cur1[i] = cur1[i-1] * l %mod;
		cur2[i] = cur2[i-1] * (l-1) %mod;
	}
	for(int i=0;i<=n;i++) {
		for(int j=0;j<=i;j++) {
			if(j == 0) C[i][j] = 1;
			else C[i][j] = (C[i-1][j-1] + C[i-1][j])%mod; 
		}
	}
	dp[0][0] =1;
	for(int i = 1;i<=n;i++) {
		for(int j = 0;j<=n;j++) {
			if(dp[i-1][j] == 0) continue;
			for(int k = 0;k<=n;k++) {
				if(j + k > n) break;
				if(k == 0) dp[i][j+k] += dp[i-1][j] * (cur1[j] - cur2[j] + mod) %mod * cur2[n-j] %mod;
				else {
					dp[i][j+k] += dp[i-1][j] * (cur1[j] * C[n-j][k] %mod * (cur2[n-j-k]) %mod) %mod;
				}
				dp[i][j+k] %= mod;
			}
		}
	}
	printf("%lld\n",dp[n][n]);
	return 0;
}