hdu4746:2013杭州网络赛I 莫比乌斯反演

题意:

有5000组询问,每组询问求有多少i,j满足i∈[1,n],j∈[1,m]且gcd(i,j)的质因子数目<=p。 n,m<=500000

思路:

首先预处理出每个数的质因子数目分别等于多少,则问题转化为求给定区间内,gcd等于某一堆数的i,j有多少组

发现很像一个基础莫比乌斯反演题:hdu1695。但是此题在某组询问中可能要处理很多个gcd,所以需要进行一些预处理

我们首先筛出每个数的莫比乌斯函数和它的质因子个数

通过容斥的公式可以看出如果要求的gcd为d,那么d*i的倍数对答案的贡献为 mobi[i]。

这样就可以预处理出每个p对应位置的莫比乌斯函数系数之和p[19][500000]

注意这里容易想到p是小于等于18的,因为50W之内的数最多有18个质因子(2^19>500000)

然后对p数组求前缀和,可以使单组查询复杂度变为p*sqrt(n),具体为什么是分块sqrt(n)可以参考hdu1695的题解。交了一发超时了。。

还以为写搓了,后来发现可以再求一次前缀和。。这样单组查询变成了sqrt(n),终于过了

代码:

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 const int maxn=500000;
 4 int mb[maxn+10];
 5 int notprime[maxn+10];
 6 int num[maxn+10];
 7 int prime[maxn];
 8 vector<int>v[20];
 9 int np=0;
10 void setprime()
11 {
12     memset(notprime,0,sizeof(notprime));
13     mb[1]=1;
14     num[1]=0;
15     for(int i=2; i<=maxn; i++)
16     {
17         if(!notprime[i])
18         {
19             prime[np++]=i;
20             num[i]=1;
21             mb[i]=-1;
22         }
23         for(int j=0; j<np&&i*prime[j]<=maxn; j++)
24         {
25             notprime[i*prime[j]]=1;
26             num[i*prime[j]]=num[i]+1;
27             //v[num[i]+1].push_back(i*prime[j]);
28             if(i%prime[j]==0)
29             {
30                 mb[i*prime[j]]=0;
31                 break;
32             }
33             else
34             {
35                 mb[i*prime[j]]=-mb[i];
36             }
37         }
38     }
39 }
40 int p[20][500050];
41 int sum[20][500050];
42 int f[20][500050];
43 int main()
44 {
45     freopen("in.txt","r",stdin);
46     setprime();
47     for(int i=1; i<=maxn; i++)
48     {
49         for(int j=1; i*j<=maxn; j++)
50         {
51             p[num[i]][i*j]+=mb[j];
52         }
53     }
54     for(int i=0; i<=18; i++)
55     {
56         sum[i][0]=0;
57         for(int j=1; j<=maxn; j++)
58         {
59             sum[i][j]=sum[i][j-1]+p[i][j];
60         }
61     }
62     for(int i=1; i<=maxn; i++)
63     {
64         f[0][i]=sum[0][i];
65         for(int j=1;j<=18;j++)
66         {
67             f[j][i]=f[j-1][i]+sum[j][i];
68         }
69     }
70     int n,m,k,q;
71     scanf("%d",&q);
72     while(q--)
73     {
74         long long ans=0;
75         scanf("%d%d%d",&n,&m,&k);
76         if(k>18)
77         {
78             printf("%I64d\n",(long long)n*m);
79             continue;
80         }
81         if(n>m)
82             swap(n,m);
83         for(int j=1; j<=n; j++)
84         {
85             int to=min(n/(n/j),m/(m/j));
86             ans+=(long long)(n/j)*(m/j)*(f[k][to]-f[k][j-1]);
87             j=to;
88         }
89         printf("%I64d\n",ans);
90     }
91     return 0;
92 }
View Code

 

posted @ 2015-09-23 21:10  PlasticSpirit  阅读(172)  评论(0编辑  收藏  举报