結果

問題 No.886 Direct
ユーザー hasegawa1hasegawa1
提出日時 2021-08-24 00:50:22
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 212 ms / 4,000 ms
コード長 5,266 bytes
コンパイル時間 1,118 ms
コンパイル使用メモリ 92,676 KB
実行使用メモリ 98,488 KB
最終ジャッジ日時 2024-04-25 21:34:21
合計ジャッジ時間 6,229 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 80 ms
26,664 KB
testcase_01 AC 76 ms
26,820 KB
testcase_02 AC 72 ms
26,644 KB
testcase_03 AC 74 ms
27,048 KB
testcase_04 AC 73 ms
26,648 KB
testcase_05 AC 73 ms
26,644 KB
testcase_06 AC 71 ms
26,668 KB
testcase_07 AC 74 ms
26,756 KB
testcase_08 AC 78 ms
26,644 KB
testcase_09 AC 79 ms
26,784 KB
testcase_10 AC 75 ms
26,700 KB
testcase_11 AC 68 ms
26,732 KB
testcase_12 AC 68 ms
26,756 KB
testcase_13 AC 78 ms
26,728 KB
testcase_14 AC 71 ms
26,664 KB
testcase_15 AC 67 ms
26,740 KB
testcase_16 AC 69 ms
26,824 KB
testcase_17 AC 71 ms
27,256 KB
testcase_18 AC 72 ms
27,248 KB
testcase_19 AC 73 ms
27,208 KB
testcase_20 AC 68 ms
27,008 KB
testcase_21 AC 68 ms
27,516 KB
testcase_22 AC 70 ms
27,296 KB
testcase_23 AC 132 ms
60,332 KB
testcase_24 AC 139 ms
63,388 KB
testcase_25 AC 114 ms
48,976 KB
testcase_26 AC 127 ms
56,504 KB
testcase_27 AC 190 ms
93,484 KB
testcase_28 AC 190 ms
90,548 KB
testcase_29 AC 204 ms
97,332 KB
testcase_30 AC 206 ms
98,488 KB
testcase_31 AC 209 ms
98,420 KB
testcase_32 AC 210 ms
97,188 KB
testcase_33 AC 204 ms
97,808 KB
testcase_34 AC 212 ms
97,376 KB
testcase_35 AC 211 ms
97,396 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <numeric>
#include <cstdint>
#include <cassert>
#include "atcoder/modint"

using namespace std;
using mint = atcoder::modint1000000007;

class Sieve {
private:
    const int _n;
    std::vector<int> _max_prime_factor;
    std::vector<int> _euler_phi;
public:
    explicit Sieve(int n = 2'000'000): _n(n+1), _max_prime_factor(_n), _euler_phi(_n) {
        iota(_euler_phi.begin(), _euler_phi.end(), 0);
        for(int i=2; i<_n; i++) {
            if(_max_prime_factor[i]) continue;
            for(int j=i; j<_n; j+=i) {
                _max_prime_factor[j] = i;
                _euler_phi[j] -= _euler_phi[j]/i;
            }
        }
    }

    bool is_prime(int64_t n) const {
        assert(n > 0);
        if(n < _n) return (_max_prime_factor[n] == n);
        return atcoder::internal::is_prime_constexpr(n);
    }

    std::vector<std::pair<int64_t,int64_t>> prime_factorize(int64_t n) const {
        assert(n > 0);
        std::vector<std::pair<int64_t,int64_t>> res;
        if(n < _n) {
            while(n > 1) {
                int64_t p = _max_prime_factor[n];
                int64_t exp = 0;
                while(_max_prime_factor[n] == p) {
                    n /= p;
                    ++exp;
                }
                res.emplace_back(p, exp);
            }
            std::reverse(res.begin(), res.end());
        } else {
            for(int64_t i=2; i*i<=n; i++) {
                if(n%i == 0) {
                    int exp = 0;
                    while(n%i == 0) {
                        n /= i;
                        ++exp;
                    }
                    res.emplace_back(i, exp);
                }
            }
            if(n != 1) res.emplace_back(n, 1);
        }
        return res;
    }

