結果
問題 | No.1299 Random Array Score |
ユーザー |
![]() |
提出日時 | 2020-11-28 00:53:51 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 22 ms / 2,000 ms |
コード長 | 5,750 bytes |
コンパイル時間 | 12,893 ms |
コンパイル使用メモリ | 398,840 KB |
実行使用メモリ | 9,184 KB |
最終ジャッジ日時 | 2024-07-26 21:31:54 |
合計ジャッジ時間 | 15,376 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
コンパイルメッセージ
warning: unused variable: `n` --> src/main.rs:20:10 | 20 | let (n, k) = (v[0], v[1]); | ^ help: if this is intentional, prefix it with an underscore: `_n` | = note: `#[warn(unused_variables)]` on by default warning: associated items `set_modulus` and `inv` are never used --> src/main.rs:46:8 | 45 | impl Modulo { | ----------- associated items in this implementation 46 | fn set_modulus(m: i128) { | ^^^^^^^^^^^ ... 76 | fn inv(self) -> Modulo { | ^^^ | = note: `#[warn(dead_code)]` on by default
ソースコード
#![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 v = read_vec::<i128>();let (n, k) = (v[0], v[1]);let a = read_vec::<i128>().iter().map(|&x| Modulo(x)).sum::<Modulo>();let ans = a * Modulo(2).pow(k);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(i128);static mut MODULUS: i128 = 998244353;impl Modulo {fn set_modulus(m: i128) {unsafe {MODULUS = m;}}fn get_modulus() -> i128 {unsafe { MODULUS }}fn new(x: i128) -> 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: i128) -> 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<i128> for Modulo {fn $assign_method(&mut self, other: i128) {std::ops::$assign_imp::$assign_method(self, Modulo::new(other));}}impl<'a> std::ops::$assign_imp<&'a i128> for Modulo {fn $assign_method(&mut self, other: &'a i128) {std::ops::$assign_imp::$assign_method(self, *other);}}impl std::ops::$imp<i128> for Modulo {type Output = Modulo;fn $method(self, other: i128) -> Modulo {let mut x = self;std::ops::$assign_imp::$assign_method(&mut x, other);x}}impl<'a> std::ops::$imp<&'a i128> for Modulo {type Output = Modulo;fn $method(self, other: &'a i128) -> Modulo {std::ops::$imp::$method(self, *other)}}impl<'a> std::ops::$imp<i128> for &'a Modulo {type Output = Modulo;fn $method(self, other: i128) -> Modulo {std::ops::$imp::$method(*self, other)}}impl<'a, 'b> std::ops::$imp<&'b i128> for &'a Modulo {type Output = Modulo;fn $method(self, other: &'b i128) -> 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) -> SelfwhereI: Iterator<Item = Modulo>,{iter.fold(Modulo(0), |a, b| a + b)}}impl<'a> Sum<&'a Modulo> for Modulo {fn sum<I>(iter: I) -> SelfwhereI: Iterator<Item = &'a Self>,{iter.fold(Modulo(0), |a, b| a + b)}}