【JZOJ A组】b

Description

给定 n 个正整数序列 ，每个序列长度为m。
选择至少 1 个序列，在每个被选择的序列中选择一个元素，求出所有被选择的元素的 gcd。
求所有方案的结果之和，答案对 1e9+7 取模。两种方案不同，当且仅当存在至少一个元素，在一种方案中被选择，在另一种中没有。

Input

第一行，两个正整数n，m。
接下来n 行，每行m 个正整数，第i 行代表序列 。

Output

第一行，一个整数，代表答案对 1e9+7 取模的结果。

Sample Input

见下发文件

Sample Output

见下发文件

Data Constraint

思路

这里记 a 的最大值为 x。
将问题转化为“对 i∈[1,1e5]，求多少种选择方案使得 gcd=i。
继续转化为“对 i∈[1,1e5]，求多少种选择方案使得 i|gcd。这里可以O(x)或直接暴力O(n log n)容斥回去。
等价于“对 i∈[1,1e5]，求多少种选择方案使得选的所有数均为 i 的倍数。
O(n(m + x log x))处理出第 i 行有多少个数是 j 的倍数，记为cnt i,j。
那么，共有∏（1–n） (cnt i,j + 1) − 1种方案，使得选的所有数均为 j 的倍数，这一步为O(nm)

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
const int N=37,M=1e5+77,inf=1e5+77,mod=1e9+7;
int n,m;
long long a[N][M],sum[M][477],f[N][M],g[M],ans,mx;
int main()
{
    freopen("b.in","r",stdin);
    freopen("b.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++) for(int j=1; j<=m; j++)
    {
        scanf("%d",&a[i][j]); mx=max(a[i][j],mx);
    }   
    for(int i=1; i<=inf; i++)
        for(int j=1; j<=sqrt(i); j++)
            if(i%j==0)
            {
                sum[i][++sum[i][0]]=j;
                if(j*j!=i) sum[i][++sum[i][0]]=i/j;
            }
    for(int i=1; i<=n; i++)
        for(int j=1; j<=m; j++)
            for(int k=1; k<=sum[a[i][j]][0]; k++)
                f[i][sum[a[i][j]][k]]++;
    for(int i=1; i<=mx; i++)
    {
        g[i]=1;
        for(int j=1; j<=n; j++) g[i]=(g[i]*(f[j][i]+1))%mod;
        g[i]=(g[i]-1+mod)%mod;
    }
    for(int i=mx; i>=1; i--)
    {
        for (int j=2;j<=mx/i;j++) g[i]=(g[i]-g[i*j]+mod)%mod;
        ans=(ans+g[i]*i%mod)%mod;
    }
    printf("%d",ans);
}