QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#511799 | #9170. Cycle Game | ucup-team635# | WA | 0ms | 2264kb | Rust | 6.6kb | 2024-08-10 11:22:55 | 2024-08-10 11:22:55 |
Judging History
answer
use std::collections::*;
use std::io::Write;
type Map<K, V> = BTreeMap<K, V>;
type Set<T> = BTreeSet<T>;
type Deque<T> = VecDeque<T>;
fn main() {
input! {
h: usize,
w: usize,
q: usize,
p: [(usize, usize); q],
}
let mut z = p.clone();
z.sort();
z.dedup();
let index = |x, y| {
let pos = z.binary_search(&(x, y)).unwrap();
pos
};
let mut dsu = DSU::new(z.len());
let mut set = Set::<(usize, usize)>::new();
let mut ans = vec![];
for (x, y) in p {
let mut u = DSU::new(9);
let pos = |i, j| i * 3 + j;
let elem = |i, j| set.contains(&(x - 1 + i, y - 1 + j));
for &(dx, dy) in [(0, 1), (1, 0), (1, 1), (1, !0)].iter() {
for i in 0..3 {
for j in 0..3 {
let nx = i + dx;
let ny = j + dy;
if nx < 3 && ny < 3 && elem(i, j) && elem(nx, ny) {
u.unite(pos(i, j), pos(nx, ny));
}
}
}
}
let mut c = vec![];
for i in 0..3 {
for j in 0..3 {
if elem(i, j) {
c.push((dsu.root(index(x - 1 + i, y - 1 + j)), u.root(pos(i, j))));
}
}
}
c.sort();
c.dedup();
if c.windows(2).any(|c| c[0].0 == c[1].0) && h.min(w) > 1 {
ans.push(0);
} else {
ans.push(1);
let v = index(x, y);
for i in 0..3 {
for j in 0..3 {
if set.contains(&(x - 1 + i, y - 1 + j)) {
let u = index(x - 1 + i, y - 1 + j);
dsu.unite(v, u);
}
}
}
set.insert((x, y));
}
}
use util::*;
println!("{}", ans.iter().join(""));
}
// ---------- begin input macro ----------
// reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8
#[macro_export]
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_export]
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_export]
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 ----------
// undo をuniteした情報がないときに呼ぶとREになる
// snap を取るとそれまでのunite した情報が消える
//---------- begin union_find ----------
pub struct DSU {
parent: Vec<u32>,
size: Vec<u32>,
stack: Vec<Option<(u32, u32)>>,
}
impl DSU {
pub fn new(n: usize) -> DSU {
assert!(n < std::u32::MAX as usize);
let mut res = DSU {
parent: vec![0; n],
size: vec![1; n],
stack: vec![],
};
res.init();
res
}
pub fn init(&mut self) {
self.stack.clear();
for (i, (parent, size)) in self.parent.iter_mut().zip(self.size.iter_mut()).enumerate() {
*parent = i as u32;
*size = 1;
}
}
pub fn root(&self, mut x: usize) -> usize {
assert!(x < self.parent.len());
while self.parent[x] != x as u32 {
x = self.parent[x] as usize;
}
x
}
pub fn same(&self, x: usize, y: usize) -> bool {
assert!(x < self.parent.len());
assert!(y < self.parent.len());
self.root(x) == self.root(y)
}
pub fn unite(&mut self, x: usize, y: usize) -> Option<(usize, usize)> {
assert!(x < self.parent.len());
assert!(y < self.parent.len());
let mut x = self.root(x);
let mut y = self.root(y);
if x == y {
self.stack.push(None);
return None;
}
if (self.size[x], x) < (self.size[y], y) {
std::mem::swap(&mut x, &mut y);
}
self.size[x] += self.size[y];
self.parent[y] = x as u32;
self.stack.push(Some((x as u32, y as u32)));
Some((x, y))
}
pub fn parent(&self, x: usize) -> Option<usize> {
assert!(x < self.parent.len());
let p = self.parent[x];
if p != x as u32 {
Some(p as usize)
} else {
None
}
}
pub fn size(&self, x: usize) -> usize {
assert!(x < self.parent.len());
let r = self.root(x);
self.size[r] as usize
}
pub fn undo(&mut self) -> Option<(usize, usize)> {
self.stack.pop().unwrap().map(|(x, y)| {
let x = x as usize;
let y = y as usize;
self.size[x] -= self.size[y];
self.parent[y] = y as u32;
(x, y)
})
}
pub fn snap(&mut self) {
self.stack.clear();
}
pub fn rollback(&mut self) {
while !self.stack.is_empty() {
self.undo();
}
}
}
//---------- end union_find ----------
mod util {
pub trait Join {
fn join(self, sep: &str) -> String;
}
impl<T, I> Join for I
where
I: Iterator<Item = T>,
T: std::fmt::Display,
{
fn join(self, sep: &str) -> String {
let mut s = String::new();
use std::fmt::*;
for (i, v) in self.enumerate() {
if i > 0 {
write!(&mut s, "{}", sep).ok();
}
write!(&mut s, "{}", v).ok();
}
s
}
}
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 2100kb
input:
4 3 7 2 1 2 2 2 3 3 1 3 2 4 1 4 2
output:
1111111
result:
ok "1111111"
Test #2:
score: 0
Accepted
time: 0ms
memory: 2264kb
input:
3 3 8 1 1 1 2 1 3 2 3 3 3 3 2 3 1 2 1
output:
11111110
result:
ok "11111110"
Test #3:
score: 0
Accepted
time: 0ms
memory: 2156kb
input:
10 10 7 9 1 6 6 3 8 8 7 5 10 1 7 1 2
output:
1111111
result:
ok "1111111"
Test #4:
score: -100
Wrong Answer
time: 0ms
memory: 2072kb
input:
9 10 50 1 9 1 6 2 3 3 1 7 4 9 4 1 3 2 5 9 2 7 9 5 6 8 10 9 5 5 5 4 10 9 7 5 9 3 2 4 5 1 1 4 7 3 6 2 8 4 3 8 6 5 10 4 8 5 4 7 2 9 6 4 2 7 8 5 2 3 5 9 1 6 1 1 5 9 9 5 8 6 3 8 8 8 4 7 7 7 1 3 7 2 2 3 10 6 9 8 3 7 6
output:
11111111111111111111101111111111111111100111111101
result:
wrong answer 1st words differ - expected: '11111111111111111111111111111111111111111111111111', found: '11111111111111111111101111111111111111100111111101'