結果
問題 | No.1508 Avoid being hit |
ユーザー | akakimidori |
提出日時 | 2021-05-14 22:52:43 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 58 ms / 3,000 ms |
コード長 | 5,544 bytes |
コンパイル時間 | 13,238 ms |
コンパイル使用メモリ | 379,192 KB |
実行使用メモリ | 13,184 KB |
最終ジャッジ日時 | 2024-10-02 07:08:13 |
合計ジャッジ時間 | 16,011 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 26 ms
9,088 KB |
testcase_01 | AC | 1 ms
5,248 KB |
testcase_02 | AC | 1 ms
5,248 KB |
testcase_03 | AC | 1 ms
5,248 KB |
testcase_04 | AC | 1 ms
5,248 KB |
testcase_05 | AC | 1 ms
5,248 KB |
testcase_06 | AC | 1 ms
5,248 KB |
testcase_07 | AC | 1 ms
5,248 KB |
testcase_08 | AC | 1 ms
5,248 KB |
testcase_09 | AC | 1 ms
5,248 KB |
testcase_10 | AC | 1 ms
5,248 KB |
testcase_11 | AC | 1 ms
5,248 KB |
testcase_12 | AC | 1 ms
5,248 KB |
testcase_13 | AC | 1 ms
5,248 KB |
testcase_14 | AC | 1 ms
5,248 KB |
testcase_15 | AC | 1 ms
5,248 KB |
testcase_16 | AC | 15 ms
6,820 KB |
testcase_17 | AC | 9 ms
5,248 KB |
testcase_18 | AC | 9 ms
5,248 KB |
testcase_19 | AC | 12 ms
5,760 KB |
testcase_20 | AC | 23 ms
9,472 KB |
testcase_21 | AC | 15 ms
6,784 KB |
testcase_22 | AC | 18 ms
7,936 KB |
testcase_23 | AC | 20 ms
8,704 KB |
testcase_24 | AC | 28 ms
5,504 KB |
testcase_25 | AC | 25 ms
5,248 KB |
testcase_26 | AC | 3 ms
5,248 KB |
testcase_27 | AC | 13 ms
5,248 KB |
testcase_28 | AC | 13 ms
5,248 KB |
testcase_29 | AC | 26 ms
5,248 KB |
testcase_30 | AC | 15 ms
5,248 KB |
testcase_31 | AC | 22 ms
5,248 KB |
testcase_32 | AC | 24 ms
5,248 KB |
testcase_33 | AC | 7 ms
5,248 KB |
testcase_34 | AC | 45 ms
12,160 KB |
testcase_35 | AC | 15 ms
8,448 KB |
testcase_36 | AC | 30 ms
10,240 KB |
testcase_37 | AC | 20 ms
9,088 KB |
testcase_38 | AC | 42 ms
11,264 KB |
testcase_39 | AC | 27 ms
9,984 KB |
testcase_40 | AC | 49 ms
12,416 KB |
testcase_41 | AC | 31 ms
10,368 KB |
testcase_42 | AC | 48 ms
12,416 KB |
testcase_43 | AC | 58 ms
13,184 KB |
testcase_44 | AC | 1 ms
5,248 KB |
testcase_45 | AC | 1 ms
5,248 KB |
ソースコード
// ---------- begin persistent stack ---------- #[allow(dead_code)] mod persistent_stack { use std::rc::Rc; type Ref<T> = Rc<Node<T>>; type Link<T> = Option<Ref<T>>; struct Node<T> { value: T, next: Link<T>, } impl<T> Node<T> { fn new(value: T, next: Link<T>) -> Self { Node { value, next } } } #[derive(Clone)] pub struct Stack<T> { size: usize, node: Link<T>, } impl<T> Drop for Stack<T> { fn drop(&mut self) { let mut root = self.node.take(); while let Some(Some(node)) = root.as_mut().map(|p| Rc::get_mut(p)) { root = node.next.take(); } } } impl<T> Stack<T> { fn build(size: usize, node: Link<T>) -> Self { Stack { size, node } } pub fn new() -> Self { Stack::build(0, None) } #[must_use] pub fn push(&self, value: T) -> Self { let node = Some(Rc::new(Node::new(value, self.node.clone()))); Stack::build(self.size + 1, node) } pub fn top(&self) -> Option<&T> { self.node.as_ref().map(|p| &p.value) } #[must_use] pub fn pop(&self) -> Option<(Self, &T)> { self.node.as_ref().map(|p| { let s = Stack::build(self.size - 1, p.next.clone()); (s, &p.value) }) } pub fn len(&self) -> usize { self.size } pub fn is_empty(&self) -> bool { self.size == 0 } } pub struct Iter<'a, T> { next: Option<&'a Ref<T>>, } impl<T> Stack<T> { pub fn iter(&self) -> Iter<T> { Iter { next: self.node.as_ref(), } } } impl<'a, T> Iterator for Iter<'a, T> { type Item = &'a T; fn next(&mut self) -> Option<Self::Item> { self.next.take().map(|p| { self.next = p.next.as_ref(); &p.value }) } } } // ---------- end persistent stack ---------- // ---------- begin input macro ---------- // reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8 macro_rules! input { (source = $s:expr, $($r:tt)*) => { let mut iter = $s.split_whitespace(); input_inner!{iter, $($r)*} }; ($($r:tt)*) => { let s = { use std::io::Read; let mut s = String::new(); std::io::stdin().read_to_string(&mut s).unwrap(); s }; let mut iter = s.split_whitespace(); input_inner!{iter, $($r)*} }; } macro_rules! input_inner { ($iter:expr) => {}; ($iter:expr, ) => {}; ($iter:expr, $var:ident : $t:tt $($r:tt)*) => { let $var = read_value!($iter, $t); input_inner!{$iter $($r)*} }; } macro_rules! read_value { ($iter:expr, ( $($t:tt),* )) => { ( $(read_value!($iter, $t)),* ) }; ($iter:expr, [ $t:tt ; $len:expr ]) => { (0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>() }; ($iter:expr, chars) => { read_value!($iter, String).chars().collect::<Vec<char>>() }; ($iter:expr, bytes) => { read_value!($iter, String).bytes().collect::<Vec<u8>>() }; ($iter:expr, usize1) => { read_value!($iter, usize) - 1 }; ($iter:expr, $t:ty) => { $iter.next().unwrap().parse::<$t>().expect("Parse error") }; } // ---------- end input macro ---------- use std::io::Write; fn run() { input! { n: usize, q: usize, a: [usize; q], b: [usize; q], } type Stack = persistent_stack::Stack<(usize, usize)>; let mut dp = vec![Stack::new(); n + 2]; for i in 1..=n { dp[i] = dp[i].push((i, 0)); } let mut range = vec![(1, n + 1)]; for (i, (&a, &b)) in a.iter().zip(b.iter()).enumerate() { let t = i + 1; for p in range.iter_mut() { if p.0 > 1 && dp[p.0 - 1].is_empty() { dp[p.0 - 1] = dp[p.0].push((p.0 - 1, t)); p.0 -= 1; } if p.1 <= n && dp[p.1].is_empty() { dp[p.1] = dp[p.1 - 1].push((p.1, t)); p.1 += 1; } } range.dedup_by(|a, b| { b.1 == a.0 && { b.1 = a.1; true } }); dp[a] = Stack::new(); dp[b] = Stack::new(); for &p in [a, b].iter() { let mut next = vec![]; for (l, r) in range { if l <= p && p < r { if l < p { next.push((l, p)); } if p + 1 < r { next.push((p + 1, r)); } } else { next.push((l, r)); } } range = next; } } if let Some(x) = dp.iter().find(|s| !s.is_empty()) { println!("YES"); let mut pos = vec![0; q + 1]; for &(a, b) in x.iter() { pos[b] = a; } for i in 1..=q { if pos[i] == 0 { pos[i] = pos[i - 1]; } } let out = std::io::stdout(); let mut out = std::io::BufWriter::new(out.lock()); for p in pos.iter() { writeln!(out, "{}", *p).ok(); } } else { println!("NO"); } } fn main() { run(); }