首页 > 代码库 > Painting Storages

Painting Storages

题目链接

  • 题意:
    给n个球,每个球可以涂成红色和蓝色任意一种,求相连的红色球的个数不少于m的涂色方案有多少种,结果对MOD取模
  • 分析:
    两种方法解决,第一种直接求,也可以求,就是状态转移略麻烦;另一种求对立问题,即相连的红色球的个数小于于m的涂色方案有多少种,这个相比第一个简单些,但是第一种也可以解决。采用第二种:
    dp[i]表示i位置放蓝球,满足题意的共几种方案。这个题目如果采用当前状态更新之后状态就比较麻烦,相等于每次区间更新;如果采用由之前的状态求出当前的未知状态就比较简单,区间和,用前缀和求

const int MAXN = 1100000;
int n, m, dp[MAXN], psm[MAXN], f[MAXN];
int main()
{
    f[0] = 1;
    FF(i, 1, MAXN)
        f[i] = f[i - 1] * 2 % MOD;
    while (~RII(n, m))
    {
        dp[0] = psm[0] = 1;
        FE(i, 1, n + 1)
        {
            dp[i] = psm[i - 1];
            if (i - m - 1 >= 0)
                dp[i] = ((dp[i] - psm[i - m - 1]) % MOD + MOD) % MOD;
            psm[i] = (psm[i - 1] + dp[i]) % MOD;
        }
        cout << ((f[n] - dp[n + 1]) % MOD + MOD) % MOD << endl;
    }
    return 0;
}

Painting Storages