#604648 (Rust) No.1340 おーじ君をさがせ

提出ソース

結果

問題	No.1340 おーじ君をさがせ
ユーザー	Moss_Local
提出日時	2021-01-15 23:01:56
言語	Rust (1.77.0)
結果	AC
実行時間	1,363 ms / 2,000 ms
コード長	9,121 bytes
コンパイル時間	1,365 ms
コンパイル使用メモリ	158,096 KB
実行使用メモリ	4,384 KB
最終ジャッジ日時	2023-08-18 16:43:58
合計ジャッジ時間	21,140 ms
ジャッジサーバーID （参考情報）	judge12 / judge15

このコードへのチャレンジ（β）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 4,380 KB
testcase_01	AC	1 ms 4,376 KB
testcase_02	AC	1 ms 4,380 KB
testcase_03	AC	1 ms 4,380 KB
testcase_04	AC	1 ms 4,380 KB
testcase_05	AC	1 ms 4,376 KB
testcase_06	AC	1 ms 4,380 KB
testcase_07	AC	1 ms 4,380 KB
testcase_08	AC	1 ms 4,376 KB
testcase_09	AC	1 ms 4,384 KB
testcase_10	AC	2 ms 4,380 KB
testcase_11	AC	208 ms 4,380 KB
testcase_12	AC	37 ms 4,380 KB
testcase_13	AC	176 ms 4,376 KB
testcase_14	AC	529 ms 4,380 KB
testcase_15	AC	402 ms 4,376 KB
testcase_16	AC	28 ms 4,384 KB
testcase_17	AC	133 ms 4,380 KB
testcase_18	AC	238 ms 4,380 KB
testcase_19	AC	11 ms 4,380 KB
testcase_20	AC	3 ms 4,380 KB
testcase_21	AC	247 ms 4,376 KB
testcase_22	AC	701 ms 4,376 KB
testcase_23	AC	256 ms 4,380 KB
testcase_24	AC	1,363 ms 4,380 KB
testcase_25	AC	22 ms 4,376 KB
testcase_26	AC	15 ms 4,380 KB
testcase_27	AC	10 ms 4,376 KB
testcase_28	AC	36 ms 4,380 KB
testcase_29	AC	2 ms 4,380 KB
testcase_30	AC	279 ms 4,380 KB
testcase_31	AC	1,348 ms 4,376 KB
testcase_32	AC	1,201 ms 4,380 KB
testcase_33	AC	1,212 ms 4,380 KB
testcase_34	AC	1,119 ms 4,376 KB
testcase_35	AC	1,229 ms 4,380 KB
testcase_36	AC	1 ms 4,384 KB
testcase_37	AC	2 ms 4,380 KB
testcase_38	AC	1,326 ms 4,380 KB
testcase_39	AC	320 ms 4,380 KB
testcase_40	AC	309 ms 4,380 KB
testcase_41	AC	303 ms 4,376 KB
testcase_42	AC	1 ms 4,376 KB
testcase_43	AC	1 ms 4,380 KB
testcase_44	AC	1 ms 4,380 KB
testcase_45	AC	1 ms 4,380 KB
testcase_46	AC	232 ms 4,380 KB
testcase_47	AC	251 ms 4,380 KB
testcase_48	AC	275 ms 4,376 KB
testcase_49	AC	290 ms 4,380 KB
testcase_50	AC	307 ms 4,380 KB
testcase_51	AC	307 ms 4,380 KB
testcase_52	AC	585 ms 4,384 KB
testcase_53	AC	591 ms 4,376 KB
testcase_54	AC	596 ms 4,380 KB
testcase_55	AC	453 ms 4,376 KB
testcase_56	AC	1 ms 4,380 KB
testcase_57	AC	1 ms 4,376 KB
testcase_58	AC	1 ms 4,380 KB
testcase_59	AC	148 ms 4,376 KB
testcase_60	AC	1 ms 4,376 KB
testcase_61	AC	167 ms 4,376 KB

権限があれば一括ダウンロードができます

コンパイルメッセージ

warning: unnecessary parentheses around type
  --> Main.rs:43:15
   |
