結果

問題 No.3396 ChRisTmas memory
コンテスト
ユーザー akakimidori
提出日時 2025-12-03 00:29:20
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 1,135 ms / 4,000 ms
コード長 3,295 bytes
コンパイル時間 13,099 ms
コンパイル使用メモリ 399,728 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-12-03 00:29:47
合計ジャッジ時間 26,123 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 40
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: type alias `Map` is never used
   --> src/main.rs:102:6
    |
102 | type Map<K, V> = BTreeMap<K, V>;
    |      ^^^
    |
    = note: `#[warn(dead_code)]` on by default

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

warning: type alias `Deque` is never used
   --> src/main.rs:104:6
    |
104 | type Deque<T> = VecDeque<T>;
    |      ^^^^^

ソースコード

diff #
raw source code 

fn run<W: Write>(sc: &mut scanner::Scanner, out: &mut std::io::BufWriter<W>) {
    let q: u32 = sc.next();
    let mut rad = vec![1];
    let mut x = vec![0];
    let mut valid = vec![true];
    for _ in 0..q {
        let op = sc.next::<u32>();
        if op == 1 {
            let m = sc.next::<i64>();
            let r = sc.next::<i64>();
            let mut prod = 1;
            let mut fix = 0;
            for (p, x) in rad.iter().zip(x.iter()) {
                fix = (fix + *x * prod) % m;
                prod = prod * *p % m;
            }
            let r = (r - fix).rem_euclid(m);
            // prod * p + m * q = gcd(m, prod) (mod m)
            let (p, q) = calc(prod, m);
            let g = p * prod + m * q;
            if !*valid.last().unwrap() || r % g != 0 {
                valid.push(false);
                rad.push(m);
                x.push(0);
            } else {
                valid.push(true);
                let m = m / g;
                rad.push(m);
                let v = (p * (r / g)).rem_euclid(m);
                x.push(v);
            }
        } else if op == 2 {
            let k = sc.next::<usize>();
            for _ in 0..k {
                rad.pop();
                x.pop();
                valid.pop();
            }
        } else if op == 3 {
            let m = sc.next::<i64>();
            let ans = if *valid.last().unwrap() {
                let mut s = 0;
                let mut prod = 1;
                for (&p, &x) in rad.iter().zip(x.iter()) {
                    s = (s + prod * x) % m;
                    prod = prod * p % m;
                }
                s
            } else {
                -1
            };
            writeln!(out, "{}", ans).ok();
        }
    }
}

// return (a, b)
// ax + by = gcd(x, y)
fn calc(x: i64, y: i64) -> (i64, i64) {
    if y == 0 {
        return (x.signum(), 0);
    }
    let q = x.div_euclid(y);
    let r = x.rem_euclid(y);
    let p = calc(y, r);
    (p.1, p.0 - q * p.1)
}

// ---------- begin scannner ----------
#[allow(dead_code)]
mod scanner {
    use std::str::FromStr;
    pub struct Scanner<'a> {
        it: std::str::SplitWhitespace<'a>,
    }
    impl<'a> Scanner<'a> {
        pub fn new(s: &'a String) -> Scanner<'a> {
            Scanner {
                it: s.split_whitespace(),
            }
        }
        pub fn next<T: FromStr>(&mut self) -> T {
            self.it.next().unwrap().parse::<T>().ok().unwrap()
        }
        pub fn next_bytes(&mut self) -> Vec<u8> {
            self.it.next().unwrap().bytes().collect()
        }
        pub fn next_chars(&mut self) -> Vec<char> {
            self.it.next().unwrap().chars().collect()
        }
        pub fn next_vec<T: FromStr>(&mut self, len: usize) -> Vec<T> {
            (0..len).map(|_| self.next()).collect()
        }
    }
}
// ---------- end scannner ----------

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

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

fn main() {
    use std::io::Read;
    let mut s = String::new();
    std::io::stdin().read_to_string(&mut s).unwrap();
    let mut sc = scanner::Scanner::new(&s);
    let out = std::io::stdout();
    let mut out = std::io::BufWriter::new(out.lock());
    run(&mut sc, &mut out);
}
0