A. Black and White
题意:n个格子排在一行,每个格子里都有一枚白棋或一枚黑棋。限制:不能有连续a枚黑棋或连续b枚白棋,问有多少种方案。
tags:一开始还以为是组合数学,没想到又是dp。这mmp的还是签到题
考虑长度为 i 的合法序列与长度为 i−1 的合法序列有什么关系。定dp[i][black] 为有 i 枚棋子且最后一枚是黑色的合法方案数量,g[i]为有 i 枚棋子的合法方案数量。现假设前面 i-1 枚棋都是合法的,在 i 位置加一枚黑棋,唯一不合法的情况就是最后的a枚棋子全是黑色,即g[i]=g[i-1]-dp[i-a][white]。白棋同理,递推到g[n]即是答案。
![](https://i-blog.csdnimg.cn/blog_migrate/8f900a89c6347c561fdf2122f13be562.gif)
![](https://i-blog.csdnimg.cn/blog_migrate/961ddebeb323a10fe0623af514929fc1.gif)
#include<bits/stdc++.h> using namespace std; #pragma comment(linker, "/STACK:102400000,102400000") #define FF(i,a,b) for (int i=a;i<=b;i++) #define F(i,b,a) for (int i=b;i>=a;i--) #define mes(a,b) memset(a,b,sizeof(a)) #define INF 0x3f3f3f3f typedef long long ll; const int N = 2e6+10, mod=1e9+7; int a, b, n, T; ll dp[N][2], g[N]; int main() { scanf("%d", &T); while(T--) { scanf("%d %d %d", &a, &b, &n); dp[0][0]=dp[0][1]=1, g[0]=1; FF(i,1,n) { if(i-a>=0) dp[i][0]=g[i-1]-dp[i-a][1]; else dp[i][0]=g[i-1]; if(i-b>=0) dp[i][1]=g[i-1]-dp[i-b][0]; else dp[i][1]=g[i-1]; g[i]=(dp[i][0]+dp[i][1])%mod; } printf("%lld\n", (g[n]+mod)%mod); } return 0; }