結果

問題 No.1680 Sum and Difference
ユーザー tonyu0
提出日時 2021-09-18 10:59:55
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 1,047 bytes
コンパイル時間 13,610 ms
コンパイル使用メモリ 377,076 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-30 08:17:09
合計ジャッジ時間 14,718 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 WA * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::*;
const MOD: i64 = 1000000007;

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 mut a: i64 = itr.next().unwrap().parse::<i64>().unwrap();
    let mut b: i64 = itr.next().unwrap().parse::<i64>().unwrap();

    // -A <= x+y <= A
    // -B <= x-y <= B
    // -A-y <= x <= A-y
    // -B+y <= x <= B+y
    // max(-A-y,-B+y) <= x <= min(A-y,B+y)
    // -A-y=-B+y -> y=(-A+B)/2, x=(-A-B)/2 - bottom
    // A-y=B+y -> y=(A-B)/2, x=(A+B)/2 - top
    // -B+y=A-y -> y=(A+B)/2, x=(A-B)/2 - right
    // -A-y=B+y -> y=(-A-B)/2, x=(-A+B)/2 - left
    if a == 0 && b == 0 {
        println!("1");
    } else if a == 0 {
        println!("{}", b);
    } else if b == 0 {
        println!("{}", a);
    } else {
        a %= MOD;
        b %= MOD;
        let mut ans = ((a + 1) * (b + 1) % MOD + a * b % MOD) % MOD;
        if (a & 1) ^ (b & 1) == 1 {
            ans = (MOD + ans - 1) % MOD;
        }
        println!("{}", ans);
    }
}
0