結果

問題 No.1172 Add Recursive Sequence
ユーザー noshi91noshi91
提出日時 2020-05-09 18:09:49
言語 Rust
(1.77.0)
結果
AC  
実行時間 640 ms / 4,000 ms
コード長 2,164 bytes
コンパイル時間 2,583 ms
コンパイル使用メモリ 158,696 KB
実行使用メモリ 168,960 KB
最終ジャッジ日時 2024-04-17 06:57:46
合計ジャッジ時間 4,717 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 1 ms
5,376 KB
testcase_06 AC 4 ms
5,376 KB
testcase_07 AC 3 ms
5,376 KB
testcase_08 AC 3 ms
5,376 KB
testcase_09 AC 3 ms
5,376 KB
testcase_10 AC 32 ms
5,376 KB
testcase_11 AC 34 ms
5,376 KB
testcase_12 AC 28 ms
5,376 KB
testcase_13 AC 27 ms
5,376 KB
testcase_14 AC 215 ms
20,608 KB
testcase_15 AC 168 ms
17,920 KB
testcase_16 AC 640 ms
168,960 KB
testcase_17 AC 540 ms
166,528 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::Read;

struct Scanner<'a>(std::str::SplitWhitespace<'a>);

impl<'a> Scanner<'a> {
    fn new(s: &'a str) -> Self {
        Self(s.split_whitespace())
    }

    fn next<T>(&mut self) -> T
    where
        T: std::str::FromStr,
        T::Err: std::fmt::Debug,
    {
        let s = self.0.next().expect("found EOF");
        match s.parse() {
            Ok(v) => v,
            Err(msg) => {
                println!(
                    "parse error. T = {}, s = \"{}\": {:?}",
                    std::any::type_name::<T>(),
                    s,
                    msg
                );
                panic!()
            }
        }
    }
}

const MOD: u64 = 1000000007;

fn normalize(mut x: i32) -> u64 {
    if x < 0 {
        x += MOD as i32;
    }
    return x as u64;
}

fn main() {
    let mut s = String::new();
    std::io::stdin().read_to_string(&mut s).unwrap();
    let mut sc = Scanner::new(&s);

    let k = sc.next::<usize>();
    let n = sc.next::<usize>();
    let m = sc.next::<usize>();
    let a = (0..k).map(|_| normalize(sc.next())).collect::<Vec<_>>();
    let c = (0..k).map(|_| normalize(sc.next())).collect::<Vec<_>>();

    let mut ls = vec![0; n + 1];
    let mut rs = vec![Vec::new(); n + 1];
    for _ in 0..m {
        let (l, r) = (sc.next::<usize>(), sc.next::<usize>());
        ls[l] += 1;
        rs[r].push(r - l);
    }

    let step = |seq: &mut Vec<u64>| {
        let next = (0..k).fold(0, |x, i| (x + c[i] * seq[k - 1 - i]) % MOD);
        seq.remove(0);
        seq.push(next);
    };

    let base = {
        let mut state = a.clone();
        (0..n)
            .map(move |_| {
                let res = state.clone();
                step(&mut state);
                res
            })
            .collect::<Vec<_>>()
    };

    let mut now = vec![0; k];

    for i in 0..n {
        step(&mut now);
        for j in 0..k {
            now[j] += ls[i] * a[j];
            now[j] %= MOD;
        }
        for &r in &rs[i] {
            for j in 0..k {
                now[j] += MOD - base[r][j];
                now[j] %= MOD;
            }
        }
        println!("{}", now[0]);
    }
}
0