結果

問題 No.1136 Four Points Tour
ユーザー phspls
提出日時 2020-09-27 22:51:10
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 1 ms / 2,000 ms
コード長 1,196 bytes
コンパイル時間 12,646 ms
コンパイル使用メモリ 394,268 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-30 14:15:46
合計ジャッジ時間 13,410 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 41
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: variable `I` should have a snake case name
  --> src/main.rs:18:9
   |
18 |     let I: Vec<Vec<u64>> = vec![vec![1, 0, 0, 0], vec![0, 1, 0, 0], vec![0, 0, 0, 1], vec![0, 0, 0, 1]];
   |         ^ help: convert the identifier to snake case (notice the capitalization): `i`
   |
   = note: `#[warn(non_snake_case)]` on by default

warning: variable `F` should have a snake case name
  --> src/main.rs:19:9
   |
19 |     let F: Vec<Vec<u64>> = vec![vec![0, 1, 1, 1], vec![1, 0, 1, 1], vec![1, 1, 1, 0], vec![1, 1, 1, 0]];
   |         ^ help: convert the identifier to snake case (notice the capitalization): `f`

warning: variable `F` should have a snake case name
  --> src/main.rs:36:9
   |
36 |     let F: Vec<Vec<u64>> = vec![vec![0, 1, 1, 1], vec![1, 0, 1, 1], vec![1, 1, 1, 0], vec![1, 1, 1, 0]];
   |         ^ help: convert the identifier to snake case (notice the capitalization): `f`

ソースコード

diff # 

const DIVISOR: u64 = 1_000_000_007;

fn calc(a: Vec<Vec<u64>>, b: Vec<Vec<u64>>) -> Vec<Vec<u64>> {
    let length = 4usize;
    let mut result: Vec<Vec<u64>> = vec![vec![0; length]; length];
    for i in 0..length {
        for j in 0..length {
            for k in 0..length {
                result[i][j] += a[i][k] * b[k][j];
                result[i][j] %= DIVISOR;
            }
        }
    }
    result
}

fn power(val: Vec<Vec<u64>>, p: u64) -> Vec<Vec<u64>> {
    let I: Vec<Vec<u64>> = vec![vec![1, 0, 0, 0], vec![0, 1, 0, 0], vec![0, 0, 0, 1], vec![0, 0, 0, 1]];
    let F: Vec<Vec<u64>> = vec![vec![0, 1, 1, 1], vec![1, 0, 1, 1], vec![1, 1, 1, 0], vec![1, 1, 1, 0]];
    if p == 0 { return I; }
    if p == 1 { return F; }
    let temp = power(val, p/2);
    let temp2 = temp.clone();
    let mut ret = calc(temp, temp2);
    if p % 2 == 1 {
        ret = calc(ret, F);
    }
    ret
}

fn main() {
    let mut n = String::new();
    std::io::stdin().read_line(&mut n).ok();
    let n: u64 = n.trim().parse().unwrap();

    let F: Vec<Vec<u64>> = vec![vec![0, 1, 1, 1], vec![1, 0, 1, 1], vec![1, 1, 1, 0], vec![1, 1, 1, 0]];
    let op = power(F, n);
    println!("{}", op[0][0]);
}
0