[CQOI2015]选数(数论分块+杜教筛)
problem
洛谷鏈接
solution
將 L,HL,HL,H 的范圍放縮 1K\frac 1 KK1?,都除掉 KKK,特殊的 LLL 邊界注意一下。
H←H/K,L←(L?1)/K+1H\leftarrow H/K,L\leftarrow (L-1)/K+1H←H/K,L←(L?1)/K+1。
問題轉(zhuǎn)化為 [L,H][L,H][L,H] 中任選 NNN 個數(shù) gcd=1\text{gcd}=1gcd=1 的方案數(shù)。
T(i):T(i):T(i): 從 [L,H][L,H][L,H] 中選 NNN 個數(shù) i∣gcdi|\text{gcd}i∣gcd 的方案數(shù),即 (?Hi???L?1i?)N(\lfloor\frac H i\rfloor-\lfloor\frac {L-1}i\rfloor)^N(?iH????iL?1??)N。
t(i):t(i):t(i): 從 [L,H][L,H][L,H] 中選 NNN 個數(shù) i=gcdi=\text{gcd}i=gcd 的方案數(shù)。
根據(jù)定義顯然有,T(i)=∑i∣dt(d)T(i)=\sum_{i|d}t(d)T(i)=∑i∣d?t(d)
莫比烏斯反演得 t(i)=∑i∣dμ(di)T(d)t(i)=\sum_{i|d}\mu(\fracze8trgl8bvbq{i})T(d)t(i)=∑i∣d?μ(id?)T(d)
答案即為 t(1)=∑i=1infμ(i)T(i)=∑i=1infμ(i)(?Hi???L?1i?)Nt(1)=\sum_{i=1}^{inf}\mu(i)T(i)=\sum_{i=1}^{inf}\mu(i)(\lfloor\frac H i\rfloor-\lfloor\frac {L-1}i\rfloor)^Nt(1)=∑i=1inf?μ(i)T(i)=∑i=1inf?μ(i)(?iH????iL?1??)N。
H,LH,LH,L 非常大,不能直接線篩后整除分塊。
但可杜教篩,設(shè)定一個閥值 MMM 預(yù)處理出 MMM 以內(nèi)的 μ\muμ,同樣整除分塊后杜教篩能做到 O(H23)O(H^{\frac 2 3})O(H32?)。
按 T(i)T(i)T(i) 分類,假設(shè) i∈[l,r]i\in[l,r]i∈[l,r] 的 T(i)T(i)T(i) 都是一樣的,那么對 ∑i=lrμ(i)\sum_{i=l}^r\mu(i)∑i=lr?μ(i) 進行杜教篩迅速求和。
∑i=lrμ(i)=∑i=1rμ(i)?∑i=1l?1μ(i)\sum_{i=l}^r\mu(i)=\sum_{i=1}^r\mu(i)-\sum_{i=1}^{l-1}\mu(i)∑i=lr?μ(i)=∑i=1r?μ(i)?∑i=1l?1?μ(i)。這樣就化成了標準的杜教篩形式。
∑μ(i):?=μ?I\sum\mu(i):\epsilon=\mu*I∑μ(i):?=μ?I。h??;f?μ;g?Ih\leftrightarrow \epsilon;f\leftrightarrow \mu;g\leftrightarrow Ih??;f?μ;g?I。
s(n)=∑i=1nf(i)=∑i=1nμ(i)s(n)=\sum_{i=1}^nf(i)=\sum_{i=1}^n\mu(i)s(n)=∑i=1n?f(i)=∑i=1n?μ(i)
g(1)s(n)=∑i=1n?(i)?∑i=2ng(i)s(?ni?)?s(n)=1?∑i=2ns(?ni?)g(1)s(n)=\sum_{i=1}^n\epsilon(i)-\sum_{i=2}^ng(i)s(\lfloor\frac n i\rfloor)\Leftrightarrow s(n)=1-\sum_{i=2}^ns(\lfloor\frac n i\rfloor)g(1)s(n)=∑i=1n??(i)?∑i=2n?g(i)s(?in??)?s(n)=1?∑i=2n?s(?in??)。
對 sss 函數(shù)進行記憶化遞歸,以及同樣的整除分塊。
這樣子連 H?L≤1e5H-L\le 1e5H?L≤1e5 的性質(zhì)都沒有用上!^_^ 這性質(zhì)好像是拿來容斥遞推用的。
一些省掉的推導(dǎo)?莫比烏斯反演,杜教篩。
code
#include <bits/stdc++.h> using namespace std; #define int long long #define mod 1000000007 #define maxn 1000005 int mu[maxn], prime[maxn]; bool vis[maxn]; unordered_map < int, int > mp; int N, M, K, L, H, cnt;int qkpow( int x, int y ) {int ans = 1;while( y ) {if( y & 1 ) ans = ans * x % mod;x = x * x % mod;y >>= 1;}return ans; }int solve( int n ) {if( n <= M ) return mu[n];if( mp.find( n ) != mp.end() ) return mp[n];int ans = 1;for( int l = 2, r;l <= n;l = r + 1 ) {r = n / ( n / l );( ans -= solve( n / l ) * ( r - l + 1 ) ) %= mod;}return mp[n] = ans; }signed main() {scanf( "%lld %lld %lld %lld", &N, &K, &L, &H );H /= K, L = ( L - 1 ) / K + 1;M = min( (int)1e6, H );mu[1] = 1; for( int i = 2;i <= M;i ++ ) {if( ! vis[i] ) prime[++ cnt] = i, mu[i] = -1;for( int j = 1;j <= cnt and i * prime[j] <= M;j ++ ) {vis[i * prime[j]] = 1;if( i % prime[j] == 0 ) { mu[i * prime[j]] = 0; break; }else mu[i * prime[j]] = -mu[i];}}for( int i = 1;i <= M;i ++ ) mu[i] += mu[i - 1];L --; int ans = 0;for( int l = 1, r;l <= H;l = r + 1 ) {if( ! ( L / l ) ) r = H / ( H / l );else r = min( H / ( H / l ), L / ( L / l ) );( ans += qkpow( H / l - L / l, N ) * ( solve( r ) - solve( l - 1 ) ) ) %= mod;}printf( "%lld\n", ( ans + mod ) % mod );return 0; }總結(jié)
以上是生活随笔為你收集整理的[CQOI2015]选数(数论分块+杜教筛)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 山东大学学报稿件查询
- 下一篇: 创维25TI9000工厂模式