Bzoj4652: [Noi2016]循环之美

题面

传送门

Sol

\(设x,y且gcd(x, y)=1\)若使\(\frac{x}{y}\)\(k\)进制小数是纯循环小数
则一定存在某次除法中余数在之前出现过
也就是存在\(L>0\)\(x\equiv x*k^L(mod\ y)\)
\(x,y互质\)那么同时乘上x的逆元则\(k^L\equiv1(mod\ y)\)
所以\(k^L=a*y+1(a \in Z)\)
\(y\)\(a*y+1\)互质,辗转相除法可证
所以\(k^L与y\)互质,易得\(k与y也互质\)

所以我们就证明了伪证\(k与y\)互质时,才会出现\(\frac{x}{y}\)\(k\)进制小数是纯循环小数

接下来就解题辣
暴力求就不讲了
直接推柿子:
要求的就是\(\sum_{i=1}^{n}\sum_{j=1}^{m}[gcd(i, j)==1, gcd(j, k)==1]\)

\[=\sum_{i=1}^{m}[gcd(i, k)==1]\sum_{j=1}^{n}[gcd(i, j)==1] \]

莫比乌斯反演一波

\[=\sum_{i=1}^{m}[gcd(i, k)==1]\sum_{j=1}^{n}\sum_{d|i,d|j}\mu(d) \]

\[=\sum_{i=1}^{m}[gcd(i, k)==1]\sum_{d|i}^n\mu(d)\lfloor\frac{n}{d}\rfloor \]

\[=\sum_{d=1}^{n}\mu(d)\lfloor\frac{n}{d}\rfloor\sum_{d|i}[gcd(i, k)==1] \]

\[=\sum_{d=1}^{n}\mu(d)\lfloor\frac{n}{d}\rfloor[gcd(d, k)==1]\sum_{i=1}^{\lfloor\frac{m}{d}\rfloor}[gcd(i, k)==1] \]

如果\(\sum_{i=1}^{\lfloor\frac{m}{d}\rfloor}[gcd(i, k)==1]\)可以预处理\(O(1)\)
那么就有了\(O(n)\)的解法

\(x=\lfloor\frac{m}{d}\rfloor\)
如果\(x<=k\)直接\(O(k)\)处理
否则
因为\(gcd(i, k)=gcd(i\%k, k)\),所以\(O(k)\)处理后乘上\(\lfloor\frac{x}{k}\rfloor\)

$O(n)$84分常数很大

# include <bits/stdc++.h>
# define RG register
# define IL inline
# define Fill(a, b) memset(a, b, sizeof(a))
using namespace std;
typedef long long ll;
const int _(2e7 + 1);

IL ll Read(){
    RG ll x = 0, z = 1; RG char c = getchar();
    for(; c < '0' || c > '9'; c = getchar()) z = c == '-' ? -1 : 1;
    for(; c >= '0' && c <= '9'; c = getchar()) x = (x << 1) + (x << 3) + (c ^ 48);
    return x * z;
}

int N, num, prime[_], mu[_], k, f[2010];
bool isprime[_];

IL int Gcd(RG int x, RG int y){  return !y ? x : Gcd(y, x % y);  }

IL void Prepare(){
	isprime[1] = 1; mu[1] = 1;
	for(RG int i = 2; i < N; ++i){
		if(!isprime[i]) prime[++num] = i, mu[i] = -1;
		for(RG int j = 1; j <= num && i * prime[j] < N; ++j){
			isprime[i * prime[j]] = 1;
			if(i % prime[j]) mu[i * prime[j]] = -mu[i];
			else{  mu[i * prime[j]] = 0; break;  }
		}
	}
	for(RG int i = 1; i <= k; ++i) f[i] = f[i - 1] + (Gcd(i, k) == 1);
}

int main(RG int argc, RG char* argv[]){
	RG int n = Read(), m = Read(); k = Read();
	N = min(m + 1, _); Prepare(); RG ll ans = 0;
	for(RG int i = 1; i <= n; ++i){
		if(Gcd(i, k) != 1) continue;
		RG ll x = m / i, F = f[x % k] + 1LL * (x / k) * f[k];
		ans += 1LL * mu[i] * (n / i) * F;
	}
	printf("%lld\n", ans);
    return 0;
}

怎么优化?

\[\sum_{d=1}^{n}\mu(d)\lfloor\frac{n}{d}\rfloor[gcd(d, k)==1]\sum_{i=1}^{\lfloor\frac{m}{d}\rfloor}[gcd(i, k)==1] \]

这玩意儿后面都可以\(O(1)\)求了
又看到\(\lfloor\frac{n}{d}\rfloor和\lfloor\frac{m}{d}\rfloor\)可以分块
如果能求出\(\mu(d)[gcd(d, k)==1]\)的前缀和就好了

