#1169171 (D) No.2211 Frequency Table of GCD

• 問題: No.2211 Frequency Table of GCD
• ユーザー: InTheBloom
• 提出日時: 2026-06-16 18:08:53
• 言語: D (dmd 2.112.0)
• 結果: WA
• 実行時間: -
• コード長: 4,881 bytes
• コンパイル時間: 4,861 ms
• コンパイル使用メモリ: 205,056 KB
• 実行使用メモリ: 14,336 KB
• 最終ジャッジ日時: 2026-06-16 18:09:02
• 合計ジャッジ時間: 7,039 ms
• ジャッジサーバーID （参考情報）: judge3_0 / judge1_0

ファイルパターン	結果
sample	AC * 2 WA * 1
other	AC * 2 WA * 24

ソースコード

import std;

void main () {
    const long MOD = 998244353;
    int N, M;
    readln.read(N, M);
    auto A = readln.split.to!(int[]);

    // 倍数ゼータ変換が使えそう？
    // gcd = iとして、まずiの倍数が含まれるのは必要
    // このままだとgcdがiの何倍かのものも含まれるので、約数包除をやる

    auto mul = new int[](M + 1);
    foreach (a; A) {
        mul[a]++;
    }

    alias ls = LinearSieve;
    ls.build(M + 10);

    auto ps = new int[](0);
    foreach (i; 2 .. M + 1) {
        if (ls.is_prime(i)) {
            ps ~= i;
        }
    }

    foreach (p; ps) {
        const int up = M / p;
        foreach_reverse (i; 1 .. up + 1) {
            mul[i] += mul[i * p];
        }
    }

    auto pow2 = new long[](N + 1);
    pow2[0] = 1;
    foreach (i; 0 .. N) {
        pow2[i + 1] = 2 * pow2[i];
    }

    auto ans = new long[](M + 1);
    foreach (i; 1 .. M + 1) {
        ans[i] = pow2[mul[i]] - 1;
    }

    // 倍数メビウス変換
    foreach (p; ps) {
        const int up = M / p;
        foreach (i; 1 .. up + 1) {
            ans[i] -= ans[i * p];
        }
    }

    writefln("%(%s\n%)", ans[1 .. $]);
}

void read (T...) (string S, ref T args) {
    import std.conv : to;
    import std.array : split;
    auto buf = S.split;
    foreach (i, ref arg; args) {
        arg = buf[i].to!(typeof(arg));
    }
}

import std.typecons : Tuple, tuple;

class LinearSieve {
    /// methods
    /// - void build (ulong N_)
    /// - Tuple!(long, long)[] prime_factors (ulong N_)
    /// - bool is_prime (ulong N_)
    /// - long[] divisors (ulong N_)

    private:
        static int N = 0;
        static int[] lpf;
        static int[] primes;
        static int[] lpf_ord;
        static int[] lpf_pow;

    import std.conv : to;
    import std.format : format;

    public:
        @disable this () {}

        static void build (ulong N_)
        in {
            assert(2 <= N_ && N_ <= int.max, format("Argument N_ = %s does not meet condition.", N_));
        }
        do {
            // Linear sieve.
            if (N+1 <= lpf.length) return;
            N = N_.to!int;

            primes.length = 0;
            lpf.length = N+1;

            lpf[0] = lpf[1] = 1;

            for (int i = 2; i <= N; i++) {
                if (lpf[i] == 0) {
                    lpf[i] = i;
                    primes ~= i;
                }

                foreach (p; primes) {
                    if (lpf[i] < p) break;
                    if (N < 1L * i * p) break;
                    lpf[i * p] = p;
                }
            }

            // Precomputation of prime factorization.
            lpf_ord.length = lpf_pow.length = N+1;
            lpf_pow[] = 1;

            for (int i = 2; i <= N; i++) {
                int prev = i / lpf[i];

                if (lpf[i] == lpf[prev]) {
                    lpf_ord[i] = lpf_ord[prev] + 1;
                    lpf_pow[i] = lpf_pow[prev] * lpf[i];
                }
                else {
                    lpf_ord[i] = 1;
                    lpf_pow[i] = lpf[i];
                }
            }
        }

        static Tuple!(long, long)[] prime_factors (ulong N_)
        in {
            assert(2 <= N_ && N_ <= N, format("Argument N_ = %s is not out of range. The valid range is [2, %s].", N_, N));
        }
        do {
            int n = N_.to!int;
            Tuple!(long, long)[] res;

            while (1 < n) {
                res ~= tuple(1L*lpf[n], 1L*lpf_ord[n]);
                n /= lpf_pow[n];
            }

            return res;
        }

        static bool is_prime (ulong N_)
        in {
            assert(2 <= N_ && N_ <= N, format("Argument N_ = %s is not out of range. The valid range is [2, %s].", N_, N));
        }
        do {
            int N = N_.to!int;
            return lpf[N] == N;
        }

        static long[] divisors (ulong N_)
        in {
            assert(N_ <= N, format("Argument N_ = %s is not out of range. The valid range is [2, %s].", N_, N));
        }
        do {
            if (N_ == 1) return [1L];

            import std.container : SList;
            import std.algorithm : sort;

            auto fac = prime_factors(N_);

            static SList!(Tuple!(int, long)) Q;
            Q.insertFront(tuple(0, 1L)); // (処理済み階層, 値)

            long[] res;
            while (!Q.empty) {
                auto h = Q.front; Q.removeFront;
                if (h[0] == fac.length) {
                    res ~= h[1];
                    continue;
                }

                auto p = fac[h[0]];
                long prod = 1;
                foreach (i; 0..p[1] + 1) {
                    Q.insertFront(tuple(h[0] + 1, h[1] * prod));
                    prod *= p[0];
                }
            }

            res.sort;
            return res;
        }
}