题解 CF1559E 【Mocha and Stars】


推广个人博客,欢迎大家来玩。

题目意思:

给定 \(n\)\(m\),求 \(n\) 个数满足以下 \(3\) 个条件的方案数(对 \(998244353\) 取模):

  1. \(l_i\leq a_i\leq r_i(1\leq i\leq n)\)
  2. \(\sum\limits_{i=1}^{n} a_i \leq m\)
  3. \(\gcd(a_1,a_2,\cdots,a_n)=1\)

考虑对 \(\gcd\) 大力容斥,这样就变成了让 \(\gcd(a_1,a_2,\cdots,a_n)\) 为某个数的倍数的方案数,因为 \(\gcd\) 要为某个数的倍数。所以 \(a_i\) 必须为那个数的倍数,我们可以考虑直接除一下。

考虑 \(dp\)\(f_{i,j}\) 表示前 \(i\) 个质数和(除了之后)为 \(j\) 的方案数。

求出每个数的合法区间(除来之后的),枚举它,设它为 \(k\),转移就是 \(f_{i,j}\gets f_{i-1,j-k}\)

最后再提一下,容斥乘的数是 。

#include
#define log(a) cerr<<"\033[32m[DEBUG] "<<#a<<'='<<(a)<<" @ line "<<__LINE__<<"\033[0m"<=(b);--i)
using namespace std;
inline int read(){char ch=getchar(); int w=1,c=0;
	for(;!isdigit(ch);ch=getchar()) if (ch=='-') w=-1;
	for(;isdigit(ch);ch=getchar()) c=(c<<1)+(c<<3)+(ch^48);
	return w*c;
}
const int N=55,M=1e5+10,MOD=998244353;
int l[N],r[N],ll[N],rr[N],prim[M],cnt,s[M];
bool is[M];
int f[N][M],g[N][M],ans,h[M];
signed main(){
	int n=read(),m=read();
	h[1]=1;
	F(i,2,m){
		if(!is[i]){
			h[i]=-1;
			prim[++cnt]=i;
		}
		for(int j=1;j<=cnt&&i*prim[j]
  • Update:复杂度分析:

    我们把枚举容斥的数设为 \(x\),一次 \(dp\),和的上限是 \(\frac{m}{x}\)。所以单次 dp 的复杂度是 \(O(n\times\frac{m}{x})\),但 \(x\) 会从 \(1\) 一直到 \(\frac{m}{n}\)

    即:

    \[\begin{aligned} &\sum_{x=1}^{\frac{m}{n}} n\times\frac{m}{x}\\ =&n\times m\sum_{x=1}^{\frac{m}{n}} \frac{1}{x}\\ =&n\times m\times\log\frac{m}{n} \end{aligned} \]