結果
問題 | No.1258 コインゲーム |
ユーザー | fukafukatani |
提出日時 | 2020-10-16 22:20:33 |
言語 | Rust (1.77.0) |
結果 |
AC
|
実行時間 | 226 ms / 2,000 ms |
コード長 | 6,029 bytes |
コンパイル時間 | 1,064 ms |
コンパイル使用メモリ | 157,124 KB |
実行使用メモリ | 4,384 KB |
最終ジャッジ日時 | 2023-09-28 03:38:20 |
合計ジャッジ時間 | 10,898 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 42 ms
4,380 KB |
testcase_01 | AC | 53 ms
4,380 KB |
testcase_02 | AC | 20 ms
4,376 KB |
testcase_03 | AC | 175 ms
4,376 KB |
testcase_04 | AC | 199 ms
4,376 KB |
testcase_05 | AC | 33 ms
4,376 KB |
testcase_06 | AC | 146 ms
4,376 KB |
testcase_07 | AC | 15 ms
4,376 KB |
testcase_08 | AC | 13 ms
4,376 KB |
testcase_09 | AC | 73 ms
4,376 KB |
testcase_10 | AC | 176 ms
4,380 KB |
testcase_11 | AC | 200 ms
4,380 KB |
testcase_12 | AC | 145 ms
4,376 KB |
testcase_13 | AC | 45 ms
4,376 KB |
testcase_14 | AC | 14 ms
4,380 KB |
testcase_15 | AC | 169 ms
4,376 KB |
testcase_16 | AC | 33 ms
4,376 KB |
testcase_17 | AC | 192 ms
4,376 KB |
testcase_18 | AC | 70 ms
4,380 KB |
testcase_19 | AC | 66 ms
4,376 KB |
testcase_20 | AC | 144 ms
4,380 KB |
testcase_21 | AC | 177 ms
4,380 KB |
testcase_22 | AC | 126 ms
4,376 KB |
testcase_23 | AC | 90 ms
4,380 KB |
testcase_24 | AC | 33 ms
4,376 KB |
testcase_25 | AC | 86 ms
4,376 KB |
testcase_26 | AC | 76 ms
4,376 KB |
testcase_27 | AC | 177 ms
4,380 KB |
testcase_28 | AC | 44 ms
4,380 KB |
testcase_29 | AC | 49 ms
4,380 KB |
testcase_30 | AC | 215 ms
4,380 KB |
testcase_31 | AC | 56 ms
4,376 KB |
testcase_32 | AC | 24 ms
4,380 KB |
testcase_33 | AC | 146 ms
4,380 KB |
testcase_34 | AC | 184 ms
4,380 KB |
testcase_35 | AC | 66 ms
4,380 KB |
testcase_36 | AC | 183 ms
4,380 KB |
testcase_37 | AC | 70 ms
4,380 KB |
testcase_38 | AC | 166 ms
4,380 KB |
testcase_39 | AC | 46 ms
4,380 KB |
testcase_40 | AC | 223 ms
4,380 KB |
testcase_41 | AC | 215 ms
4,376 KB |
testcase_42 | AC | 223 ms
4,380 KB |
testcase_43 | AC | 219 ms
4,376 KB |
testcase_44 | AC | 225 ms
4,380 KB |
testcase_45 | AC | 220 ms
4,380 KB |
testcase_46 | AC | 220 ms
4,380 KB |
testcase_47 | AC | 222 ms
4,376 KB |
testcase_48 | AC | 223 ms
4,376 KB |
testcase_49 | AC | 226 ms
4,384 KB |
コンパイルメッセージ
warning: unused variable: `i` --> Main.rs:20:9 | 20 | for i in 0..s { | ^ help: if this is intentional, prefix it with an underscore: `_i` | = note: `#[warn(unused_variables)]` on by default warning: associated function `set_modulus` is never used --> Main.rs:53:8 | 53 | fn set_modulus(m: i64) { | ^^^^^^^^^^^ | = note: `#[warn(dead_code)]` on by default warning: 2 warnings emitted
ソースコード
#![allow(unused_imports)] use std::cmp::*; use std::collections::*; use std::io::Write; use std::ops::Bound::*; #[allow(unused_macros)] macro_rules! debug { ($($e:expr),*) => { #[cfg(debug_assertions)] $({ let (e, mut err) = (stringify!($e), std::io::stderr()); writeln!(err, "{} = {:?}", e, $e).unwrap() })* }; } fn main() { let s = read::<usize>(); for i in 0..s { let v = read_vec::<i64>(); let (n, m, x) = (v[0], v[1], v[2]); let ans; if x == 0 { ans = (Modulo(Modulo::get_modulus() + 1 - m).pow(n) + Modulo(1 + m).pow(n)) * Modulo(2).inv(); } else { ans = Modulo(1 + m).pow(n) - (Modulo(Modulo::get_modulus() + 1 - m).pow(n) + Modulo(1 + m).pow(n)) * Modulo(2).inv(); } println!("{}", ans); } } 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() } fn read_vec<T: std::str::FromStr>() -> Vec<T> { read::<String>() .split_whitespace() .map(|e| e.parse().ok().unwrap()) .collect() } #[derive(Clone, Copy, Debug, Default, Eq, Hash, Ord, PartialEq, PartialOrd)] struct Modulo(i64); static mut MODULUS: i64 = 1000_000_000 + 7; impl Modulo { fn set_modulus(m: i64) { unsafe { MODULUS = m; } } fn get_modulus() -> i64 { unsafe { MODULUS } } fn new(x: i64) -> Modulo { let m = Modulo::get_modulus(); if x < 0 { Modulo(x % m + m) } else if x < m { Modulo(x) } else { Modulo(x % m) } } fn pow(self, p: i64) -> Modulo { if p == 0 { Modulo(1) } else { let mut t = self.pow(p / 2); t *= t; if p & 1 == 1 { t *= self; } t } } fn inv(self) -> Modulo { self.pow(Modulo::get_modulus() - 2) } } impl std::fmt::Display for Modulo { fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result { self.0.fmt(f) } } impl std::ops::AddAssign for Modulo { fn add_assign(&mut self, other: Modulo) { let m = Modulo::get_modulus(); self.0 += other.0; if self.0 >= m { self.0 -= m; } } } impl std::ops::MulAssign for Modulo { fn mul_assign(&mut self, other: Modulo) { let m = Modulo::get_modulus(); self.0 *= other.0; self.0 %= m; } } impl std::ops::SubAssign for Modulo { fn sub_assign(&mut self, other: Modulo) { let m = Modulo::get_modulus(); self.0 += m - other.0; if self.0 >= m { self.0 -= m; } } } macro_rules! impl_modulo_ops { ($imp:ident, $method:ident, $assign_imp:ident, $assign_method:ident) => { impl<'a> std::ops::$assign_imp<&'a Modulo> for Modulo { fn $assign_method(&mut self, other: &'a Modulo) { std::ops::$assign_imp::$assign_method(self, *other); } } impl std::ops::$imp for Modulo { type Output = Modulo; fn $method(self, other: Modulo) -> Modulo { let mut x = self; std::ops::$assign_imp::$assign_method(&mut x, other); x } } impl<'a> std::ops::$imp<Modulo> for &'a Modulo { type Output = Modulo; fn $method(self, other: Modulo) -> Modulo { std::ops::$imp::$method(*self, other) } } impl<'a> std::ops::$imp<&'a Modulo> for Modulo { type Output = Modulo; fn $method(self, other: &'a Modulo) -> Modulo { std::ops::$imp::$method(self, *other) } } impl<'a, 'b> std::ops::$imp<&'b Modulo> for &'a Modulo { type Output = Modulo; fn $method(self, other: &'b Modulo) -> Modulo { std::ops::$imp::$method(*self, *other) } } impl std::ops::$assign_imp<i64> for Modulo { fn $assign_method(&mut self, other: i64) { std::ops::$assign_imp::$assign_method(self, Modulo::new(other)); } } impl<'a> std::ops::$assign_imp<&'a i64> for Modulo { fn $assign_method(&mut self, other: &'a i64) { std::ops::$assign_imp::$assign_method(self, *other); } } impl std::ops::$imp<i64> for Modulo { type Output = Modulo; fn $method(self, other: i64) -> Modulo { let mut x = self; std::ops::$assign_imp::$assign_method(&mut x, other); x } } impl<'a> std::ops::$imp<&'a i64> for Modulo { type Output = Modulo; fn $method(self, other: &'a i64) -> Modulo { std::ops::$imp::$method(self, *other) } } impl<'a> std::ops::$imp<i64> for &'a Modulo { type Output = Modulo; fn $method(self, other: i64) -> Modulo { std::ops::$imp::$method(*self, other) } } impl<'a, 'b> std::ops::$imp<&'b i64> for &'a Modulo { type Output = Modulo; fn $method(self, other: &'b i64) -> Modulo { std::ops::$imp::$method(*self, *other) } } }; } impl_modulo_ops!(Add, add, AddAssign, add_assign); impl_modulo_ops!(Mul, mul, MulAssign, mul_assign); impl_modulo_ops!(Sub, sub, SubAssign, sub_assign); use std::iter::Sum; impl Sum for Modulo { fn sum<I>(iter: I) -> Self where I: Iterator<Item = Modulo>, { iter.fold(Modulo(0), |a, b| a + b) } } impl<'a> Sum<&'a Modulo> for Modulo { fn sum<I>(iter: I) -> Self where I: Iterator<Item = &'a Self>, { iter.fold(Modulo(0), |a, b| a + b) } }