那就继续推柿子:
设$$G(n,k)=\sum_{d=1}^{n}\mu(d)[gcd(d, k)==1]$$
后面的\(gcd\)和之前推的时候类似,这里是一样的,所以莫比乌斯反演一波得

\[\sum_{d=1}^{n}\mu(d)\sum_{i|d, i|k}\mu(i) \]

\[=\sum_{i|k}^{k}\mu(i)\sum_{i|d}^{n}\mu(d) \]

\[=\sum_{i|k}^{k}\mu(i)\sum_{j=1}^{\lfloor\frac{n}{i}\rfloor}\mu(i*j) \]

后面的\(\sum_{j=1}^{\lfloor\frac{n}{i}\rfloor}\mu(i*j)\),显然只有当\(gcd(i, j)=1\)\(\mu(i*j)\)才不为\(0\)
所以可以写成\(\sum_{j=1}^{\lfloor\frac{n}{i}\rfloor}\mu(i)\mu(j)[gcd(i,j)==1]\)
代入

\[=\sum_{i|k}^{k}\mu(i)\sum_{j=1}^{\lfloor\frac{n}{i}\rfloor}\mu(i)\mu(j)[gcd(i,j)==1] \]

\[=\sum_{i|k}^{k}\mu^2(i)\sum_{j=1}^{\lfloor\frac{n}{i}\rfloor}\mu(j)[gcd(i,j)==1] \]

\[=\sum_{i|k}^{k}\mu^2(i)G(\lfloor\frac{n}{i}\rfloor,i) \]

前面的也可以\(O(k)\)处理出来,\(G\)貌似可以杜教筛似的递归处理
暴力在筛里搞一搞k的约数
那不就做完了,复杂度不会证明\(O(跑的过)\)

某些OJ的数据太强啦要卡一卡预处理的大小才能过

# include <bits/stdc++.h>
# define RG register
# define IL inline
# define Fill(a, b) memset(a, b, sizeof(a))
using namespace std;
typedef long long ll;
const int _(4e6 + 1);

IL ll Read(){
    RG ll x = 0, z = 1; RG char c = getchar();
    for(; c < '0' || c > '9'; c = getchar()) z = c == '-' ? -1 : 1;
    for(; c >= '0' && c <= '9'; c = getchar()) x = (x << 1) + (x << 3) + (c ^ 48);
    return x * z;
}

int N, num, prime[_], mu[_], k, f[2010], smu[_];
map < pair <int, int> , ll > G;
bool isprime[_];

IL int Gcd(RG int x, RG int y){  return !y ? x : Gcd(y, x % y);  }

IL void Sieve(){
	isprime[1] = 1; smu[1] = mu[1] = 1;
	for(RG int i = 2; i < N; ++i){
		if(!isprime[i]) prime[++num] = i, mu[i] = -1;
		for(RG int j = 1; j <= num && i * prime[j] < N; ++j){
			isprime[i * prime[j]] = 1;
			if(i % prime[j]) mu[i * prime[j]] = -mu[i];
			else{  mu[i * prime[j]] = 0; break;  }
		}
		smu[i] = smu[i - 1] + mu[i];
	}
	for(RG int i = 1; i <= k; ++i) f[i] = f[i - 1] + (Gcd(i, k) == 1);
}

IL ll Du_Sieve(RG int n, RG int m){
	if(!n) return 0;
	if(m == 1 && n < _) return smu[n];
	if(G[make_pair(n, m)]) return G[make_pair(n, m)];
	RG ll ans = 0;
	if(m == 1){
		ans = 1;
		for(RG int i = 2, j; i <= n; i = j + 1){
			j = n / (n / i);
			ans -= Du_Sieve(n / i, 1) * (j - i + 1);
		}
	}
	else{
		for(RG int i = 1; i * i <= m; ++i){
			if(m % i) continue;
			if(mu[i] != 0) ans += Du_Sieve(n / i, i);
			if(i * i != m && mu[m / i] != 0) ans += Du_Sieve(n / (m / i), m / i);
		}
	}
	return G[make_pair(n, m)] = ans;
}

int main(RG int argc, RG char* argv[]){
	RG int n = Read(), m = Read(); k = Read();
	N = min(m + 1, _); Sieve(); RG ll ans = 0, nxt, lst = 0;
	for(RG int i = 1, j; i <= n; i = j + 1){
		j = n / (n / i); if(m / i) j = min(j, m / (m / i));
		RG ll x = m / i, F = f[x % k] + 1LL * (x / k) * f[k];
		nxt = Du_Sieve(j, k);
		ans += 1LL * (nxt - lst) * (n / i) * F;
		lst = nxt;
	}
	printf("%lld\n", ans);
    return 0;
}

柿子比较多,如有错误请指正

posted @ 2018-01-17 20:38  Cyhlnj  阅读(308)  评论(0编辑  收藏  举报