#381812 (Rust) No.891 隣接3項間の漸化式

• 問題: No.891 隣接3項間の漸化式
• ユーザー: akiradeveloper
• 提出日時: 2019-09-20 23:18:41
• 言語: Rust (1.77.0 + proconio)
• 結果: AC
• 実行時間: 1 ms / 2,000 ms
• コード長: 5,936 bytes
• コンパイル時間: 13,970 ms
• コンパイル使用メモリ: 379,520 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-09-14 20:05:39
• 合計ジャッジ時間: 15,408 ms
• ジャッジサーバーID （参考情報）: judge5 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,248 KB
testcase_02	AC	1 ms; 5,248 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	1 ms; 5,376 KB
testcase_07	AC	1 ms; 5,376 KB
testcase_08	AC	1 ms; 5,376 KB
testcase_09	AC	1 ms; 5,376 KB
testcase_10	AC	1 ms; 5,376 KB
testcase_11	AC	1 ms; 5,376 KB
testcase_12	AC	1 ms; 5,376 KB
testcase_13	AC	1 ms; 5,376 KB
testcase_14	AC	1 ms; 5,376 KB
testcase_15	AC	1 ms; 5,376 KB
testcase_16	AC	1 ms; 5,376 KB
testcase_17	AC	1 ms; 5,376 KB
testcase_18	AC	1 ms; 5,376 KB
testcase_19	AC	1 ms; 5,376 KB
testcase_20	AC	1 ms; 5,376 KB
testcase_21	AC	1 ms; 5,376 KB
testcase_22	AC	1 ms; 5,376 KB
testcase_23	AC	1 ms; 5,376 KB
testcase_24	AC	1 ms; 5,376 KB
testcase_25	AC	1 ms; 5,376 KB
testcase_26	AC	1 ms; 5,376 KB
testcase_27	AC	1 ms; 5,376 KB
testcase_28	AC	1 ms; 5,376 KB
testcase_29	AC	1 ms; 5,376 KB
testcase_30	AC	1 ms; 5,376 KB
testcase_31	AC	1 ms; 5,376 KB
testcase_32	AC	1 ms; 5,376 KB
testcase_33	AC	1 ms; 5,376 KB
testcase_34	AC	1 ms; 5,376 KB
testcase_35	AC	1 ms; 5,376 KB
testcase_36	AC	1 ms; 5,376 KB
testcase_37	AC	1 ms; 5,376 KB
testcase_38	AC	1 ms; 5,376 KB
testcase_39	AC	1 ms; 5,376 KB
testcase_40	AC	1 ms; 5,376 KB
testcase_41	AC	1 ms; 5,376 KB

コンパイルメッセージ

warning: unnecessary parentheses around assigned value
   --> src/main.rs:166:18
    |
166 |             n -= (1 << k);
    |                  ^      ^
    |
    = note: `#[warn(unused_parens)]` on by default
help: remove these parentheses
    |
166 -             n -= (1 << k);
166 +             n -= 1 << k;
    |

warning: unused variable: `i`
   --> src/main.rs:139:9
    |
