hdu1695(莫比乌斯反演模板)

题目链接: http://acm.hdu.edu.cn/showproblem.php?pid=1695

题意: 对于 a, b, c, d, k . 有 x 属于 [a, b],  y 属于 [c, d], 求 gcd(x, y) = k 的 x, y 的对数 . 其中 a = b = 1 .

注意: (x, y), (y, x) 算一种情况 .

思路: 莫比乌斯反演

可以参考一下: http://blog.csdn.net/lixuepeng_001/article/details/50577932

公式: F(n) = sigma f(d) , 其中 (n | d) ==> f(n) = sigma u(d / n) * F(d) , 其中 (n | d) .

其中 u 为莫比乌斯函数, 定义为:

　　若 d = 1, 则 u(d) = 1;

　　若 d = p1p2...pk, 则 u(d) = (-1)^k , 其中 pi 为互异质数;

　　其他 u(d) = 0;

对于 gcd(x, y) = k, 显然有 gcd(x / k, y / k) = 1 . 那么原题等价于求 gcd(x, y) = 1, 其中 x 属于 (1, b / k), y 属于 (1, d / k) .

然后定义 f(n) 表示满足条件的 gcd(x,y) = n 的 (x, y) 对数,

再定义 F(n) 表示满足 n | gcd(x,y) 的 (x, y) 对数, 即 gcd(x, y) % n = 0 的x, y对数 .

那么我们要求的就是 f(1) .

通过前面推理不难发现 F(x) = n / x * m / x, 其中 n = b / k, m = d / k;

那么 f(1) = sigma u(d / 1) * F(d), 其中 1 | d 且 d <= min(n, m);

即 f(1) = sigma u(d) * (n / d) * (m / d);

关于去重: f'(1) = sigma u(d) * (n / d) * (n / d), 其中 n 为 m, n中的最小值, 则 sol = f(1) - f'(1) / 2;

代码:

 #include <iostream>
 #include <stdio.h>
 #include <string.h>
 #define ll long long
 using namespace std;

 const int MAXN = 1e6 + ;

 bool check[MAXN];
 int mu[MAXN], prime[MAXN];

 void Moblus(void){
     memset(check, false, sizeof(check));
     int tot = ;
     mu[] = ;
     for(int i = ; i < MAXN; i++){
         if(!check[i]){
             prime[tot++] = i;
             mu[i] = -;
         }
         for(int j = ; j < tot && i * prime[j] < MAXN; j++){
             check[i * prime[j]] = true;
             if(i % prime[j] == ){
                 mu[i * prime[j]] = ;
                 break;
             }else mu[i * prime[j]] = -mu[i];
         }
     }
 }

 int main(void){
     int t, a, b, c, d, k;
     Moblus();
     scanf("%d", &t);
     for(int i = ; i <= t; i++){
         scanf("%d%d%d%d%d", &a, &b, &c, &d, &k);
         if(k == ){
             printf("Case %d: 0\n", i);
             continue;
         }
         if(b > d) swap(b, d);
         b /= k;
         d /= k;
         ll ans1 = , ans2 = ;
         for(int j = ; j <= b; j++){
             ans1 += (ll)mu[j] * (b / j) * (d / j);
         }
         for(int j = ; j <= b; j++){
             ans2 += (ll)mu[j] * (b / j) * (b / j);
         }
         printf("Case %d: %lld\n",i, ans1 - (ans2 >> ));
     }
     return ;
 }

因为对于某些连续的 i 会有 b / i, d / i 是相同的, 这种情况可以通过前缀和优化一下 .

优化代码:

 #include <iostream>
 #include <stdio.h>
 #include <string.h>
 #define ll long long
 using namespace std;

 const int MAXN = 1e5 + ;

 bool check[MAXN];
 int prime[MAXN], mu[MAXN], sum[MAXN];

 void Moblus(void){
     memset(check, false, sizeof(check));
     int tot = ;
     mu[] = ;
     sum[] = ;
     for(int i = ; i < MAXN; i++){
         if(!check[i]){
             prime[tot++] = i;
             mu[i] = -;
         }
         for(int j = ; j < tot && prime[j] * i < MAXN; j++){
             check[i * prime[j]] = true;
             if(i % prime[j] == ){
                 mu[i * prime[j]] = ;
                 break;
             }else mu[i * prime[j]] = -mu[i];
         }
         sum[i] = sum[i - ] + mu[i];
     }
 }

 ll solve(int b, int d){
     ll ans = ;
     for(int i = , la = ; i <= b; i = la + ){
         la = min(b / (b / i), d / (d / i));
         ans += (ll)(sum[la] - sum[i - ]) * (b / i) * (d / i);
     }
     return ans;
 }

 int main(void){
     Moblus();
     int t, a, b, c, d, k;
     scanf("%d", &t);
     for(int i = ; i <= t; i++){
         scanf("%d%d%d%d%d", &a, &b, &c, &d, &k);
         if(k == ){
             printf("Case %d: 0\n", i);
             continue;
         }
         b /= k;
         d /= k;
         if(b > d) swap(b, d);
         ll ans1 = solve(b, d);
         ll ans2 = solve(b, b);
          printf("Case %d: %lld\n", i, ans1 - (ans2 >> ));
     }
     return ;
 }

就本题而言时间从 31ms 优化到了 15 ms .