結果

問題 No.1358 [Zelkova 2nd Tune *] 語るなら枚数を...
ユーザー tonyu0tonyu0
提出日時 2021-01-23 08:50:37
言語 Rust
(1.72.1)
結果
AC  
実行時間 409 ms / 2,000 ms
コード長 1,514 bytes
コンパイル時間 1,462 ms
コンパイル使用メモリ 147,388 KB
実行使用メモリ 4,380 KB
最終ジャッジ日時 2023-08-28 17:51:15
合計ジャッジ時間 4,158 ms
ジャッジサーバーID
(参考情報)
judge11 / judge14
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 1 ms
4,380 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 1 ms
4,380 KB
testcase_04 AC 1 ms
4,376 KB
testcase_05 AC 1 ms
4,380 KB
testcase_06 AC 4 ms
4,376 KB
testcase_07 AC 3 ms
4,376 KB
testcase_08 AC 3 ms
4,380 KB
testcase_09 AC 3 ms
4,376 KB
testcase_10 AC 3 ms
4,376 KB
testcase_11 AC 365 ms
4,376 KB
testcase_12 AC 322 ms
4,376 KB
testcase_13 AC 409 ms
4,376 KB
testcase_14 AC 294 ms
4,376 KB
testcase_15 AC 319 ms
4,380 KB
testcase_16 AC 234 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

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