結果

問題 No.1358 [Zelkova 2nd Tune *] 語るなら枚数を...
コンテスト
ユーザー tonyu0
提出日時 2021-01-22 23:45:19
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 2,217 bytes
コンパイル時間 15,063 ms
コンパイル使用メモリ 380,024 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-28 09:25:12
合計ジャッジ時間 28,701 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10 WA * 2 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::*;
const MOD: i64 = 1_000_000_07;
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, b: i64, c: i64, y: i64) -> i64 {
    let mut ans = 0;
    for i in 0..=y / a {
        let d = y - i * a;

        // d = b * x + c * y
        let mut x0 = 0;
        let mut y0 = 0;
        let g = ext_gcd(b, c, &mut x0, &mut y0);
        if d % g != 0 {
            continue;
        }
        x0 *= d / g;
        y0 *= d / g;
        let bs = b / g;
        let cs = c / g;
        if x0 < 0 {
            // while x0 < 0 {
            //     x0 += cs;
            //     y0 -= bs;
            // }
            let u = (-x0 + cs - 1) / cs;
            y0 -= u * bs;
            // while y0 >= 0 {
            //     ans += 1;
            //     y0 -= bs;
            // }
            if y0 >= 0 {
                ans = (ans + (y0 / bs + 1) % MOD) % MOD;
            }
        } else {
            // while y0 < 0 {
            //     y0 += bs;
            //     x0 -= cs;
            // }
            let u = (-y0 + bs - 1) / bs;
            x0 -= u * cs;
            // while x0 >= 0 {
            //     ans += 1;
            //     x0 -= cs;
            // }
            if x0 >= 0 {
                ans = (ans + (x0 / cs + 1) % MOD) % 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 n: i64 = itr.next().unwrap().parse().unwrap();
        let k: i64 = itr.next().unwrap().parse().unwrap();
        let h: i64 = itr.next().unwrap().parse().unwrap();
        let y: i64 = itr.next().unwrap().parse().unwrap();
        let nc = y / n;
        let kc = y / k;
        if nc <= 1000000 {
            println!("{}", calc(n, k, h, y));
        } else if kc <= 1000000 {
            println!("{}", calc(k, h, n, y));
        } else {
            println!("{}", calc(h, n, k, y));
        }
    }
}
0