結果
問題 | No.1097 Remainder Operation |
ユーザー |
|
提出日時 | 2020-06-28 09:13:01 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 305 ms / 2,000 ms |
コード長 | 3,072 bytes |
コンパイル時間 | 18,672 ms |
コンパイル使用メモリ | 404,344 KB |
実行使用メモリ | 67,072 KB |
最終ジャッジ日時 | 2024-07-07 03:24:54 |
合計ジャッジ時間 | 16,593 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 |
ソースコード
#![allow(non_snake_case)]#![allow(dead_code)]#![allow(unused_macros)]#![allow(unused_imports)]use std::str::FromStr;use std::io::*;use std::collections::*;use std::cmp::*;struct Scanner<I: Iterator<Item = char>> {iter: std::iter::Peekable<I>,}macro_rules! exit {() => {{exit!(0)}};($code:expr) => {{if cfg!(local) {writeln!(std::io::stderr(), "===== Terminated =====").expect("failed printing to stderr");}std::process::exit($code);}}}impl<I: Iterator<Item = char>> Scanner<I> {pub fn new(iter: I) -> Scanner<I> {Scanner {iter: iter.peekable(),}}pub fn safe_get_token(&mut self) -> Option<String> {let token = self.iter.by_ref().skip_while(|c| c.is_whitespace()).take_while(|c| !c.is_whitespace()).collect::<String>();if token.is_empty() {None} else {Some(token)}}pub fn token(&mut self) -> String {self.safe_get_token().unwrap_or_else(|| exit!())}pub fn get<T: FromStr>(&mut self) -> T {self.token().parse::<T>().unwrap_or_else(|_| exit!())}pub fn vec<T: FromStr>(&mut self, len: usize) -> Vec<T> {(0..len).map(|_| self.get()).collect()}pub fn mat<T: FromStr>(&mut self, row: usize, col: usize) -> Vec<Vec<T>> {(0..row).map(|_| self.vec(col)).collect()}pub fn char(&mut self) -> char {self.iter.next().unwrap_or_else(|| exit!())}pub fn chars(&mut self) -> Vec<char> {self.get::<String>().chars().collect()}pub fn mat_chars(&mut self, row: usize) -> Vec<Vec<char>> {(0..row).map(|_| self.chars()).collect()}pub fn line(&mut self) -> String {if self.peek().is_some() {self.iter.by_ref().take_while(|&c| !(c == '\n' || c == '\r')).collect::<String>()} else {exit!();}}pub fn peek(&mut self) -> Option<&char> {self.iter.peek()}}const MAX: usize = 40;fn main() {let cin = stdin();let cin = cin.lock();let mut sc = Scanner::new(cin.bytes().map(|c| c.unwrap() as char));let N: usize = sc.get();let A: Vec<usize> = sc.vec(N);let mut to = vec![vec![0; N]; MAX+1];let mut sum = vec![vec![0; N]; MAX+1];for i in 0..N {let nx = (i + A[i]) % N;to[0][i] = nx;sum[0][i] = A[i];}for k in 0..MAX {for i in 0..N {to[k+1][i] = to[k][to[k][i]];sum[k+1][i] = sum[k][i] + sum[k][to[k][i]];}}let Q: usize = sc.get();for _ in 0..Q {let K: usize = sc.get();let mut now = 0;let mut ans = 0;for i in 0..MAX {if (K>>i) & 1 != 0 {ans += sum[i][now];now = to[i][now];}}println!("{}", ans);}}