結果

問題 No.186 中華風 (Easy)
ユーザー pred
提出日時 2021-03-30 16:28:39
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,409 bytes
コンパイル時間 2,165 ms
コンパイル使用メモリ 198,764 KB
最終ジャッジ日時 2025-01-20 00:55:55
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

std::pair<long long, long long> crt(std::vector<long long> &a, std::vector<long long> &m){
    assert(a.size() == m.size());
    long long ax = a[0], mx = m[0];

    auto extgcd = [](long long r, long long s) -> std::tuple<long long, long long, long long>{
        bool rev = false;
        if(s < r){
            std::swap(r, s);
            rev = true;
        }
        long long a = 1, b = 0, c = 0, d = 1, k;
        while(s){
            k = r / s;
            a -= k * c;
            b -= k * d;
            std::swap(a, c);
            std::swap(b, d);
            r %= s;
            std::swap(r, s);
        }
        if(rev) return std::make_tuple(b, a, r);
        else return std::make_tuple(a, b, r);
    };

    for(int i = 1, sz = a.size(); i < sz; i++){
        long long my = m[i];
        long long ay = (a[i] % my + my) % my;

        auto [rx, ry, g] = extgcd(mx, my);
        if((ay - ax) % g != 0) return std::make_pair(0, 0);

        long long l = my / g;
        ax = ax + ((ay - ax) / g) % l * rx % l * mx;
        mx *= l;
        if(ax < 0) ax += mx;
    }

    return std::make_pair(ax, mx);
}

int main(){
    std::vector a(3, 0LL), b(3, 0LL);
    for(int i = 0; i < 3; i++) std::cin >> a[i] >> b[i];
    auto res = crt(a, b);
    if(res.second == 0) res.first = -1;
    if(res.first == 0) res.first = res.second;
    std::cout << res.first << "\n";
}
0