2021.08.31 CF1558E Up the Strip

the last blog of the summer holiday of 2021

$$\text{CF1558E Up the Strip}$$

$\text{[Problem]}$

$\text{[Solution]}$

可能会比较容易看出要用 dp 吧。

记 $f_x$ 表示从 $x$ 转移到 $1$ 的方案数，考虑如何转移。

显然，考虑对于任意一个点 $u(x<u\le n)$，$u$ 都可以通过方式 $1$ 转移到 $x$。

现在我们需要考虑哪些 $u$ 可以通过方式 $2$ 转移到 $x$。

显然，对于 $\forall u=at+b(0\le b<a)$，都有 $\lfloor \frac{u}{t}\rfloor =a$，反之亦然。

故而有满足 $\lfloor \frac{u}{z}\rfloor =x$ 的 $u\in [zx,zx+z-1]$。

所以，我们可以愉快地写出状态转移方程。

由于所有的 $u$ 都集中在一个区间中，所以我们可以愉快地使用后缀和优化算法到 $\mathcal{O}(n)$。

$\text{[code]}$

const int N=4e6+100;
int f[N],sum[N],n,mod;
int main(){
	scanf("%d%d",&n,&mod);
	f[n]=1;sum[n]=1;sum[n+1]=0;
	for(int i=n-1;i;i--){
		f[i]=(f[i]+sum[i+1])%mod;
		for(int j=2;1ll*i*j<=n;j++){
			f[i]=(f[i]+sum[i*j]-sum[min((i+1)*j,n+1)])%mod;
			f[i]=(f[i]+mod)%mod;
		}
		sum[i]=(sum[i+1]+f[i])%mod;
	}
	printf("%d",f[1]);
	return 0;
}