agc045_c Range Set

agc045_c Range Set

https://atcoder.jp/contests/agc045/tasks/agc045_c

Tutorial

https://img.atcoder.jp/agc045/editorial.pdf

考虑如何判断是否能得到某个01串

对于长度大于等于\(A\)的连续的0,可以把它看作1,这样变化后问题是等价的(最后将其染回0即可)

考虑在这样变换之后,如果存在一个长度大于等于\(B\)的连续的1,可以以它为中心按如下方式扩展

但是,考虑如果 \(A>B\) ,那么在边界的时候,这样的方法就有一点问题.

发现我们其实可以先将所有字符染为1,这样就得到了一个对称的问题.所以我们可以保证 \(A \le B\) .

用DP统计满足上述条件的序列数即可

Code

#include <cstdio>
#include <iostream>
#define debug(...) fprintf(stderr,__VA_ARGS__)
#define upd(a,b) a=add(a+b)
using namespace std;
inline char nc() {
//	return getchar();
	static char buf[100000],*l=buf,*r=buf;
	return l==r&&(r=(l=buf)+fread(buf,1,100000,stdin),l==r)?EOF:*l++;
}
template<class T> void read(T &x) {
	x=0; int f=1,ch=nc();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=nc();}
	while(ch>='0'&&ch<='9'){x=x*10-'0'+ch;ch=nc();}
	x*=f;
}
typedef long long ll;
const int mod=1e9+7;
const int maxn=5000+5;
int n,A,B;
int f[maxn][2],g[maxn][2];
int dp[maxn][2][2];
inline int add(int x) {return x>=mod?x-mod:x;}
void DP(int dp[maxn][2]) {
	for(int i=0;i<n;++i) {
		upd(dp[i+1][1],dp[i][0]);
		upd(dp[i+1][1],dp[i][1]);
		for(int k=A;i+k<=n;++k) upd(dp[i+k][0],dp[i][1]);
	}
}
int main() {
	read(n),read(A),read(B);
	if(A>B) swap(A,B);
	f[0][0]=1,DP(f);
	g[0][0]=1;
	for(int i=A;i<=n;++i) g[i][0]=1;
	DP(g);
	dp[n][1][0]=g[n][0];
	for(int i=1;i<=n;++i) dp[i][i>=B][1]=g[i][1];
	for(int i=1;i<A;++i) dp[i][0][0]=1;
	for(int i=1;i<=n;++i) for(int k=0;k<2;++k) {
		for(int j=1;j<A&&i+j<=n;++j) upd(dp[i+j][k][0],dp[i][k][1]);
		for(int j=1;i+j<=n;++j) {
			int r=f[j][1]; if(i+j==n) upd(r,f[j][0]);
			upd(dp[i+j][k||j>=B][1],(ll)dp[i][k][0]*r%mod);
		}
	}
	printf("%d\n",add(dp[n][1][0]+dp[n][1][1]));
	return 0;
}