結果

問題 No.1358 [Zelkova 2nd Tune *] 語るなら枚数を...
コンテスト
ユーザー tonyu0
提出日時 2021-01-23 08:50:37
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 435 ms / 2,000 ms
コード長 1,514 bytes
コンパイル時間 14,217 ms
コンパイル使用メモリ 390,944 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-30 01:25:54
合計ジャッジ時間 17,336 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::*;
const MOD: i64 = 1_000_000_007;

fn ext_gcd(a: i64, b: i64, x: &mut i64, y: &mut i64) -> i64 {
    if b == 0 {
        *x = 1;
        *y = 0;
        a
    } else {
        let d = ext_gcd(b, a % b, y, x);
        *y -= (a / b) * *x;
        d
    }
}

fn calc(a: i64, mut b: i64, mut c: i64, y: i64) -> i64 {
    let mut ans = 0;
    let mut x0 = 0;
    let mut y0 = 0;
    let g = ext_gcd(b, c, &mut x0, &mut y0);
    b /= g;
    c /= g;
    for i in 0..=y / a {
        let mut d = y - i * a;
        if d % g != 0 {
            continue;
        }
        d /= g;

        // d = b * x + c * y
        // b(x - x0) = -c(y - y0)
        // x = x0 + c*i >= 0
        // y = y0 - b*i >= 0
        // -x0 / c <= i <= y0 / b
        if x0 < y0 {
            std::mem::swap(&mut x0, &mut y0);
            std::mem::swap(&mut b, &mut c);
        }
        ans = (ans + (x0 * d) / c - (-y0 * d + b - 1) / b + 1) % MOD;
    }
    ans
}

fn main() {
    let mut s: String = String::new();
    std::io::stdin().read_to_string(&mut s).ok();
    let mut itr = s.trim().split_whitespace();
    let t: usize = itr.next().unwrap().parse().unwrap();
    for _ in 0..t {
        let mut a = vec![0; 3];
        a[0] = itr.next().unwrap().parse().unwrap();
        a[1] = itr.next().unwrap().parse().unwrap();
        a[2] = itr.next().unwrap().parse().unwrap();
        let y: i64 = itr.next().unwrap().parse().unwrap();
        a.sort_by_key(|x| -x);
        println!("{}", calc(a[0], a[1], a[2], y));
    }
}
0