LA 6801 Sequence(DP)

2015-01-27 06:14:36 · 作者: · 浏览: 7
FILE 6801 - Sequence

题目大意:

给定n个开关(0/1)的初始状态,执行k次操作,每次可以任意选择一个,将其状态反转(0-1, 1-0)。问使得最终状态全是0的方法数%1000000007。


解题思路:

动态规划。用dp[i][j]表示第i次操作后,还有j个1的方案数。

状态转移方程: dp[i][j+1] = dp[i-1][j] * (n-j)

dp[i][j-1] = dp[i-1][j] * j


参考代码:

// Author: Yuan Zhu
#include 
  
   
#include 
   
     #include 
    
      #define ll long long #define mod 1000000007 using namespace std; int t, n, k; int a[1100]; ll dp[1100][1100]; int main() { scanf("%d", &t); for (int ca = 1; ca <= t; ca++) { memset(dp, 0, sizeof dp); scanf("%d%d", &n, &k); int one = 0; for (int i = 1; i <= n; i++) { scanf("%d", a + i); if(a[i] == 1) one++; } dp[0][one] = 1; for (int i = 0; i < k; i++) { for (int j = 0; j <= n; j++) { if (j) dp[i + 1][j - 1] = (dp[i + 1][j - 1] + dp[i][j] * j) % mod; if (j < n) dp[i + 1][j + 1] = (dp[i + 1][j + 1] + dp[i][j] * (n - j)) % mod; } } printf("Case #%d: %lld\n", ca, (dp[k][0] % mod + mod) % mod); } return 0; }