結果

問題 No.187 中華風 (Hard)
ユーザー suibakasuibaka
提出日時 2018-04-14 14:49:06
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 105 ms / 3,000 ms
コード長 3,323 bytes
コンパイル時間 2,479 ms
コンパイル使用メモリ 217,796 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-26 22:11:53
合計ジャッジ時間 4,717 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 16 ms
5,248 KB
testcase_01 AC 16 ms
5,376 KB
testcase_02 AC 84 ms
5,376 KB
testcase_03 AC 82 ms
5,376 KB
testcase_04 AC 93 ms
5,376 KB
testcase_05 AC 96 ms
5,376 KB
testcase_06 AC 99 ms
5,376 KB
testcase_07 AC 95 ms
5,376 KB
testcase_08 AC 78 ms
5,376 KB
testcase_09 AC 76 ms
5,376 KB
testcase_10 AC 77 ms
5,376 KB
testcase_11 AC 97 ms
5,376 KB
testcase_12 AC 97 ms
5,376 KB
testcase_13 AC 16 ms
5,376 KB
testcase_14 AC 16 ms
5,376 KB
testcase_15 AC 103 ms
5,376 KB
testcase_16 AC 105 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 13 ms
5,376 KB
testcase_19 AC 3 ms
5,376 KB
testcase_20 AC 76 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 97 ms
5,376 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

using ll = long long;

template <typename T>
T extgcd(T a, T b, T& x, T& y) {
    T g = a; x = 1; y = 0;
    if(b != 0) {
        g = extgcd(b, a % b, y, x);
        y -= (a / b) * x;
    }
    return g;
}

template <typename T>
T mod_inv(T a, T m) {
    T x, y;
    extgcd(a, m, x, y);
    return (m + x % m) % m;
}

ll mod_pow(ll a, ll n, const ll mod) {
    ll res = 1;
    ll p = a % mod;
    while(n > 0) {
        if(n & 1) (res *= p) %= mod;
        (p *= p) %= mod;
        n >>= 1;
    }
    return res;
}

ll garner(std::vector<ll> m, std::vector<ll> u, int mod) {
    const int n = m.size();
    std::vector<ll> inv_prod(n);
    {
        for(int i = 1; i < n; ++i) {
            ll prod = m[0] % m[i];
            for(int j = 1; j < i; ++j) {
                prod = (prod * m[j]) % m[i];
            }
            inv_prod[i] = mod_inv(prod, m[i]);
        }
    }

    std::vector<ll> v(n), constants(n);
    v[0] = u[0];
    for(int i = 1; i < n; ++i) {
        ll tmp = v[i - 1];
        for(int j = i - 2; j >= 0; --j) {
            tmp = (tmp * m[j] + v[j]) % m[i];
        }
        v[i] = ((u[i] - tmp) * inv_prod[i]) % m[i];
        if(v[i] < 0) v[i] += m[i];
    }

    ll res = v[n - 1];
    for(int i = n - 2; i >= 0; --i) {
        res = (res * m[i] + v[i]) % mod;
    }
    return res;
}

int main() {
    constexpr int mod = 1e9 + 7;
    int n;
    cin >> n;
    vector<ll> a(n), b(n);
    for(int i = 0; i < n; ++i) {
        cin >> a[i] >> b[i];
    }

    if(count(begin(a), end(a), 0) == n) {
        for(int i = 0; i < n; ++i) {
            for(int j = i + 1; j < n; ++j) {
                const ll g = __gcd(b[i], b[j]);
                b[j] /= g;
            }
        }

        ll ans = 1;
        for(int i = 0; i < n; ++i) {
            ans = (ans * b[i]) % mod;
        }
        cout << ans << endl;
        return 0;
    }

    constexpr int p_max = 40000;
    vector<int> primes;
    vector<bool> is_prime(p_max, true);
    is_prime[0] = is_prime[1] = false;
    for(int i = 2; i < p_max; ++i) {
        if(is_prime[i]) {
            primes.push_back(i);
            for(int j = i + i; j < p_max; j += i) {
                is_prime[j] = false;
            }
        }
    }

    set<int> use_p;
    for(int i = 0; i < n; ++i) {
        int v = b[i];
        for(auto p : primes) {
            if(v % p == 0) {
                use_p.insert(p);
                while(v % p == 0) v /= p;
            }
        }
        if(v > 1) use_p.insert(v);
    }

    for(auto p : use_p) {
        vector<ll> tmp(n, 1);
        int max_pos = 0;
        for(int i = 0; i < n; ++i) {
            int v = b[i];
            while(v % p == 0) {
                v /= p;
                tmp[i] *= p;
            }
            if(tmp[max_pos] < tmp[i]) max_pos = i;
        }
        for(int i = 0; i < n; ++i) {
            if(i == max_pos) continue;
            if(a[i] % tmp[i] != a[max_pos] % tmp[i]) {
                cout << -1 << endl;
                return 0;
            }
            b[i] /= tmp[i];
            a[i] %= b[i];
        }
    }

    vector<ll> aa, bb;
    for(int i = 0; i < n; ++i) {
        if(b[i] > 1) {
            aa.push_back(a[i]);
            bb.push_back(b[i]);
        }
    }

    cout << garner(bb, aa, mod) << endl;
}
0