HDU 6053 - TrickGCD
阿新 • • 發佈:2017-07-28
需要 時間 print 必須 def int mem ... bit
/* HDU 6053 - TrickGCD [ 莫比烏斯函數,篩法分塊 ] 題意: 給出數列 A[N],問滿足: 1 <= B[i] <= A[i] ; 對任意(l, r) (1<=l<=r<=n) ,使得 gcd(bl,...br) >= 2 ; 的 B[N] 數列的個數 分析: 設 gcd(b1,...bn) = k (k >= 2),此時 k 對答案的貢獻為 (a1/k)*(a2/k)*(a3/k)*...*(an/k) 根據容斥原理,ans = +[k=一個素數之積 時對答案的貢獻] -[k=兩個素數之積 時對答案的貢獻] +[k=三個素數之積 時對答案的貢獻] ... 故任意k對答案的貢獻系數 μ(k) = 0 , k是完全平方數的倍數 = (-1)^(n-1) , k = p1*p2*p3*...*pn ,p是素數 貢獻系數可以O(nsqrt(n)) 或者 O(nlogn) 預處理,再或者可以看出μ(k) 是莫比烏斯函數的相反數 現在枚舉k需要O(n)的時間,計算k對答案的貢獻必須在O(sqrt(n))的時間之內 將a[]處理成權值數組,並求前綴和,設為 sum[] 對於每個k,對sum[]進行埃式篩法的分塊,即根據k的倍數分塊 此時每個k的貢獻 = 1^(sum[2k-1]-sum[k-1]) * 2^(sum[3k-1]-sum[2k-1]) * 3^(sum[4k-1]-sum[3k-1]) ... 就做到 O(n(logn)^2) 編碼時長: 40分鐘(0) */ #include <bits/stdc++.h> using namespace std; #define LL long long const LL MOD = 1e9+7; const int N = 1e5+4; bool notp[N]; int prime[N], pnum, mu[N]; void Mobius() { memset(notp, 0, sizeof(notp)); mu[1] = 1; for (int i = 2; i < N; i++) { if (!notp[i]) prime[++pnum] = i, mu[i] = -1; for (int j = 1; prime[j]*i < N; j++) { notp[prime[j]*i] = 1; if (i%prime[j] == 0) { mu[prime[j]*i] = 0; break; } mu[prime[j]*i] = -mu[i]; } } for (int i = 0; i < N; i++) mu[i] = -mu[i]; } LL PowMod(LL a, int m) { if (a == 1 || m == 0) return 1; if (a == 0) return 0; LL res = 1; while (m) { if (m&1) res = res*a % MOD; a = a*a % MOD; m >>= 1; } return res; } int a[N]; int t, n; int sum[N]; int Max, Min; LL ans; void solve() { int i, j, k, p; ans = 0; for (i = 2; i <= Min; ++i) { if (!mu[i]) continue; LL res = 1; j = min(i, Max), k = min((i<<1)-1, Max); for (p = 1; ; ++p) { if (sum[k] - sum[j-1]) res = res*PowMod(p, sum[k] - sum[j-1]) % MOD; if (k == Max) break; j += i; k += i; if (k > Max) k = Max; } ans += mu[i]*res; if (ans > MOD) ans -= MOD; if (ans < 0) ans += MOD; } } int main() { int i; Mobius(); scanf("%d", &t); for (int tt = 1; tt <= t; ++tt) { scanf("%d", &n); for (i = 0; i < N; ++i) sum[i] = 0; for (i = 1; i <= n; ++i) { scanf("%d", &a[i]); ++sum[a[i]]; } Max = Min = a[1]; for (i = 2; i <= n; ++i) { Max = max(Max, a[i]); Min = min(Min, a[i]); } for (i = 1; i <= Max; ++i) sum[i] += sum[i-1]; solve(); printf("Case #%d: %lld\n", tt, ans); } }
HDU 6053 - TrickGCD