QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#308773 | #8130. Yet Another Balanced Coloring Problem | ucup-team635# | AC ✓ | 64ms | 20292kb | Rust | 4.5kb | 2024-01-20 12:34:47 | 2024-01-20 12:34:48 |
Judging History
answer
// ---------- begin quick find ----------
pub struct QuickFind {
size: usize,
id: Vec<usize>,
list: Vec<Vec<usize>>,
}
impl QuickFind {
pub fn new(size: usize) -> Self {
let id = (0..size).collect::<Vec<_>>();
let list = (0..size).map(|x| vec![x]).collect::<Vec<_>>();
QuickFind { size, id, list }
}
pub fn root(&self, x: usize) -> usize {
assert!(x < self.size);
self.id[x]
}
pub fn same(&self, x: usize, y: usize) -> bool {
assert!(x < self.size);
assert!(y < self.size);
self.root(x) == self.root(y)
}
pub fn unite(&mut self, x: usize, y: usize) -> Option<(usize, usize)> {
assert!(x < self.size);
assert!(y < self.size);
let mut x = self.root(x);
let mut y = self.root(y);
if x == y {
return None;
}
if (self.list[x].len(), x) < (self.list[y].len(), y) {
std::mem::swap(&mut x, &mut y);
}
let mut z = std::mem::take(&mut self.list[y]);
z.iter().for_each(|y| self.id[*y] = x);
self.list[x].append(&mut z);
Some((x, y))
}
pub fn enumerate(&self, x: usize) -> &[usize] {
assert!(x < self.size);
&self.list[self.root(x)]
}
pub fn size(&self, x: usize) -> usize {
assert!(x < self.size);
self.enumerate(x).len()
}
}
// ---------- end quick find ----------
// ---------- begin scannner ----------
#[allow(dead_code)]
mod scanner {
use std::str::FromStr;
pub struct Scanner<'a> {
it: std::str::SplitWhitespace<'a>,
}
impl<'a> Scanner<'a> {
pub fn new(s: &'a String) -> Scanner<'a> {
Scanner {
it: s.split_whitespace(),
}
}
pub fn next<T: FromStr>(&mut self) -> T {
self.it.next().unwrap().parse::<T>().ok().unwrap()
}
pub fn next_bytes(&mut self) -> Vec<u8> {
self.it.next().unwrap().bytes().collect()
}
pub fn next_chars(&mut self) -> Vec<char> {
self.it.next().unwrap().chars().collect()
}
pub fn next_vec<T: FromStr>(&mut self, len: usize) -> Vec<T> {
(0..len).map(|_| self.next()).collect()
}
}
}
// ---------- end scannner ----------
use std::io::Write;
use std::collections::*;
type Map<K, V> = BTreeMap<K, V>;
type Set<T> = BTreeSet<T>;
type Deque<T> = VecDeque<T>;
fn main() {
use std::io::Read;
let mut s = String::new();
std::io::stdin().read_to_string(&mut s).unwrap();
let mut sc = scanner::Scanner::new(&s);
let out = std::io::stdout();
let mut out = std::io::BufWriter::new(out.lock());
run(&mut sc, &mut out);
}
fn run<W: Write>(sc: &mut scanner::Scanner, out: &mut std::io::BufWriter<W>) {
let t: u32 = sc.next();
for _ in 0..t {
let n: usize = sc.next();
let m: usize = sc.next();
let p = sc.next_vec::<usize>(n - 1);
let q = sc.next_vec::<usize>(m - 1);
let ans = solve(n, m, p, q);
writeln!(out, "{}", ans).ok();
}
}
fn solve(n: usize, m: usize, p: Vec<usize>, q: Vec<usize>) -> String {
let x = calc(n, p.clone());
let y = calc(m, q);
let k = p.iter().min().unwrap() - 1;
let mut dsu = QuickFind::new(2 * k);
for &(x, y) in x.iter().chain(y.iter()) {
dsu.unite(x, k + y);
dsu.unite(y, k + x);
}
let mut used = vec![false; 2 * k];
let mut ans = vec![0; k];
for i in 0..k {
let r = dsu.root(i);
if used[r] {
continue;
}
used[r] = true;
used[dsu.root(i + k)] = true;
for &v in dsu.enumerate(i) {
if v >= k {
ans[v - k] = 1;
}
}
}
ans.iter().map(|a| if *a == 0 {'B'} else {'R'}).collect()
}
fn calc(n: usize, p: Vec<usize>) -> Vec<(usize, usize)> {
let mut g = vec![vec![]; n];
for (i, p) in p.iter().enumerate() {
g[*p - 1].push(i);
}
let mut ans = vec![];
let mut dp = vec![n; n];
for v in 0..n {
if g[v].is_empty() {
dp[v] = v;
continue;
}
let mut c = g[v].iter().map(|p| dp[*p]).collect::<Vec<_>>();
c.retain(|c| *c < n);
while c.len() >= 2 {
let a = c.pop().unwrap();
let b = c.pop().unwrap();
ans.push((a, b));
}
dp[v] = c.pop().map_or(n, |p| p);
}
ans
}
这程序好像有点Bug,我给组数据试试?
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 2168kb
input:
2 7 7 5 5 6 6 7 7 5 6 5 6 7 7 5 4 4 4 5 5 4 4 4
output:
BRRB BRB
result:
ok ok (2 test cases)
Test #2:
score: 0
Accepted
time: 13ms
memory: 2484kb
input:
10000 6 6 5 5 5 5 6 5 6 6 6 6 9 6 7 9 7 7 6 8 9 9 6 6 6 6 6 9 8 9 9 8 7 7 8 8 9 7 7 8 7 7 7 8 6 10 4 5 5 6 6 4 6 5 7 8 9 8 9 10 6 9 6 6 6 6 6 6 9 7 8 8 9 9 9 8 9 8 6 6 7 6 7 8 7 9 7 6 7 8 9 9 9 8 6 7 7 5 8 8 8 9 7 5 6 5 8 7 8 8 7 6 6 5 5 6 7 8 5 7 6 6 7 7 9 9 8 9 8 8 8 8 9 9 9 9 8 8 9 9 8 9 9 8 8 8 ...
output:
BRBR BBRBR BRRBRB BBR BBRBR BBBRR BBRR BRBR BBRBRBR BRBBRRB BRB BRBBRR BBR BBRBRBR BRRRBB BRRB BBR BRBRB BBR BRBRB BBRBR BRBR BRBBR BBR BRBRB BBRR BRBRBR BRBRRB BRBR BRBRBR BBRRBR BBRRB BRBRBR BRRBBR BRBRBR BRR BRRB BRR BBR BBRBR BBRR BRBRBR BRB BBRBR BRR BRBRBR BRBRBR BBR BRBRBR BRB BRR BBRR BRBBR ...
result:
ok ok (10000 test cases)
Test #3:
score: 0
Accepted
time: 14ms
memory: 2732kb
input:
1000 98 96 41 39 52 47 34 37 45 33 68 54 74 35 65 58 49 46 53 42 87 30 43 48 38 36 56 40 88 66 32 31 72 44 91 96 51 85 83 61 60 59 80 63 70 80 75 61 51 83 50 69 86 55 79 62 67 57 73 93 96 64 69 91 78 73 80 83 81 91 91 71 76 81 75 90 92 77 82 89 82 86 98 84 96 89 97 96 91 97 94 93 95 97 97 95 96 97 9...
output:
BBRBRRBBRBBRBRRRRRBBBBBBRBRRR BRBRBRRRBBBBBRRRBBRBBRRR BBBRRRBRBBBBBBRRRBBRBRBRRRRRRRBBBBBRBBBRRRRBRBBRBRRBBBBRRBRBBRRRRRRBBBBBRRRRRBB BBRBBBBBRRBRRRRRBRBBRBRRB BBRRRBBBBRRBRBRRRRRBBRBRBRRBBBBBRBRBRBRRBRBRRBRBBR BBBRRBBBBRRBBBRBBRRRRRRRRRBBBBRRRBRB BRRBRBRBBRRRBBRBBBBRBRBBBRRRBBRRRBRRRB BRBRRBR BBRR...
result:
ok ok (1000 test cases)
Test #4:
score: 0
Accepted
time: 21ms
memory: 4812kb
input:
10 9442 9473 6729 7355 6467 7301 7964 7025 7066 7206 8711 8044 7401 6634 6594 9405 7767 7253 7611 6730 6630 8250 6872 6720 8868 8644 9280 7272 6808 8887 7965 7384 6376 9115 8340 7618 8377 9351 8690 8842 9014 6913 7207 7552 8087 9013 9340 6509 8152 6963 8666 8716 7681 6447 8097 7014 6854 8576 6915 92...
output:
BRBBBRBRRBRBBRBBRBBBBBRRBBRBRBBRBRRRRBBRBRBBRRRBBRBBRBBRBBRBRBBBBRBRBBBBBBRRBRRBRBBBBBRRRBRBRRRBBRBBBBBRBRRBBBBRBBBRRRRRRBBBBRRRRBBRBRRRBRBBBRBBRBRBBRRRBBBBBBBBBRBRBRRBRBBRRRBRRBBRRRRRRRBRBRRRRRBRBBBBBBBRBRRBBRRBBRRRRRBRBRBRBRRRRRBRRRBRBBBBRRBRBBRRRRBBRBBBRRRBRBRBRRRRRRRBBRRRBBBBRBRRBRBBRBBRBRBRRBRB...
result:
ok ok (10 test cases)
Test #5:
score: 0
Accepted
time: 13ms
memory: 13180kb
input:
1 100000 100000 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 9...
output:
B
result:
ok ok (1 test case)
Test #6:
score: 0
Accepted
time: 16ms
memory: 13112kb
input:
1 100000 100000 27 17 44 12 22 19 14 21 15 11 48 13 16 20 34 18 24 26 25 28 43 23 33 29 31 30 46 45 41 36 32 38 90 35 40 37 39 55 47 42 59 52 65 72 49 50 54 53 51 64 56 57 66 58 63 82 62 61 60 69 86 95 85 71 68 67 83 70 74 73 77 75 81 76 78 88 89 79 80 84 94 96 123 106 110 87 92 91 99 102 93 98 101 ...
output:
BBBBRBRRRR
result:
ok ok (1 test case)
Test #7:
score: 0
Accepted
time: 18ms
memory: 13240kb
input:
1 100000 100000 218 381 317 660 186 296 679 224 357 361 193 231 232 288 310 183 209 206 182 300 400 344 250 440 519 563 203 333 346 543 427 447 245 289 202 386 221 249 256 359 257 370 200 393 263 270 340 191 240 644 489 522 380 453 241 207 345 719 261 336 364 260 211 541 178 215 383 220 886 403 441 ...
output:
BBRBRRBRBRBRBBRRBRRRRBBBBRBRRRRBBBRRBRBRRBRRBBRBRBBBRBBBRBBBRRBRRRRRRRBRBRBBRRBBBRBBBRBRRRBBBRRRRRBRBBBBBBBBRRRRRBRRBRBRRBBRBBRBRBBRBRBBRBBRBRBBBBRBBRBRRRRBBBRRBRBRRRBRBRRRBBRR
result:
ok ok (1 test case)
Test #8:
score: 0
Accepted
time: 13ms
memory: 13168kb
input:
1 100000 100000 1421 2788 1423 1160 1112 1488 1236 5948 1163 1757 1410 2312 1361 2923 1593 1342 1999 1293 1081 2501 1942 2294 1893 1059 1610 1844 1256 1287 3866 2443 1668 2351 2721 1191 1618 1058 2556 1182 4821 1303 1528 2149 1564 1294 2339 1490 2179 3047 1645 1178 1536 3950 1275 8455 1068 2870 2234...
output:
BBBRRRBBBBBBRRBBRBRRRBRRRBBBRRBRBBBBRRBBBRBBRRRBRBRBBBBBRRRBRRRRBBRBBRRBRRBRRRBRRBRRBRRRRBBBBRRBBRRRRRRRRRBBBRRRRRBRRBBRRBRRRRRRBRRBRRRRBBRRBBBRRBRRRBBBBRBBBRBRBRRRBBRBRBBRBRBBRRBBRRBRBBBBBBBRBBRBBRBBBRRBRBRBRRBBBRRBRBRBRRRBRBRBBRBBRBRBBRBBBRRBBRRRRRBRBRBBRRRBBBBBBRRRRBBBBRBBBRBRBBBBRBRRRBRBRRBRBRRB...
result:
ok ok (1 test case)
Test #9:
score: 0
Accepted
time: 26ms
memory: 12864kb
input:
1 100000 100000 13732 23027 27029 18787 13613 22155 13102 24134 16692 14811 14984 38995 53389 15109 14007 13571 39616 23329 57926 78696 15054 18000 18665 15849 48065 15896 12897 14354 18446 27792 19390 19744 32923 16980 18489 27104 17926 43496 25322 21020 16210 38766 16458 32680 31819 51489 16147 16...
output:
BBBBRBBRBBRRRRBBBBRBBRBBRRRRRBBBBBRBBRRBRBRBRRBBRBBRBRBBRRBBBBRBRBBBRBBBRRRRRBRBBRRBRRBBBBBRRBBBBBRRRBBBBRBRRRBRRRRBBRRBBRRBRRRBBRRBRBRRRRBBRRBBBRBRBRBBBBRRBBBBRBBBRRRRRRBRRRBRBRBRRRRRRRBBBBRBBBRBRBBBBRRRBBBRRBRRRRRRBBRRBRBRBBRBRBRBRBRRBRBRRRBBRRBRRRRRBBRRRBBBRBRRBBBBBRBRBBRBBRRBRBRBRRBRBRRRRRBBBRRR...
result:
ok ok (1 test case)
Test #10:
score: 0
Accepted
time: 64ms
memory: 20292kb
input:
1 100000 100000 96980 98039 98580 93297 93757 97307 99757 94452 92927 98214 98011 93924 93860 95890 92565 93639 93376 92841 93009 93010 98649 93905 99583 97162 96430 99638 99287 94535 94294 98896 92712 98796 96197 94388 99111 96453 94989 97747 95737 96731 97796 98914 95663 93712 93423 94795 96869 98...
output:
BBBBRBBRRBBBRBRRRRRBRRBRBBBBBBRBBRBRRRBRBRRRBRRRRRRRBRRRBBBBRBRRRRBRBRRRRRRBRBBRBBBBBRRBBRBRRBBBRRBRRRRRBRBBRRRRRRRBBBRRRBRRBBBRBRRRRRBRRBRBRBBRBRRBRBRRBRRBRBRRRBRBRRBBBBRBBBRBBRRBBBRRRBBBBRBRRRBBBBBRRBBBRBRBBRRRRRRRRRBBRRBBBRBRBBBBBBBRBRBBBBRRRRBBBRBRRBRRBRBRBRRRRBBBRRBRBBRBBRBRRRRRBRRBRBBRBBRBBBBR...
result:
ok ok (1 test case)
Test #11:
score: 0
Accepted
time: 10ms
memory: 11820kb
input:
1 100000 30 51 28 32 13 22 18 19 20 12 25 21 14 15 16 23 17 29 33 27 24 39 35 26 34 37 46 42 31 30 36 41 40 38 50 48 71 44 79 45 58 43 68 47 93 54 49 53 60 66 57 52 55 63 56 62 67 76 59 64 61 78 65 69 74 70 89 75 118 77 72 73 81 80 84 111 95 87 82 98 85 83 86 90 96 121 91 88 94 92 99 97 100 101 102 ...
output:
BRRBBBRRRRB
result:
ok ok (1 test case)
Test #12:
score: 0
Accepted
time: 0ms
memory: 2172kb
input:
1 3 3 3 3 3 3
output:
BR
result:
ok ok (1 test case)
Extra Test:
score: 0
Extra Test Passed