【題目大意】
給定一個n*n的棋盤,求放置k個互不攻擊的象的方法數。其中n <= 8,k <= n ^
2。
【題目分析】
對于棋盤放車問題可以用組合數學的知識來解決,但是對于含禁區的擺放問題,雖然組合數學給出了經典的棋盤多項式+容斥原理的解法,但是實際中棋盤多項式的求解是很困難的,因此一般需要借助狀態壓縮動態規劃求解。
現在題目中要求出互不攻擊的象的方法數,象的攻擊路線是斜的,是不是可以考慮采用放車的方法來解呢?將棋盤黑白染色,如果一個象在黑色的格子里面,那么它一定不會攻擊到白色的格子,這樣的話可以分開計數,然后最后利用乘法原理加起來就行了。把棋盤旋轉45度,這樣象的攻擊路線就是直的了,如果只考慮一種顏色的話,那么問題就轉變成了經典的放車問題了,可以利用動態規劃解決。
設dp[i][j]表示前i行放了j個車的方法數,c[i]表示第i行可以放置的棋子數量,那么轉移方程為:
dp[i][j]
= dp[i-1][j] + dp[i-1][j-1] * (c[i] - (j -
1))
需要注意的是c數組應該是增序的,這樣才能保證前面的j-1行放了車,對應這一行就有j-1個位置不可放了。
這個題目的dp方程不難想,但是如何把模型轉化到放車問題是不容易想到的,尤其是將棋盤黑白染色后分開計數的想法,非常巧妙。
題目代碼:
1 #include <iostream>
2 #include <algorithm>
3 using namespace std;
4 const int N = 70;
5
6 void init(int n, int c1[N], int c2[N])
7 {
8 memset(c1, 0, sizeof(int) * N);
9 memset(c2, 0, sizeof(int) * N);
10 for (int i = 1; i <= n; i++)
11 {
12 for (int j = 1; j <= n; j++)
13 {
14 if ((i + j) & 1)
15 c2[(i+j)/2]++;
16 else
17 c1[(i+j)/2]++;
18 }
19 }
20 }
21 void bishops(int n, int dp[N][N], int c[N])
22 {
23 for (int i = 0; i <= n; i++)
24 dp[i][0] = 1;
25 for (int i = 1; i <= n; i++)
26 for (int j = 1; j <= c[i]; j++)
27 dp[i][j] = dp[i-1][j] + dp[i-1][j-1] * (c[i] - j + 1);
28 }
29
30 int main()
31 {
32 int n, k, c1[N], c2[N], dp1[N][N], dp2[N][N], ans;
33
34 while (scanf("%d %d", &n, &k) == 2)
35 {
36 if (n == 0 && k == 0)
37 break;
38 init(n, c1, c2);
39 sort(c1 + 1, c1 + n + 1);
40 sort(c2 + 1, c2 + n);
41 memset(dp1, 0, sizeof(dp1));
42 memset(dp2, 0, sizeof(dp2));
43 bishops(n, dp1, c1);
44 bishops(n - 1, dp2, c2);
45 ans = 0;
46 for (int i = 0; i <= k; i++)
47 ans += dp1[n][i] * dp2[n-1][k-i];
48 printf("%d\n", ans);
49 }
50
51 return 0;
52 }
注:本文作于2009年6月23日 19點51分