do_while_true

一言(ヒトコト)

「题解」Codeforces 1559E Mocha and Stars

莫比乌斯反演的另外一种形式:

如果有:

\[f(n)=\sum_{n|d}g(d) \]

则有:

\[g(n)=\sum_{n|d}\mu(\frac{d}{n})f(d) \]

这里的 \(d\) 通常是小于等于一个界限,也就是题目中所给定的 "\(n\)"。

特别地,当 \(n=1\) 时,\(g(1)=\sum \mu(d)f(d)\),也可以理解为容斥系数为 \(\mu\) 的容斥。


对第三个条件进行莫比乌斯反演,得:

\[ans=\sum_{d=1}^m \mu(d)f(d) \]

其中 \(f(d)\)\(\gcd\)\(d\) 的倍数时的答案。

对于 \(f(d)\),每次可以做一下背包,用前缀和优化易做到 \(\mathcal{O}(\frac{nm}{d})\)​,总时间复杂度就是 \(\mathcal{O}(nm\log nm)\),注意到 \(\mu(d)\)\(0\) 的时候不用跑背包,所以这个远远跑不满。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<vector>
#define int long long
#define mp std::make_pair
typedef long long ll;
const ll mod = 998244353;
void cAdd(ll &x, ll y) { x = (x + y >= mod) ? (x + y - mod) : (x + y); }
ll Add(ll x, ll y) { return (x + y >= mod) ? (x + y - mod) : (x + y); }
ll Mod(ll x) { return (x+mod)%mod; }
typedef std::pair<int, int> pii; 
template <typename T> T Max(T x, T y) { return x > y ? x : y; }
template <typename T> T Min(T x, T y) { return x < y ? x : y; }
template <typename T>
T& read(T& r) {
	r = 0; bool w = 0; char ch = getchar();
	while(ch < '0' || ch > '9') w = ch == '-' ? 1 : 0, ch = getchar();
	while(ch >= '0' && ch <= '9') r = r * 10 + (ch ^ 48), ch = getchar();
	return r = w ? -r : r;
}
const int N = 500010;
int n, m, L[N], R[N], l[N], r[N];
int ct, prime[N], mu[N];
bool vis[N];
void pre() {
	vis[1] = 1; mu[1] = 1;
	for(int i = 2; i <= 500000; ++i) {
		if(!vis[i]) {
			prime[++ct] = i;
			mu[i] = -1;
		}
		for(int j = 1; j <= ct && i * prime[j] <= 500000; ++j) {
			vis[i * prime[j]] = 1;
			if(i % prime[j] == 0) mu[i * prime[j]] = 0;
			else mu[i * prime[j]] = mu[i] * mu[prime[j]];
			if(i % prime[j] == 0) break;
		}
	}
}
ll f[N], g[N], ans;
signed main() {
	pre();
	read(n); read(m);
	for(int i = 1; i <= n; ++i) read(L[i]), read(R[i]);
	for(int d = 1; d <= m; ++d) {
		if(!mu[d]) continue ;
		for(int i = 1; i <= n; ++i) l[i] = (L[i]+d-1)/d, r[i] = R[i]/d;
		int s = m/d;
		for(int i = 0; i <= s; ++i) g[i] = 1;
		for(int i = 1; i <= n; ++i) {
			for(int j = 1; j <= s; ++j) f[j] = 0;
			for(int j = l[i]; j <= s; ++j)
				f[j] = Add(g[j-l[i]], (j>=r[i]+1)?Mod(-g[j-r[i]-1]):0);
			g[0] = 0;
			for(int j = 1; j <= s; ++j)	g[j] = Add(g[j-1], f[j]);
		}
		ans = Add(ans, Mod(mu[d] * g[s] % mod));
	}
	printf("%lld\n", ans);
	return 0;
}
posted @ 2021-08-17 16:08  do_while_true  阅读(84)  评论(0编辑  收藏  举报