結果

問題 No.3516 Very Large Range Mod
コンテスト
ユーザー akakimidori
提出日時 2026-04-24 22:51:51
言語 Rust
(1.94.0 + proconio + num + itertools)
コンパイル:
/usr/bin/rustc_custom
実行:
./target/release/main
結果
WA  
実行時間 -
コード長 3,169 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 11,894 ms
コンパイル使用メモリ 206,804 KB
実行使用メモリ 9,216 KB
最終ジャッジ日時 2026-04-24 22:52:11
合計ジャッジ時間 15,016 ms
ジャッジサーバーID
（参考情報） 		judge4_0 / judge2_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other WA * 30
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: unused import: `std::io::Write`
 --> src/main.rs:4:5
  |
4 | use std::io::Write;
  |     ^^^^^^^^^^^^^^
  |
  = note: `#[warn(unused_imports)]` (part of `#[warn(unused)]`) on by default

warning: type alias `Map` is never used
 --> src/main.rs:6:6
  |
6 | type Map<K, V> = BTreeMap<K, V>;
  |      ^^^
  |
  = note: `#[warn(dead_code)]` (part of `#[warn(unused)]`) on by default

warning: type alias `Set` is never used
 --> src/main.rs:7:6
  |
7 | type Set<T> = BTreeSet<T>;
  |      ^^^

ソースコード

diff #
raw source code 

// 尺取りしろという話だがどういうことか

use std::collections::*;
use std::io::Write;

type Map<K, V> = BTreeMap<K, V>;
type Set<T> = BTreeSet<T>;
type Deque<T> = VecDeque<T>;

fn main() {
    input! {
        n: usize,
        k: i64,
        m: i64,
        b: [i64; n],
        c: [i64; n],
    }
    let mut deq = Deque::<(i64, i64)>::new();
    let mut cnt = 0;
    let mut sum = 0;
    let mut ans = 0;
    for (mut c, b) in b.into_iter().zip(c) {
        while deq.len() > 0 && cnt + c > k {
            let (p, q) = deq[0];
            let q = q.min(cnt + c - k);
            // 0 <= i < q, sum + (b - p)i = 0 (mod M)
            let (_x, y, g) = ext_gcd(m, b - p);
            if sum % g == 0 {
                let i = ((-sum / g) * y).rem_euclid(m / g);
                if i < q {
                    ans += (q - i - 1) / (m / g) + 1;
                }
            }
            if q == deq[0].1 {
                deq.pop_front();
            } else {
                deq[0].1 -= q;
            }
            cnt -= q;
            sum = (sum + m - p * q % m) % m;
        }
        if c > k {
            if k * b % m == 0 {
                ans += c - k;
            }
            c = k;
        }
        deq.push_back((b, c));
        cnt += c;
        sum = (sum + b * c) % m;
    }
    if sum % m == 0 {
        ans += 1;
    }
    println!("{}", ans);
}

// ---------- begin input macro ----------
// reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8
#[macro_export]
macro_rules! input {
    (source = $s:expr, $($r:tt)*) => {
        let mut iter = $s.split_whitespace();
        input_inner!{iter, $($r)*}
    };
    ($($r:tt)*) => {
        let s = {
            use std::io::Read;
            let mut s = String::new();
            std::io::stdin().read_to_string(&mut s).unwrap();
            s
        };
        let mut iter = s.split_whitespace();
        input_inner!{iter, $($r)*}
    };
}

#[macro_export]
macro_rules! input_inner {
    ($iter:expr) => {};
    ($iter:expr, ) => {};
    ($iter:expr, $var:ident : $t:tt $($r:tt)*) => {
        let $var = read_value!($iter, $t);
        input_inner!{$iter $($r)*}
    };
}

#[macro_export]
macro_rules! read_value {
    ($iter:expr, ( $($t:tt),* )) => {
        ( $(read_value!($iter, $t)),* )
    };
    ($iter:expr, [ $t:tt ; $len:expr ]) => {
        (0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>()
    };
    ($iter:expr, chars) => {
        read_value!($iter, String).chars().collect::<Vec<char>>()
    };
    ($iter:expr, bytes) => {
        read_value!($iter, String).bytes().collect::<Vec<u8>>()
    };
    ($iter:expr, usize1) => {
        read_value!($iter, usize) - 1
    };
    ($iter:expr, $t:ty) => {
        $iter.next().unwrap().parse::<$t>().expect("Parse error")
    };
}
// ---------- end input macro ----------
// return (a, b, g)
// ax + by = gcd(x, y)
// gcd(x, y) = 0 の時は(0, 0) が返る
pub fn ext_gcd(x: i64, y: i64) -> (i64, i64, i64) {
    if y == 0 {
        return (x.signum(), 0, x.abs());
    }
    let q = x.div_euclid(y);
    let r = x.rem_euclid(y);
    let p = ext_gcd(y, r);
    (p.1, p.0 - q * p.1, p.2)
}
0