牛客练习赛 41 B 题 【简单的dp计数】

传送门
题意: 就n个数，开始分数为0, 第i次操作可以有两种选择
1: 给分数加上a[i]
2: 给分数乘-1
问n次过后分数为-666并且任意一次操作后分数不能是666的方案数是多少? (%1e8+7)

思路: 很明显的dp计数, dp[i][j] 表示 i 次操作后分数为 j 的方案数, 那么转移方程是
dp[i][j] = dp[i-1][j-a[i]] + dp[i-1][-j];
去掉中间状态为666的方案

因为有负数的原因, 我们利用偏移量解决, 并且二维开不下, 还需要用滚动数组.
还需要注意把状态666的去掉
详细实现请看代码

const int maxn = 300*700;
int dp[2][maxn*2 + 700];
void solve() {
    int n;
    cin >> n;
    dp[0][maxn] = 1;
    for (int i = 1 ; i <= n ; i ++) {
        int x; cin >> x;
        Fill(dp[i&1], 0); // 因为是滚动数组, 又是累加, 所以每一次得先把下一次要用的清空了才行
        for (int j = -maxn ; j <= maxn ; j ++) {
            if (j != -666) dp[i&1][j+maxn] += dp[!(i&1)][maxn-j];
            if (j - x >= -maxn && j - x != 666) dp[i&1][j+maxn] += dp[!(i&1)][maxn+j-x];
            dp[i&1][j+maxn] %= mod;
        }
    }
    cout << dp[n&1][maxn-666] << endl;
}