数列的 GCD [计数问题]

GCD数列的 GCD


Descriptionmathcal{Description}


Solutionmathcal{Solution}

由于有 KK 个位置两数列不相同, 于是 使用 {an}{a_n} 构造 {bn}{b_n},

枚举 dd, 即假设 dd 是已知量,
{an}{a_n}xx 个位置不能整除 dd,
由于2条件2的限制, 这 xx 个位置需要改变,
即至少改变xx个位置, 若 x>Kx>K, 说明无解 .

于是若存在答案, 则 x<=Kcolor{red}{x <= K},
xx 个位置的数要改变, 方案数为
(1)Mdxlfloor frac{M}{d} floor ^x ag{1}
剩下还需改变 KxK-x 个位置上的数, 方案数为
(2)(NNx)Md1Kxdbinom{N}{N-x}*lfloor frac{M}{d}-1 floor^{K-x} ag{2}

( (1) * (2) )tmp[d]=Mdx(NNx)Md1Kx herefore tmp[d] = lfloor frac{M}{d} floor ^x dbinom{N}{N-x}lfloor frac{M}{d}-1 floor^{K-x} ag{ (1) * (2) }

由于 gcdgcddd, 所以要减去 gcdgcdAns[d]Ans[d] 倍数的方案
Ans[d]=tmp[d]tmp[2d]tmp[3d]...tmp[nd]Ans[d]=tmp[d]-tmp[2d]-tmp[3d]-...-tmp[nd]


Code with bugmathcal{Code with bug}

#include<bits/stdc++.h>
#define reg register

int read(){
        char c;
        int s = 0, flag = 1;
        while((c=getchar()) && !isdigit(c))
                if(c == '-'){ c = getchar(), flag = -1; break ; }
        while(isdigit(c)) s = s*10 + c-'0', c = getchar();
        return s * flag;
}

const int maxn = 300005;
const int mod = 1000000007;

int N;
int M;
int K;
int x;
int A[maxn];
int jc[maxn];
int rev[maxn];
int tmp[maxn];
int Ans[maxn];

int C(int n, int m){
        int fz = jc[n];
        int fm = 1ll*jc[n-m]*jc[m] % mod;
        fz = 1ll*fz*rev[fm] % mod;
        return fz;
}

int KSM(int a, int b){
        int s = 1;
        while(b){
                if(b & 1) s = 1ll*s*a % mod;
                a = 1ll*a*a % mod;
                b >>= 1;
        }
        return s;
}

void Init(){
        rev[1] = 1;
        for(reg int i = 2; i < maxn; i ++) rev[i] = ((1ll*(-mod/i) * 1ll*rev[mod%i]) %mod + mod) % mod;
        jc[0] = 1;
        for(reg int i = 1; i < maxn; i ++) jc[i] = 1ll*jc[i-1]*i % mod;
}

int main(){
        freopen("gcd.in", "r", stdin);
        freopen("gcd.out", "w", stdout);
        Init();
        N = read(); M = read(); K = read();
        for(reg int i = 1; i <= N; i ++) A[i] = read();
        for(reg int d = 1; d <= M; d ++){
                x = 0;
                for(reg int i = 1; i <= N; i ++) x += ((A[i]%d) != 0);
                if(x > K){ Ans[d] = 0; continue ; }
                int tmp_1 = KSM(M/d, x), tmp_2 = C(N, N-x), tmp_3 = KSM(M/d-1, K-x);
                tmp[d] = (1ll*tmp_1*tmp_2%mod)*1ll*tmp_3 % mod;
        }
        for(reg int d = 1; d <= M; d ++){
                Ans[d] = tmp[d];
                for(reg int i = 2; i*d <= M; i ++) Ans[d] = ((Ans[d] - tmp[i*d])%mod + mod) % mod;
                printf("%d
", Ans[d]);
        }
        return 0;
}

原文地址:https://www.cnblogs.com/zbr162/p/11822575.html