| Task: | Niitty |
| Sender: | EmuBird |
| Submission time: | 2024-11-10 22:27:46 +0200 |
| Language: | Rust (2021) |
| Status: | READY |
| Result: | 0 |
| group | verdict | score |
|---|---|---|
| #1 | RUNTIME ERROR | 0 |
| #2 | RUNTIME ERROR | 0 |
| #3 | RUNTIME ERROR | 0 |
| #4 | RUNTIME ERROR | 0 |
| #5 | RUNTIME ERROR | 0 |
| #6 | RUNTIME ERROR | 0 |
| test | verdict | time | group | |
|---|---|---|---|---|
| #1 | RUNTIME ERROR | 0.00 s | 1, 2, 3, 4, 5, 6 | details |
| #2 | WRONG ANSWER | 0.00 s | 1, 2, 3, 4, 5, 6 | details |
| #3 | RUNTIME ERROR | 0.00 s | 1, 2, 3, 4, 5, 6 | details |
| #4 | RUNTIME ERROR | 0.00 s | 1, 2, 3, 4, 5, 6 | details |
| #5 | ACCEPTED | 0.00 s | 1, 2, 3, 4, 5, 6 | details |
| #6 | RUNTIME ERROR | 0.00 s | 2, 3, 4, 5, 6 | details |
| #7 | RUNTIME ERROR | 0.00 s | 2, 3, 4, 5, 6 | details |
| #8 | RUNTIME ERROR | 0.00 s | 2, 3, 4, 5, 6 | details |
| #9 | WRONG ANSWER | 0.00 s | 2, 3, 4, 5, 6 | details |
| #10 | RUNTIME ERROR | 0.00 s | 3, 4, 5, 6 | details |
| #11 | RUNTIME ERROR | 0.00 s | 3, 4, 5, 6 | details |
| #12 | RUNTIME ERROR | 0.00 s | 3, 4, 5, 6 | details |
| #13 | WRONG ANSWER | 0.01 s | 3, 4, 5, 6 | details |
| #14 | RUNTIME ERROR | 0.00 s | 4, 5, 6 | details |
| #15 | RUNTIME ERROR | 0.00 s | 4, 5, 6 | details |
| #16 | RUNTIME ERROR | 0.00 s | 4, 5, 6 | details |
| #17 | RUNTIME ERROR | 0.00 s | 4, 5, 6 | details |
| #18 | RUNTIME ERROR | 0.01 s | 5, 6 | details |
| #19 | RUNTIME ERROR | 0.01 s | 5, 6 | details |
| #20 | RUNTIME ERROR | 0.01 s | 5, 6 | details |
| #21 | RUNTIME ERROR | 0.01 s | 5, 6 | details |
| #22 | RUNTIME ERROR | 0.01 s | 6 | details |
| #23 | RUNTIME ERROR | 0.01 s | 6 | details |
| #24 | RUNTIME ERROR | 0.01 s | 6 | details |
| #25 | RUNTIME ERROR | 0.01 s | 6 | details |
Code
use std::collections::VecDeque;
use std::{cmp, io};
use std::cell::RefCell;
use std::rc::Rc;
fn main() {
let stdin = io::stdin();
let n: usize = {
let mut input: String = String::new();
stdin.read_line(&mut input).unwrap();
input.trim().parse().unwrap()
};
let mut tree_last_row: Vec<Vec<u32>> = Vec::with_capacity(2 * n);
for _ in 0..n {
let mut row_columns: Vec<u32> = Vec::with_capacity(n);
let mut input: String = String::new();
stdin.read_line(&mut input).unwrap();
let mut chars = input.trim().chars();
for _ in 0..n {
let char = chars.next().unwrap().to_ascii_uppercase();
debug_assert!(char.is_ascii_alphabetic());
let index = get_char_index(char);
let flower = 1 << index;
row_columns.push(flower);
}
let subtree = build_tree(row_columns, sum_flower_cells);
tree_last_row.push(subtree);
}
let tree = build_tree(tree_last_row, sum_flower_rows);
let all_flower_types: u32 = tree[0][0];
let counter = Rc::new(RefCell::new(0u128));
for start_y in 0..n {
for end_y in start_y..n {
let vertical = get_flowers(start_y, end_y, n, &tree, sum_flower_rows);
process_tree(&vertical, 0, 1, 0, &all_flower_types, 0, Some(&counter));
}
}
println!("{}", counter.take());
}
/// Returns a pair (left, right) where first two describe how many selections starting from that side can be made with all flower types included.
fn process_tree(tree: &Vec<u32>, root: usize, root_level_length: usize, root_level_start: usize, all_flower_types: &u32, provided_flower_types: u32, counter: Option<&Rc<RefCell<u128>>>) -> (u128, u128) {
if tree[root_level_start + root] | provided_flower_types != *all_flower_types {
return (0, 0);
}
let second_level_start = root_level_start + root_level_length;
let second_level_len = root_level_length * 2;
let third_level_start = second_level_start + second_level_len;
let left_child_index = root * 2;
let right_child_index = left_child_index + 1;
if tree.len() > second_level_start && tree[second_level_start + right_child_index] == 0 {
// There is only 1 real child (the left one). Calculations can be delegated to that one.
return process_tree(&tree, left_child_index, second_level_len, second_level_start, all_flower_types, provided_flower_types, counter);
}
// Selecting the whole area represented by this node is guaranteed to yield all flowers, thus counters are 1 by default.
let mut left_selections = 1;
let mut right_selections = 1;
if counter.is_some() {
*counter.unwrap().borrow_mut() += 1;
}
if tree.len() > second_level_start {
// This is not the last level.
let left_child = process_tree(&tree, left_child_index, second_level_len, second_level_start, all_flower_types, provided_flower_types, counter);
let right_child = process_tree(&tree, right_child_index, second_level_len, second_level_start, all_flower_types, provided_flower_types, counter);
left_selections += left_child.0;
right_selections += right_child.1;
if tree.len() > third_level_start {
// Root's children have children.
let has_right_right = tree[third_level_start + right_child_index * 2 + 1] != 0; // Whether right has a right child. i.e. left + right's left child isn't the full area.
{
// Try the left child and the right child's left child.
let right_left_child_index = right_child_index * 2;
let right_left_child_node = &tree[third_level_start + right_left_child_index];
let left_child_again = process_tree(&tree, left_child_index, second_level_len, second_level_start, all_flower_types, provided_flower_types | right_left_child_node, None);
let addition = left_child_again.1;
if addition > 1 {
// Selection [left completely] + [right's left child] is possible.
// There can only be one way to select that.
left_selections += 1;
if counter.is_some() {
// Selection [some rightmost children of left] + [right's left child] is possible.
*counter.unwrap().borrow_mut() += addition;
}
}
}
{
// Try the right child and the left child's right child.
let left_right_child_index = left_child_index * 2 + 1;
let left_right_child_node = &tree[third_level_start + left_right_child_index];
let right_child_again = process_tree(&tree, right_child_index, second_level_len, second_level_start, all_flower_types, provided_flower_types | left_right_child_node, None);
let mut addition = right_child_again.0;
if addition > 1 {
if !has_right_right {
// Deduct full area so it isn't counted twice.
addition -= 1;
}
// Selection [right completely] + [left's right child] is possible.
// There can only be one way to select that.
right_selections += 1;
if counter.is_some() {
// Selection [some leftmost children of right] + [left's right child] is possible.
*counter.unwrap().borrow_mut() += addition;
}
}
}
// If left's right and right's left together contain all flowers, they were counted twice. This deducts 1 for that.
{
let right_left_child_index = right_child_index * 2;
let right_left_child_node = &tree[third_level_start + right_left_child_index];
let left_right_child_index = left_child_index * 2 + 1;
let left_right_child_node = &tree[third_level_start + left_right_child_index];
if *right_left_child_node | *left_right_child_node | provided_flower_types == *all_flower_types {
*counter.unwrap().borrow_mut() -= 1;
}
}
}
}
(left_selections, right_selections)
}
fn build_tree<T : Default>(mut bottom_row: Vec<T>, sum_function: fn(&T, &T) -> T) -> Vec<T> {
let bottom_row_len = bottom_row.len().next_power_of_two();
for _ in bottom_row.len()..bottom_row_len {
bottom_row.push(T::default());
}
debug_assert!(bottom_row.len().is_power_of_two());
debug_assert_eq!(bottom_row.len(), bottom_row_len);
let mut tree: VecDeque<T> = VecDeque::with_capacity(bottom_row_len * 2);
tree.extend(bottom_row);
let default_t = T::default();
let mut sum_start = 0;
let mut sum_len = bottom_row_len;
while sum_len > 1 {
let dest_len = sum_len / 2;
for i in (0..dest_len).rev() {
let a = sum_start + i * 2;
let b = a + 1;
let a_item = &tree[a];
let b_item = if b >= (sum_start + sum_len) { &default_t } else { &tree[b] };
tree.push_front(sum_function(a_item, b_item));
sum_start += 1;
}
sum_len = dest_len;
sum_start -= dest_len;
}
Vec::from(tree)
}
fn get_flowers<T : Default>(mut start: usize, mut end: usize, n: usize, tree: &Vec<T>, sum_function: fn(&T, &T) -> T) -> T {
let mut level_len = n.next_power_of_two();
let mut level_start = tree.len() - level_len;
let mut sum = T::default();
while start <= end {
debug_assert!(end < level_len);
if start % 2 == 1 {
sum = sum_function(&sum, &tree[level_start + start]);
start += 1;
}
if end % 2 == 0 {
sum = sum_function(&sum, &tree[level_start + end]);
if end == 0 {
break;
} else {
end -= 1;
}
}
// Move one level up
level_len /= 2;
level_start -= level_len;
start /= 2;
end /= 2;
}
sum
}
// Valid indices start from 1. 0 is reserved for an invalid/dummy state without actual flowers.
fn get_char_index(flower: char) -> u32 {
if 'A' <= flower && flower <= 'Z' {
flower as u32 - 'A' as u32 + 1
} else {
panic!("Illegal flower: {}", flower);
}
}
#[inline]
fn sum_flower_cells(a: &u32, b: &u32) -> u32 {
*a | *b
}
fn sum_flower_rows(a: &Vec<u32>, b: &Vec<u32>) -> Vec<u32> {
let len = cmp::max(a.len(), b.len());
let mut vec = Vec::with_capacity(len);
if a.len() == b.len() {
for i in 0..len {
vec.push(sum_flower_cells(&a[i], &b[i]));
}
} else {
// This would also work for the other case, but this is probably slower.
debug_assert_eq!(cmp::min(a.len(), b.len()), 0);
let mut a_iter = a.iter().chain([0].iter().cycle());
let mut b_iter = b.iter().chain([0].iter().cycle());
for _ in 0..len {
vec.push(sum_flower_cells(a_iter.next().unwrap(), b_iter.next().unwrap()));
}
}
vec
}
Test details
Test 1
Group: 1, 2, 3, 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 10 TNCTNPNTPC NPPNTNTPTP NTNTTCNTCT NPCPNPPNTT ... |
| correct output |
|---|
| 2035 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 2
Group: 1, 2, 3, 4, 5, 6
Verdict: WRONG ANSWER
| input |
|---|
| 10 NFWQLWNWYS DZOQJVXFPJ CNHXPXMCQD QRTBVNLTQC ... |
| correct output |
|---|
| 9 |
| user output |
|---|
| 5 |
Test 3
Group: 1, 2, 3, 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 10 XXXXXXXXXX XXXXXXXXXX XXXXXXXXXX XXXXXXXXXX ... |
| correct output |
|---|
| 3025 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 4
Group: 1, 2, 3, 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 10 FFFFFFFFFF FFFFFCFFFF FFFFFFJFFF FFFFFFFFFF ... |
| correct output |
|---|
| 12 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 5
Group: 1, 2, 3, 4, 5, 6
Verdict: ACCEPTED
| input |
|---|
| 1 X |
| correct output |
|---|
| 1 |
| user output |
|---|
| 1 |
Test 6
Group: 2, 3, 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 20 BBCBUBOUOBBCUUBBCOUO BOUCOOCUBCOOOCOBOCUO UCCUUUOBCOCBCBUBUCOO BUOBUCUCUOOBCOOUBUOO ... |
| correct output |
|---|
| 38724 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 7
Group: 2, 3, 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 20 CBGLSHGZHYZDWBNDBJUG SMUXOJQYPXZDTMJUIWOJ XIDSTNBGHKRKOVUVMINB MTQGCFRUHQKALXRNCQGS ... |
| correct output |
|---|
| 8334 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 8
Group: 2, 3, 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 20 KKKKKKKKKKKKKKKKKKKK KKKKKKKKKKKKKKKKKKKK KKKKKKKKKKKKKKKKKKKK KKKKKKKKKKKKKKKKKKKK ... |
| correct output |
|---|
| 44100 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 9
Group: 2, 3, 4, 5, 6
Verdict: WRONG ANSWER
| input |
|---|
| 20 AAAAAAAAXAAAAAAAAAAA AAAWAAAAAAAAAAAAAOAA AAAAAAAAAAAAAAAAAPAA AAAAAAAAKAAAAAAAAAAZ ... |
| correct output |
|---|
| 18 |
| user output |
|---|
| 4 |
Test 10
Group: 3, 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 50 GRGREEEGREGXRXXEGXXREXGRRRGRRR... |
| correct output |
|---|
| 1584665 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 11
Group: 3, 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 50 AITIISJUHCCRZNKSDCNQKYSQRINFWJ... |
| correct output |
|---|
| 1077746 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 12
Group: 3, 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 50 OOOOOOOOOOOOOOOOOOOOOOOOOOOOOO... |
| correct output |
|---|
| 1625625 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 13
Group: 3, 4, 5, 6
Verdict: WRONG ANSWER
| input |
|---|
| 50 FFFFFFFFFFFFFFFFFFFFFFFFFFFFFF... |
| correct output |
|---|
| 1680 |
| user output |
|---|
| 56 |
Test 14
Group: 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 100 NNCMDCDDCCNNNDNCMMNCDCDCCDCDNM... |
| correct output |
|---|
| 25325366 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 15
Group: 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 100 LIMQQIHASECROEVILNVULGWZJPPKOG... |
| correct output |
|---|
| 22342463 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 16
Group: 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 100 TTTTTTTTTTTTTTTTTTTTTTTTTTTTTT... |
| correct output |
|---|
| 25502500 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 17
Group: 4, 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 100 QXQQQQQQQQQQQQQQQQQQQQQQQQQQQQ... |
| correct output |
|---|
| 25650 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 18
Group: 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 200 NAANANMMKNKKAKMKMAKNKMNKMMNNAA... |
| correct output |
|---|
| 403292767 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 19
Group: 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 200 OMYWATTLURKQPTKEFMGGYAOONXWVSC... |
| correct output |
|---|
| 388111321 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 20
Group: 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 200 CCCCCCCCCCCCCCCCCCCCCCCCCCCCCC... |
| correct output |
|---|
| 404010000 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 21
Group: 5, 6
Verdict: RUNTIME ERROR
| input |
|---|
| 200 LLLLLLLLLLLLLLLLLHLLLLLLLLLLLL... |
| correct output |
|---|
| 14159445 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 22
Group: 6
Verdict: RUNTIME ERROR
| input |
|---|
| 500 VVHWVUHVHUWWWVUUUWVUUHUUWHWUVW... |
| correct output |
|---|
| 15683003812 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 23
Group: 6
Verdict: RUNTIME ERROR
| input |
|---|
| 500 OIMZGEQSBMBDSDXSWRFNKSGFEBBTJE... |
| correct output |
|---|
| 15575906951 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 24
Group: 6
Verdict: RUNTIME ERROR
| input |
|---|
| 500 IIIIIIIIIIIIIIIIIIIIIIIIIIIIII... |
| correct output |
|---|
| 15687562500 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Test 25
Group: 6
Verdict: RUNTIME ERROR
| input |
|---|
| 500 WWWWWWWWWWWWWWWWWWWWWWWWWWWWWW... |
| correct output |
|---|
| 3058970930 |
| user output |
|---|
| (empty) |
Error:
thread 'main' panicked at 'called `Option::unwrap()` on a `None` value', input/code.rs:137:30 note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