43 | fn readi() -> (i64) {
   |               ^   ^
   |
   = note: `#[warn(unused_parens)]` on by default
help: remove these parentheses
   |
43 - fn readi() -> (i64) {
43 + fn readi() -> i64 {
   |

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

warning: variable does not need to be mutable
   --> Main.rs:314:9
    |
314 |     let mut r = x.pow(t as u64, INF);
    |         ----^
    |         |
    |         help: remove this `mut`
    |
    = note: `#[warn(unused_mut)]` on by default

warning: variable `K` should have a snake case name
   --> Main.rs:183:17
    |
183 |             let K = self.n();
    |                 ^ help: convert the identifier to snake case (notice the capitalization): `k`
    |
    = note: `#[warn(non_snake_case)]` on by default

warning: variable `M` should have a snake case name
   --> Main.rs:184:17
    |
184 |             let M = self.m();
    |                 ^ help: convert the identifier to snake case: `m`

warning: variable `N` should have a snake case name
   --> Main.rs:185:17
    |
185 |             let N = other.n();
    |                 ^ help: convert the identifier to snake case: `n`

warning: variable `K` should have a snake case name
   --> Main.rs:250:17
    |
250 |             let K = self.n();
    |                 ^ help: convert the identifier to snake case (notice the capitalization): `k`

warning: variable `M` should have a snake case name
   --> Main.rs:251:17
    |
251 |             let M = self.m();
    |                 ^ help: convert the identifier to snake case: `m`

warning: variable `N` should have a snake case name
   --> Main.rs:252:17
    |
252 |             let N = other.n();
    |                 ^ help: convert the identifier to snake case: `n`

warning:

ソースコード

raw source code

// -*- coding:utf-8-unix -*-

#![allow(dead_code)]
#![allow(unused_imports)]
use std::collections::*;
use std::convert::*;
use std::convert::{From, Into};
use std::fmt::Debug;
use std::fs::File;
use std::io::prelude::*;
use std::io::*;
use std::marker::Copy;
use std::mem::*;
use std::ops::Bound::*;
use std::ops::{Add, Mul, Neg, Sub};
use std::str;
use std::vec;
use std::{cmp, process::Output};
use std::{cmp::Ordering, env::consts::DLL_PREFIX};
use std::{cmp::Ordering::*, f32::consts::PI};

const INF: i64 = 1223372036854775807;
// const INF: i64 = 2147483647;
const MEM_SIZE: usize = 202020;
const MOD: i64 = 1000000007;
// const MOD: i64 = 998244353;
use std::cmp::*;
use std::collections::*;
use std::io::stdin;
use std::io::stdout;
use std::io::Write;

use btree_map::{Keys, Values};
// use str::Chars;
#[allow(dead_code)]
fn read<T: std::str::FromStr>() -> T {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    s.trim().parse().ok().unwrap()
}

#[allow(dead_code)]
fn readi() -> (i64) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    iter.next().unwrap().parse::<i64>().unwrap()
}

#[allow(dead_code)]
fn read_vec<T: std::str::FromStr>() -> Vec<T> {
    read::<String>()
        .split_whitespace()
        .map(|e| e.parse().ok().unwrap())
        .collect()
}
#[allow(dead_code)]
fn read_vec2<T: std::str::FromStr>(n: u32) -> Vec<Vec<T>> {
    (0..n).map(|_| read_vec()).collect()
}

#[allow(dead_code)]
fn readii() -> (i64, i64) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<i64>().unwrap(),
        iter.next().unwrap().parse::<i64>().unwrap(),
    )
}

fn readff() -> (f64, f64) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<f64>().unwrap(),
        iter.next().unwrap().parse::<f64>().unwrap(),
    )
}

#[allow(dead_code)]
fn readiii() -> (i64, i64, i64) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<i64>().unwrap(),
        iter.next().unwrap().parse::<i64>().unwrap(),
        iter.next().unwrap().parse::<i64>().unwrap(),
    )
}
#[allow(dead_code)]
fn readuu() -> (usize, usize) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<usize>().unwrap(),
    )
}

fn readcc() -> (char, char) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<char>().unwrap(),
        iter.next().unwrap().parse::<char>().unwrap(),
    )
}

#[allow(dead_code)]
fn readuuu() -> (usize, usize, usize) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<usize>().unwrap(),
    )
}

#[allow(dead_code)]
fn readuuuu() -> (usize, usize, usize, usize) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<usize>().unwrap(),
    )
}

fn readiiii() -> (i64, i64, i64, i64) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<i64>().unwrap(),
        iter.next().unwrap().parse::<i64>().unwrap(),
        iter.next().unwrap().parse::<i64>().unwrap(),
        iter.next().unwrap().parse::<i64>().unwrap(),
    )
}

fn readin() -> (usize, char) {
    let mut str = String::new();
    let _ = stdin().read_line(&mut str).unwrap();
    let mut iter = str.split_whitespace();
    (
        iter.next().unwrap().parse::<usize>().unwrap(),
        iter.next().unwrap().parse::<char>().unwrap(),
    )
}

