結果
問題 | No.1313 N言っちゃダメゲーム (4) |
ユーザー | ikd |
提出日時 | 2020-12-10 20:35:53 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 6 ms / 2,000 ms |
コード長 | 2,433 bytes |
コンパイル時間 | 13,466 ms |
コンパイル使用メモリ | 399,712 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-09-19 20:41:04 |
合計ジャッジ時間 | 13,611 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 35 |
ソースコード
fn main() { let stdin = std::io::stdin(); let mut rd = ProconReader::new(stdin.lock()); let n: usize = rd.get(); let k: usize = rd.get(); let s: String = rd.get(); let s: Vec<char> = s.chars().collect(); let s = [vec!['o'], s].concat(); let mut win = vec![0; n + k]; for i in n..(n + k) { win[i] = 1; } let mut sum = k; for i in (1..=(n - 1)).rev() { if s[i] == 'x' { win[i] = 1; } else if sum < k { win[i] = 1; } else { win[i] = 0; } sum += win[i]; sum -= win[i + k]; } let ans: Vec<usize> = (1..=k).filter(|&i| win[i] == 0).collect(); if ans.is_empty() { println!("{}", 0); } else { ans.iter().for_each(|a| { println!("{}", a); }); } } pub struct ProconReader<R> { r: R, line: String, i: usize, } impl<R: std::io::BufRead> ProconReader<R> { pub fn new(reader: R) -> Self { Self { r: reader, line: String::new(), i: 0, } } pub fn get<T>(&mut self) -> T where T: std::str::FromStr, <T as std::str::FromStr>::Err: std::fmt::Debug, { self.skip_blanks(); assert!(self.i < self.line.len()); assert_ne!(&self.line[self.i..=self.i], " "); let line = &self.line[self.i..]; let end = line.find(' ').unwrap_or(line.len()); let s = &line[..end]; self.i += end; s.parse() .unwrap_or_else(|_| panic!("parse error `{}`", self.line)) } fn skip_blanks(&mut self) { loop { let start = self.line[self.i..].find(|ch| ch != ' '); match start { Some(j) => { self.i += j; break; } None => { self.line.clear(); self.i = 0; let num_bytes = self.r.read_line(&mut self.line).unwrap(); assert!(num_bytes > 0, "reached EOF :("); self.line = self.line.trim_end_matches(&['\r', '\n'][..]).to_string(); } } } } pub fn get_vec<T>(&mut self, n: usize) -> Vec<T> where T: std::str::FromStr, <T as std::str::FromStr>::Err: std::fmt::Debug, { (0..n).map(|_| self.get()).collect() } }