139 |     for i in 0..60 {
    |         ^ help: if this is intentional, prefix it with an underscore: `_i`
    |
    = note: `#[warn(unused_variables)]` on by default

warning: function `mat_add` is never used
   --> src/main.rs:106:4
    |
106 | fn mat_add(x: &[Vec<i64>], y: &[Vec<i64>], mo: i64) -> Vec<Vec<i64>> {
    |    ^^^^^^^
    |
    = note: `#[warn(dead_code)]` on by default

warning: variable `logN` should have a snake case name
   --> src/main.rs:160:9
    |
160 |     let logN = (n as f64).log2().floor() as usize;
    |         ^^^^ help: convert the identifier to snake case: `log_n`
    |
    = note: `#[warn(non_snake_case)]` on by default

ソースコード

#[doc = " https://github.com/hatoo/competitive-rust-snippets"]
#[allow(unused_imports)]
use std::cmp::{max, min, Ordering};
#[allow(unused_imports)]
use std::collections::{BTreeMap, BTreeSet, BinaryHeap, HashMap, HashSet, VecDeque};
#[allow(unused_imports)]
use std::io::{stdin, stdout, BufWriter, Write};
#[allow(unused_imports)]
use std::iter::FromIterator;
#[allow(unused_macros)]
macro_rules ! debug { ( $ ( $ a : expr ) ,* ) => { eprintln ! ( concat ! ( $ ( stringify ! ( $ a ) , " = {:?}, " ) ,* ) , $ ( $ a ) ,* ) ; } }
#[macro_export]
macro_rules ! input { ( source = $ s : expr , $ ( $ r : tt ) * ) => { let mut parser = Parser :: from_str ( $ s ) ; input_inner ! { parser , $ ( $ r ) * } } ; ( parser = $ parser : ident , $ ( $ r : tt ) * ) => { input_inner ! { $ parser , $ ( $ r ) * } } ; ( new_stdin_parser = $ parser : ident , $ ( $ r : tt ) * ) => { let stdin = std :: io :: stdin ( ) ; let reader = std :: io :: BufReader :: new ( stdin . lock ( ) ) ; let mut $ parser = Parser :: new ( reader ) ; input_inner ! { $ parser , $ ( $ r ) * } } ; ( $ ( $ r : tt ) * ) => { input ! { new_stdin_parser = parser , $ ( $ r ) * } } ; }
#[macro_export]
macro_rules ! input_inner { ( $ parser : ident ) => { } ; ( $ parser : ident , ) => { } ; ( $ parser : ident , $ var : ident : $ t : tt $ ( $ r : tt ) * ) => { let $ var = read_value ! ( $ parser , $ t ) ; input_inner ! { $ parser $ ( $ r ) * } } ; }
#[macro_export]
macro_rules ! read_value { ( $ parser : ident , ( $ ( $ t : tt ) ,* ) ) => { ( $ ( read_value ! ( $ parser , $ t ) ) ,* ) } ; ( $ parser : ident , [ $ t : tt ; $ len : expr ] ) => { ( 0 ..$ len ) . map ( | _ | read_value ! ( $ parser , $ t ) ) . collect ::< Vec < _ >> ( ) } ; ( $ parser : ident , chars ) => { read_value ! ( $ parser , String ) . chars ( ) . collect ::< Vec < char >> ( ) } ; ( $ parser : ident , usize1 ) => { read_value ! ( $ parser , usize ) - 1 } ; ( $ parser : ident , $ t : ty ) => { $ parser . next ::<$ t > ( ) . expect ( "Parse error" ) } ; }
use std::io;
use std::io::BufRead;
use std::str;
pub struct Parser<R> {
    reader: R,
    buf: Vec<u8>,
    pos: usize,
}
impl Parser<io::Empty> {
    pub fn from_str(s: &str) -> Parser<io::Empty> {
        Parser {
            reader: io::empty(),
            buf: s.as_bytes().to_vec(),
            pos: 0,
        }
    }
}
impl<R: BufRead> Parser<R> {
    pub fn new(reader: R) -> Parser<R> {
        Parser {
            reader: reader,
            buf: vec![],
            pos: 0,
        }
    }
    pub fn update_buf(&mut self) {
        self.buf.clear();
        self.pos = 0;
        loop {
            let (len, complete) = {
                let buf2 = self.reader.fill_buf().unwrap();
                self.buf.extend_from_slice(buf2);
                let len = buf2.len();
                if len == 0 {
                    break;
                }
                (len, buf2[len - 1] <= 0x20)
            };
            self.reader.consume(len);
            if complete {
                break;
            }
        }
    }
    pub fn next<T: str::FromStr>(&mut self) -> Result<T, T::Err> {
        loop {
            let mut begin = self.pos;
            while begin < self.buf.len() && (self.buf[begin] <= 0x20) {
                begin += 1;
            }
            let mut end = begin;
            while end < self.buf.len() && (self.buf[end] > 0x20) {
                end += 1;
            }
            if begin != self.buf.len() {
                self.pos = end;
                return str::from_utf8(&self.buf[begin..end]).unwrap().parse::<T>();
            } else {
                self.update_buf();
            }
        }
    }
}
#[allow(unused_macros)]
macro_rules ! debug { ( $ ( $ a : expr ) ,* ) => { eprintln ! ( concat ! ( $ ( stringify ! ( $ a ) , " = {:?}, " ) ,* ) , $ ( $ a ) ,* ) ; } }
#[doc = " https://github.com/hatoo/competitive-rust-snippets"]
const BIG_STACK_SIZE: bool = true;
#[allow(dead_code)]
fn main() {
    use std::thread;
    if BIG_STACK_SIZE {
        thread::Builder::new()
            .stack_size(32 * 1024 * 1024)
            .name("solve".into())
            .spawn(solve)
            .unwrap()
            .join()
            .unwrap();
    } else {
        solve();
    }
}
fn mat_mul(x: &[Vec<i64>], y: &[Vec<i64>], mo: i64) -> Vec<Vec<i64>> {
    vec![
        vec![(x[0][0]*y[0][0]+x[0][1]*y[1][0]) % mo, (x[0][0]*y[0][1]+x[0][1]*y[1][1]) % mo],
        vec![(x[1][0]*y[0][0]+x[1][1]*y[1][0]) % mo, (x[1][0]*y[0][1]+x[1][1]*y[1][1]) % mo]
    ]
}
fn mat_add(x: &[Vec<i64>], y: &[Vec<i64>], mo: i64) -> Vec<Vec<i64>> {
    vec![
        vec![(x[0][0]+y[0][0]) % mo, (x[0][1]+y[0][1]) % mo],
        vec![(x[0][1]+y[0][1]) % mo, (x[1][1]+y[1][1]) % mo],
    ]
}
fn mat_apply(x: &[Vec<i64>], y: &[i64], mo: i64) -> Vec<i64> {
    vec![
        (x[0][0]*y[0]+x[0][1]*y[1]) % mo,
        (x[1][0]*y[0]+x[1][1]*y[1]) % mo
    ]
}
fn solve() {
    input!{
        a:i64,b:i64,n:usize,
    }
    let mo = 1_000_000_007;
    if n==0 {
        println!("0");
        return
    }
    if n==1 {
        println!("1");
        return
    }

    let bd = bin_digits(n-1);
    let mut muls = vec![];
    let mut acc = vec![
        vec![a, b],
        vec![1, 0],
    ];
    muls.push(acc.clone());
    for i in 0..60 {
        acc = mat_mul(&acc, &acc, mo);
        muls.push(acc.clone());
    }
    
    let mut acc = vec![
        vec![1,0],
        vec![0,1],
    ];
    for i in 0..bd.len() {
        if bd[i] {
            acc = mat_mul(&acc, &muls[i], mo);
        }
    }
    let res = mat_apply(&acc, &vec![1,0], mo);
    println!("{}",res[0]);
}
fn bin_digits(n: usize) -> Vec<bool> {
    if n == 0 {
        return vec![];
    }
    let logN = (n as f64).log2().floor() as usize;
    let mut res = vec![false; logN + 1];
    let mut n = n;
    for k in (0..logN + 1).rev() {
        if n >= 1 << k {
            res[k] = true;
            n -= (1 << k);
        }
    }
    res
}