    std::vector<int64_t> divisors(int64_t n) const {
        assert(n > 0);
        std::vector<int64_t> res = {1};
        for(const auto [p, exp]: prime_factorize(n)) {
            int sz = res.size();
            for(int i=0; i<sz; i++) {
                int64_t now = res[i];
                for(int j=0; j<exp; j++) {
                    now *= p;
                    res.emplace_back(now);
                }
            }
        }
        sort(res.begin(), res.end());
        return res;
    }

    int64_t euler_phi(int64_t n) const {
        assert(n > 0);
        if(n < _n) return _euler_phi[n];
        int64_t res = n;
        for(const auto [p, exp]: prime_factorize(n)) {
            res -= res/p;
        }
        return res;
    }

    template<typename T>
    std::vector<T> divisor_transform(std::vector<T> v) {
        int n = v.size();
        assert(n <= _n);
        for(int i=2; i<n; i++) {
            if(is_prime(i)) {
                for(int j=1; i*j<n; j++) {
                    v[j * i] += v[j];
                }
            }
        }
        return v;
    }

    template<typename T>
    std::vector<T> inverse_divisor_transform(std::vector<T> v) {
        int n = v.size();
        assert(n <= _n);
        for(int i=2; i<n; i++) {
            if(is_prime(i)) {
                for(int j=(n-1)/i; j>0; j--) {
                    v[j * i] -= v[j];
                }
            }
        }
        return v;
    }

    template<typename T>
    std::vector<T> multiple_transform(std::vector<T> v) {
        int n = v.size();
        assert(n <= _n);
        for(int i=2; i<n; i++) {
            if(is_prime(i)) {
                for(int j=(n-1)/i; j>0; j--) {
                    v[j] += v[j * i];
                }
            }
        }
        return v;
    }


    template<typename T>
    std::vector<T> inverse_multiple_transform(std::vector<T> v) {
        int n = v.size();
        assert(n <= _n);
        for(int i=2; i<n; i++) {
            if(is_prime(i)) {
                for(int j=1; i*j<n; j++) {
                    v[j] -= v[j * i];
                }
            }
        }
        return v;
    }

    template<typename T>
    std::vector<T> gcd_convolution(const std::vector<T> &a, const std::vector<T> &b) {
        assert(a.size() == b.size());
        auto sum_a = multiple_transform(a);
        auto sum_b = multiple_transform(b);
        std::vector<T> sum_c;
        std::transform(sum_a.begin(), sum_a.end(), sum_b.begin(), std::back_inserter(sum_c), std::multiplies<T>());
        return inverse_multiple_transform(sum_c);
    }

    template<typename T>
    std::vector<T> lcm_convolution(const std::vector<T> &a, const std::vector<T> &b) {
        assert(a.size() == b.size());
        auto sum_a = divisor_transform(a);
        auto sum_b = divisor_transform(b);
        std::vector<T> sum_c;
        std::transform(sum_a.begin(), sum_a.end(), sum_b.begin(), std::back_inserter(sum_c), std::multiplies<T>());
        return inverse_divisor_transform(sum_c);
    }
};

int main(void) {
    cin.tie(nullptr);
    ios_base::sync_with_stdio(false);

    Sieve s(3'000'000);

    long long int H, W;
    cin >> H >> W;

    int M = max(H, W);
    vector<mint> h(M+1), w(M+1);
    for(int i=1; i<H; i++) {
        h[i] = H-i;
    }
    for(int i=1; i<W; i++) {
        w[i] = W-i;
    }

    mint ans = (H-1)*W + H*(W-1) + 2*s.gcd_convolution(h, w)[1];
    cout << ans.val() << endl;
}
0