結果

問題 No.510 二次漸化式
ユーザー packet0
提出日時 2017-04-30 02:25:01
言語 Rust
(1.83.0 + proconio)
結果
TLE  
実行時間 -
コード長 1,750 bytes
コンパイル時間 12,972 ms
コンパイル使用メモリ 384,592 KB
実行使用メモリ 14,024 KB
最終ジャッジ日時 2024-09-13 19:20:28
合計ジャッジ時間 34,925 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 21 TLE * 1 -- * 12
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: unnecessary parentheses around `for` iterator expression
  --> src/main.rs:27:14
   |
27 |     for i in (2..q+2) {
   |              ^      ^
   |
   = note: `#[warn(unused_parens)]` on by default
help: remove these parentheses
   |
27 -     for i in (2..q+2) {
27 +     for i in 2..q+2 {
   |

ソースコード

diff # 

use std::io::{self, BufRead};

const DIV: usize = 1000000007;

fn main() {
    let stdin = io::stdin();
    let lines: Vec<String> = stdin.lock().lines().map(|x| x.unwrap()).collect();

    let n0: usize = lines[0].parse().unwrap();
    let n1 = n0 + 1;
    let mut x = vec![0; n1];
    let mut y = vec![0; n1];
    let mut a = vec![0; n1];
    let mut b = vec![0; n1];
    let mut b2 = vec![0; n1];
    a[0] = 1;
    b[0] = 1;
    b2[0] = 1;
    let mut ai = 0;
    let mut bi = 0;
    // println!("x{:?}", x);
    // println!("y{:?}", y);
    // println!("a{:?}", a);
    // println!("b{:?}", b);

    let q: usize = lines[1].parse().unwrap();
    for i in (2..q+2) {
        let list: Vec<&str> = lines[i].split_whitespace().collect();
        let v = list[1].parse::<usize>().unwrap();
        match list[0].as_bytes()[0] as char {
            'x' => {
                x[v] = list[2].parse().unwrap();
                ai = 0;
            },
            'y' => {
                y[v] = list[2].parse().unwrap();
                ai = 0;
                bi = 0;
            },
            'a' => {
                while bi < v {
                    let next = bi + 1;
                    b[next] = b[bi] * y[bi] % DIV + 1;
                    b2[next] = b[next] * b[next] % DIV;
                    bi = next;
                }
                while ai < v {
                    let next = ai + 1;
                    a[next] = b2[ai] * x[ai] % DIV + a[ai] % DIV;
                    ai = next;
                }
                println!("{}", a[v] % DIV);
            },
            _ => panic!(),
        }
        // println!("x{:?}", x);
        // println!("y{:?}", y);
        // println!("a{:?}", a);
        // println!("b{:?}", b);
    }
}
0