pub mod matrix {
    #[derive(Clone)]
    pub struct Matrix {
        pub v: Vec<Vec<i64>>,
    }
    impl Matrix {
        pub fn identity(n: usize) -> Self {
            let mut v = vec![vec![0; n]; n];
            for i in 0..n {
                v[i][i] = 1;
            }
            Matrix { v: v }
        }
        pub fn m(&self) -> usize {
            self.v.len()
        }
        pub fn n(&self) -> usize {
            self.v[0].len()
        }
        pub fn mul_rem(&self, other: &Self, mo: i64) -> Self {
            assert!(self.n() == other.m());
            let K = self.n();
            let M = self.m();
            let N = other.n();
            let mut r = vec![vec![0; N]; M];
            for i in 0..M {
                for j in 0..N {
                    let mut v = 0;
                    for k in 0..K {
                        v += self.v[i][k] & other.v[k][j] % mo;
                        v %= mo;
                    }
                    if v > 0 {
                        v = 1;
                    }
                    r[i][j] = v;
                }
            }
            Matrix { v: r }
        }
        pub fn pow(&self, k: u64, mo: i64) -> Self {
            assert!(self.m() == self.n());
            let mut k = k;
            let mut x = Self::identity(self.m());
            let mut y = self.clone();
            while k > 0 {
                if k & 1 > 0 {
                    x = y.clone() * x;
                    x %= mo;
                }
                y = y.mul_rem(&y, mo);
                y %= mo;
                k >>= 1;
            }
            x
        }
    }

    use std::ops::*;

    impl Add for Matrix {
        type Output = Self;
        fn add(self, other: Self) -> Self {
            let mut r = self.v.clone();
            for i in 0..self.m() {
                for j in 0..self.n() {
                    r[i][j] += other.v[i][j];
                }
            }
            Matrix { v: r }
        }
    }
    impl Sub for Matrix {
        type Output = Self;
        fn sub(self, other: Self) -> Self {
            let mut r = self.v.clone();
            for i in 0..self.m() {
                for j in 0..self.n() {
                    r[i][j] -= other.v[i][j];
                }
            }
            Matrix { v: r }
        }
    }
    impl Mul for Matrix {
        type Output = Self;
        fn mul(self, other: Self) -> Self {
            assert!(self.n() == other.m());
            let K = self.n();
            let M = self.m();
            let N = other.n();
            let mut r = vec![vec![0; N]; M];
            for i in 0..M {
                for j in 0..N {
                    let mut v = 0;
                    for k in 0..K {
                        v += self.v[i][k] & other.v[k][j];
                        if v > 0 {
                            v = 1;
                        }
                    }
                    if v > 0 {
                        v = 1;
                    }
                    r[i][j] = v;
                }
            }
            Matrix { v: r }
        }
    }
    impl Rem<i64> for Matrix {
        type Output = Self;
        fn rem(self, mo: i64) -> Self {
            let mut r = self.v.clone();
            for i in 0..self.m() {
                for j in 0..self.n() {
                    r[i][j] %= mo;
                }
            }
            Matrix { v: r }
        }
    }
    impl RemAssign<i64> for Matrix {
        fn rem_assign(&mut self, mo: i64) {
            for i in 0..self.m() {
                for j in 0..self.n() {
                    self.v[i][j] %= mo;
                }
            }
        }
    }
}

pub type Matrix = matrix::Matrix;

fn solve() {
    // let (mut a, mut b, mut c) = readuuu();
    // a += 100;[]
    // let (n, m, t) = readuuu();
    let vec: Vec<u64> = read_vec();
    let n = vec[0] as usize;
    let m = vec[1] as usize;
    let t = vec[2] as u64;
    let mut vv = vec![vec![0 as i64; (n) as usize]; (n) as usize];

    for i in 0..m {
        let (a, b) = readuu();
        vv[a][b] = 1;
    }
    let x = Matrix { v: vv };
    // let x3 = x.clone() * x.clone() * x.clone();
    // assert_eq!(x.pow(3, std::i64::MAX).v, x3.v);
    let mut r = x.pow(t as u64, INF);
    let mut res = 0 as usize;
    for i in 0..1 {
        for j in 0..n {
            if r.v[i][j] >= 1 {
                res += 1;
            }
        }
    }
    // let mut y = Matrix { v: vv };
    // println!("{:?}", r.v);
    if t == 0 {
        println!("{:?}", 1);
        return;
    }
    println!("{:?}", res);

    return;
}

fn main() {
    solve()
}

yukicoder

結果

テストケース

コンパイルメッセージ

ソースコード