nyoj-715-Adjacent Bit Counts（动态规划）

原文链接:nyoj-715-Adjacent Bit Counts

---

题目大意：

给定一个01字符串，使用如下公式 fun(x) = x1*x2 + x2*x3 + x3*x 4 + … + xn-1*x n 计算其权值。例如 Fun(011101101) = 3。要求出字符串的长度和权值数，输出满足该要求字符串的个数。

---

解题思路：

一开始用递归模拟了一下，大概十几秒才输出。。。
看了别人的博客知道要用动态规划。其中

dp[i][j][0] 或 dp[i][j][1]用来存储 长度为i 权值为j 末尾为0或1 时的权值

动态转移方程就是：

dp[i][j][0] = dp[i - 1][j][0] + dp[i - 1][j][1];
dp[i][j][1] = dp[i - 1][j][0] + dp[i - 1][j - 1][1];

---

代码：

#pragma warning(disable : 4996)
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<math.h>
#include<string>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#include<set>
#define LOCAL

using namespace std;

const int MAXN = 100 + 5;

int T, n, p;
int dp[MAXN][MAXN][2];//dp[i][j][0] 或 dp[i][j][1]用来存储 长度为i 权值为j 末尾为0或1 时的权值

int main()
{
    //freopen("C:\\input.txt", "r", stdin);
    cin >> T;
    //构造
    dp[1][0][0] = dp[1][0][1] = 1;
    for (int i = 2; i < MAXN; i++) {
        dp[i][0][0] = dp[i - 1][0][0] + dp[i - 1][0][1];
        dp[i][0][1] = dp[i - 1][0][0];
        for (int j = 1; j <= i; j++) {
            //转移方程
            dp[i][j][0] = dp[i - 1][j][0] + dp[i - 1][j][1];
            dp[i][j][1] = dp[i - 1][j][0] + dp[i - 1][j - 1][1];
        }
    }
    while (T--) {
        cin >> n >> p;
        cout << dp[n][p][0] + dp[n][p][1] << endl;
    }
    return 0;
}