結果
問題 | No.1749 ラムドスウイルスの感染拡大 |
ユーザー | togatoga |
提出日時 | 2021-11-19 21:47:51 |
言語 | Rust (1.77.0 + proconio) |
結果 |
AC
|
実行時間 | 18 ms / 2,000 ms |
コード長 | 11,228 bytes |
コンパイル時間 | 11,877 ms |
コンパイル使用メモリ | 387,892 KB |
実行使用メモリ | 9,984 KB |
最終ジャッジ日時 | 2024-06-10 08:32:07 |
合計ジャッジ時間 | 13,407 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
6,812 KB |
testcase_01 | AC | 1 ms
6,940 KB |
testcase_02 | AC | 1 ms
6,944 KB |
testcase_03 | AC | 1 ms
6,940 KB |
testcase_04 | AC | 9 ms
6,940 KB |
testcase_05 | AC | 1 ms
6,944 KB |
testcase_06 | AC | 1 ms
6,940 KB |
testcase_07 | AC | 2 ms
6,944 KB |
testcase_08 | AC | 1 ms
6,944 KB |
testcase_09 | AC | 1 ms
6,940 KB |
testcase_10 | AC | 8 ms
9,984 KB |
testcase_11 | AC | 8 ms
9,856 KB |
testcase_12 | AC | 18 ms
9,728 KB |
testcase_13 | AC | 1 ms
6,940 KB |
testcase_14 | AC | 1 ms
6,940 KB |
testcase_15 | AC | 2 ms
6,940 KB |
testcase_16 | AC | 4 ms
6,944 KB |
testcase_17 | AC | 10 ms
6,940 KB |
testcase_18 | AC | 14 ms
8,832 KB |
testcase_19 | AC | 13 ms
7,424 KB |
testcase_20 | AC | 1 ms
6,944 KB |
testcase_21 | AC | 1 ms
6,944 KB |
testcase_22 | AC | 1 ms
6,940 KB |
testcase_23 | AC | 1 ms
6,940 KB |
testcase_24 | AC | 1 ms
6,940 KB |
testcase_25 | AC | 7 ms
7,552 KB |
testcase_26 | AC | 2 ms
6,940 KB |
testcase_27 | AC | 2 ms
6,940 KB |
testcase_28 | AC | 9 ms
6,944 KB |
testcase_29 | AC | 9 ms
6,940 KB |
ソースコード
use crate::mod_int::ModInt998244353; pub mod utils { const DYX: [(isize, isize); 8] = [ (0, 1), (1, 0), (0, -1), (-1, 0), (1, 1), (-1, 1), (1, -1), (-1, -1), ]; pub fn try_adj(y: usize, x: usize, dir: usize, h: usize, w: usize) -> Option<(usize, usize)> { let ny = y as isize + DYX[dir].0; let nx = x as isize + DYX[dir].1; if ny >= 0 && nx >= 0 && ny < h as isize && nx < w as isize { Some((ny as usize, nx as usize)) } else { None } } } /// Modular Integer /// NOTE /// If a modular isn't prime, you can't div. /// If you want to calculate a combination and permutation, you have to use `mod_comb`. pub mod mod_int { use std::marker::PhantomData; use std::ops::{Add, AddAssign, Div, DivAssign, Mul, MulAssign, Sub, SubAssign}; #[derive(Copy, Clone, Ord, PartialOrd, Eq, PartialEq, Hash, Debug)] pub struct Mod1000000007; #[derive(Copy, Clone, Ord, PartialOrd, Eq, PartialEq, Hash, Debug)] pub struct Mod998244353; pub trait Modulus: Copy + Eq + Copy { const VALUE: u32; } impl Modulus for Mod1000000007 { const VALUE: u32 = 1000000007; } impl Modulus for Mod998244353 { const VALUE: u32 = 998244353; } #[derive(Clone, Copy, Debug, PartialEq, Eq)] pub struct ModInt< T: Copy + Clone + Add + AddAssign + Mul + MulAssign + Sub + SubAssign, M: Modulus, > { pub val: T, phantom: std::marker::PhantomData<fn() -> M>, } /// Implementation macros #[warn(unused_macros)] macro_rules ! mod_int_impl {($ ($ t : ty ) * ) => ($ (impl < M : Modulus > ModInt <$ t , M > {pub fn new (x : $ t ) -> Self {ModInt {val : x % M :: VALUE as $ t , phantom : PhantomData } } pub fn pow (self , e : usize ) -> ModInt <$ t , M > {let mut result = ModInt ::<$ t , M >:: new (1 ) ; let mut cur = self ; let mut e = e ; while e > 0 {if e & 1 == 1 {result *= cur ; } cur *= cur ; e >>= 1 ; } result } } impl < M : Modulus > Add < ModInt <$ t , M >> for ModInt <$ t , M > {type Output = ModInt <$ t , M >; fn add (self , rhs : ModInt <$ t , M > ) -> ModInt <$ t , M > {self + rhs . val } } impl < M : Modulus > Add < ModInt <$ t , M >> for $ t {type Output = ModInt <$ t , M >; fn add (self , rhs : ModInt <$ t , M > ) -> ModInt <$ t , M > {let x = self % M :: VALUE as $ t ; let val = (x + rhs . val ) % M :: VALUE as $ t ; ModInt {val , phantom : PhantomData } } } impl < M : Modulus > Add <$ t > for ModInt <$ t , M > {type Output = ModInt <$ t , M >; fn add (self , rhs : $ t ) -> ModInt <$ t , M > {let x = rhs % M :: VALUE as $ t ; let val = (self . val + x ) % M :: VALUE as $ t ; ModInt {val , phantom : PhantomData } } } impl < M : Modulus > Sub < ModInt <$ t , M >> for ModInt <$ t , M > {type Output = ModInt <$ t , M >; fn sub (self , rhs : ModInt <$ t , M > ) -> ModInt <$ t , M > {self - rhs . val } } impl < M : Modulus > Sub < ModInt <$ t , M >> for $ t {type Output = ModInt <$ t , M >; fn sub (self , rhs : ModInt <$ t , M > ) -> ModInt <$ t , M > {let val = self % M :: VALUE as $ t ; ModInt {val , phantom : PhantomData } - rhs } } impl < M : Modulus > Sub <$ t > for ModInt <$ t , M > {type Output = ModInt <$ t , M >; fn sub (self , rhs : $ t ) -> ModInt <$ t , M > {let rhs = if rhs >= M :: VALUE as $ t {rhs % M :: VALUE as $ t } else {rhs } ; let val = if self . val < rhs {self . val + M :: VALUE as $ t - rhs } else {self . val - rhs } ; ModInt {val , phantom : PhantomData } } } impl < M : Modulus > AddAssign < ModInt <$ t , M >> for ModInt <$ t , M > {fn add_assign (& mut self , rhs : ModInt <$ t , M > ) {* self = * self + rhs ; } } impl < M : Modulus > AddAssign <$ t > for ModInt <$ t , M > {fn add_assign (& mut self , rhs : $ t ) {* self = * self + rhs ; } } impl < M : Modulus > SubAssign < ModInt <$ t , M >> for ModInt <$ t , M > {fn sub_assign (& mut self , rhs : ModInt <$ t , M > ) {* self = * self - rhs ; } } impl < M : Modulus > SubAssign <$ t > for ModInt <$ t , M > {fn sub_assign (& mut self , rhs : $ t ) {* self = * self - rhs ; } } impl < M : Modulus > Div <$ t > for ModInt <$ t , M > {type Output = ModInt <$ t , M >; fn div (self , mut rhs : $ t ) -> ModInt <$ t , M > {if rhs >= M :: VALUE as $ t {rhs %= M :: VALUE as $ t ; } self * ModInt {val : rhs , phantom : PhantomData } . pow ((M :: VALUE - 2 ) as usize ) } } impl < M : Modulus > Div < ModInt <$ t , M >> for ModInt <$ t , M > {type Output = ModInt <$ t , M >; fn div (self , rhs : ModInt <$ t , M > ) -> ModInt <$ t , M > {self / rhs . val } } impl < M : Modulus > DivAssign <$ t > for ModInt <$ t , M > {fn div_assign (& mut self , rhs : $ t ) {* self = * self / rhs } } impl < M : Modulus > DivAssign < ModInt <$ t , M >> for ModInt <$ t , M > {fn div_assign (& mut self , rhs : ModInt <$ t , M > ) {* self = * self / rhs } } impl < M : Modulus > Mul < ModInt <$ t , M >> for ModInt <$ t , M > {type Output = ModInt <$ t , M >; fn mul (self , rhs : ModInt <$ t , M > ) -> ModInt <$ t , M > {self * rhs . val } } impl < M : Modulus > Mul < ModInt <$ t , M >> for $ t {type Output = ModInt <$ t , M >; fn mul (self , rhs : ModInt <$ t , M > ) -> ModInt <$ t , M > {rhs * self } } impl < M : Modulus > Mul <$ t > for ModInt <$ t , M > {type Output = ModInt <$ t , M >; fn mul (self , rhs : $ t ) -> ModInt <$ t , M > {ModInt {val : self . val * (rhs % M :: VALUE as $ t ) % M :: VALUE as $ t , phantom : PhantomData } } } impl < M : Modulus > MulAssign < ModInt <$ t , M >> for ModInt <$ t , M > {fn mul_assign (& mut self , rhs : ModInt <$ t , M > ) {* self = * self * rhs ; } } impl < M : Modulus > MulAssign <$ t > for ModInt <$ t , M > {fn mul_assign (& mut self , rhs : $ t ) {* self = * self * rhs ; } } impl < M : Modulus > Default for ModInt <$ t , M > {fn default () -> ModInt <$ t , M > {ModInt {val : 0 , phantom : PhantomData } } } ) * ) } mod_int_impl ! (usize i64 u64 i128 ); #[allow(dead_code)] pub type ModInt1000000007 = ModInt<i64, Mod1000000007>; pub type ModInt998244353 = ModInt<i64, Mod998244353>; } #[derive(Default)] /// NOTE /// declare variables to reduce the number of parameters for dp and dfs etc. pub struct Solver {} impl Solver { pub fn solve(&mut self) { let stdin = std::io::stdin(); let mut scn = fastio::Scanner::new(stdin.lock()); let n: usize = scn.read(); let m: usize = scn.read(); let t: usize = scn.read(); let mut edges = vec![vec![]; n]; for _ in 0..m { let s: usize = scn.read(); let t: usize = scn.read(); edges[s].push(t); edges[t].push(s); } let mut dp = vec![vec![ModInt998244353::new(0); n + 1]; t+1]; dp[0][0] += 1; for i in 0..t { for j in 0..n { //dp[i + 1][j] = dp[i][j]; for &k in edges[j].iter() { let p = dp[i][k]; dp[i + 1][j] += p; } } //cecho!(dp[i][0], dp[i+1][0]); } println!("{}", dp[t][0].val); } } pub mod fastio { use std::collections::VecDeque; use std::io::BufWriter; use std::io::Write; pub struct Writer<W: std::io::Write> { writer: std::io::BufWriter<W>, } impl<W: std::io::Write> Writer<W> { pub fn new(write: W) -> Writer<W> { Writer { writer: BufWriter::new(write), } } pub fn flush(&mut self) { self.writer.flush().unwrap(); } pub fn write<S: std::string::ToString>(&mut self, s: S) { self.writer.write(s.to_string().as_bytes()).unwrap(); } pub fn writeln<S: std::string::ToString>(&mut self, s: S) { self.write(s); self.write('\n'); } } pub struct Scanner<R> { stdin: R, buffer: VecDeque<String>, } impl<R: std::io::BufRead> Scanner<R> { pub fn new(s: R) -> Scanner<R> { Scanner { stdin: s, buffer: VecDeque::new(), } } pub fn read<T: std::str::FromStr>(&mut self) -> T { while self.buffer.is_empty() { let line = self.read_line(); for w in line.split_whitespace() { self.buffer.push_back(String::from(w)); } } self.buffer.pop_front().unwrap().parse::<T>().ok().unwrap() } pub fn read_line(&mut self) -> String { let mut line = String::new(); let _ = self.stdin.read_line(&mut line); line.trim_end().to_string() } pub fn vec<T: std::str::FromStr>(&mut self, n: usize) -> Vec<T> { (0..n).map(|_| self.read()).collect() } pub fn chars(&mut self) -> Vec<char> { self.read::<String>().chars().collect() } } } pub mod macros { #[macro_export] #[allow(unused_macros)] macro_rules ! max {($ x : expr ) => ($ x ) ; ($ x : expr , $ ($ y : expr ) ,+ ) => {std :: cmp :: max ($ x , max ! ($ ($ y ) ,+ ) ) } } #[macro_export] #[allow(unused_macros)] macro_rules ! min {($ x : expr ) => ($ x ) ; ($ x : expr , $ ($ y : expr ) ,+ ) => {std :: cmp :: min ($ x , min ! ($ ($ y ) ,+ ) ) } } #[macro_export] #[allow(unused_macros)] /// Display a line of variables macro_rules ! echo {() => {{use std :: io :: {self , Write } ; writeln ! (io :: stderr () , "{}:" , line ! () ) . unwrap () ; } } ; ($ e : expr , $ ($ es : expr ) ,+ $ (, ) ? ) => {{use std :: io :: {self , Write } ; write ! (io :: stderr () , "{}:" , line ! () ) . unwrap () ; write ! (io :: stderr () , " {} = {:?}" , stringify ! ($ e ) , $ e ) . unwrap () ; $ (write ! (io :: stderr () , " {} = {:?}" , stringify ! ($ es ) , $ es ) . unwrap () ; ) + writeln ! (io :: stderr () ) . unwrap () ; } } ; ($ e : expr ) => {{use std :: io :: {self , Write } ; let result = $ e ; writeln ! (io :: stderr () , "{}: {} = {:?}" , line ! () , stringify ! ($ e ) , result ) . unwrap () ; result } } ; } #[macro_export] #[allow(unused_macros)] /// Display a line of variables with colors macro_rules ! cecho {() => {{use std :: io :: {self , Write } ; writeln ! (io :: stderr () , "\x1b[31;1m{}\x1b[m:" , line ! () ) . unwrap () ; } } ; ($ e : expr , $ ($ es : expr ) ,+ $ (, ) ? ) => {{use std :: io :: {self , Write } ; write ! (io :: stderr () , "\x1b[31;1m{}\x1b[m:" , line ! () ) . unwrap () ; write ! (io :: stderr () , " \x1b[92;1m{}\x1b[m = {:?}" , stringify ! ($ e ) , $ e ) . unwrap () ; $ (write ! (io :: stderr () , " \x1b[92;1m{}\x1b[m = {:?}" , stringify ! ($ es ) , $ es ) . unwrap () ; ) + writeln ! (io :: stderr () ) . unwrap () ; } } ; ($ e : expr ) => {{use std :: io :: {self , Write } ; let result = $ e ; writeln ! (io :: stderr () , "\x1b[31;1m{}\x1b[m: \x1b[92;1m{}\x1b[m = {:?}" , line ! () , stringify ! ($ e ) , result ) . unwrap () ; result } } ; } } fn main() { std::thread::Builder::new() .stack_size(64 * 1024 * 1024) .spawn(|| Solver::default().solve()) .unwrap() .join() .unwrap(); }