CSES - Datatähti 2025 alku - Results
Submission details
Task:Niitty
Sender:Saksunoob
Submission time:2024-11-01 14:15:01 +0200
Language:Rust (2021)
Status:READY
Result:58
Feedback
groupverdictscore
#1ACCEPTED4
#2ACCEPTED6
#3ACCEPTED10
#4ACCEPTED13
#5ACCEPTED25
#60
Test results
testverdicttimegroup
#1ACCEPTED0.00 s1, 2, 3, 4, 5, 6details
#2ACCEPTED0.00 s1, 2, 3, 4, 5, 6details
#3ACCEPTED0.00 s1, 2, 3, 4, 5, 6details
#4ACCEPTED0.00 s1, 2, 3, 4, 5, 6details
#5ACCEPTED0.00 s1, 2, 3, 4, 5, 6details
#6ACCEPTED0.00 s2, 3, 4, 5, 6details
#7ACCEPTED0.00 s2, 3, 4, 5, 6details
#8ACCEPTED0.00 s2, 3, 4, 5, 6details
#9ACCEPTED0.00 s2, 3, 4, 5, 6details
#10ACCEPTED0.01 s3, 4, 5, 6details
#11ACCEPTED0.02 s3, 4, 5, 6details
#12ACCEPTED0.00 s3, 4, 5, 6details
#13ACCEPTED0.00 s3, 4, 5, 6details
#14ACCEPTED0.02 s4, 5, 6details
#15ACCEPTED0.14 s4, 5, 6details
#16ACCEPTED0.01 s4, 5, 6details
#17ACCEPTED0.01 s4, 5, 6details
#18ACCEPTED0.08 s5, 6details
#19ACCEPTED0.82 s5, 6details
#20ACCEPTED0.03 s5, 6details
#21ACCEPTED0.07 s5, 6details
#22ACCEPTED0.67 s6details
#23--6details
#24ACCEPTED0.17 s6details
#25--6details

Code

use std::{collections::HashSet, io};
fn is_valid(sums: &Vec<Vec<Vec<usize>>>, unique_letters: &HashSet<usize>,
y_start: usize, x_start: usize, y_end: usize, x_end: usize ) -> bool {
for l in unique_letters {
let mut count = sums[y_end][x_end][*l];
if y_start != 0 && x_start != 0 {
count += sums[y_start-1][x_start-1][*l];
}
if y_start != 0 {
count -= sums[y_start-1][x_end][*l]
}
if x_start != 0 {
count -= sums[y_end][x_start-1][*l];
}
if count == 0 {
return false;
}
}
return true;
}
fn main() {
let mut n_str = String::new();
io::stdin().read_line(&mut n_str).unwrap();
n_str = n_str.trim().to_string();
let n: usize = n_str.parse().unwrap();
let mut rows: Vec<Vec<usize>> = Vec::with_capacity(n);
let mut unique_letters = HashSet::new();
for _ in 0..n {
let mut row_str = String::new();
io::stdin().read_line(&mut row_str).unwrap();
row_str = row_str.trim().to_string();
rows.push(Vec::with_capacity(n));
for c in row_str.chars() {
let c_int = (c as usize)-65;
unique_letters.insert(c_int);
rows.last_mut().unwrap().push(c_int);
}
}
let mut sums: Vec<Vec<Vec<usize>>> = vec![vec![vec![0; 26]; n]; n];
for row in 0..n {
for col in 0..n {
sums[row][col][rows[row][col]] = 1;
for l in unique_letters.iter() {
if row != 0 {
sums[row][col][*l] += sums[row-1][col][*l];
}
if col != 0 {
sums[row][col][*l] += sums[row][col-1][*l];
}
if row != 0 && col != 0 {
sums[row][col][*l] -= sums[row-1][col-1][*l];
}
}
}
}
let mut count = 0;
'y: for y_start in 0..n {
for x_start in 0..n {
if !is_valid(&sums, &unique_letters, y_start, x_start, n-1, n-1) {
continue 'y;
}
let mut last_y = y_start;
let mut x_end = n-1;
loop {
let mut low = last_y;
let mut high = n-1;
while low < high {
let mid = (low + high) / 2;
if !is_valid(&sums, &unique_letters, y_start, x_start, mid, x_end) {
low = mid+1;
} else {
high = mid;
}
}
if !is_valid(&sums, &unique_letters, y_start, x_start, low, x_end) {
break;
}
last_y = low;
let mut low = x_start;
let mut high = x_end;
while low < high {
let mid = (low + high) / 2;
if !is_valid(&sums, &unique_letters, y_start, x_start, last_y, mid) {
low = mid+1;
} else {
high = mid;
}
}
count +=(n-last_y)*(x_end-low+1);
if low == x_start {
break;
}
x_end = low-1;
}
}
}
println!("{}", count);
}

Test details

Test 1

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
10
TNCTNPNTPC
NPPNTNTPTP
NTNTTCNTCT
NPCPNPPNTT
...

correct output
2035

user output
2035

Test 2

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
10
NFWQLWNWYS
DZOQJVXFPJ
CNHXPXMCQD
QRTBVNLTQC
...

correct output
9

user output
9

Test 3

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
10
XXXXXXXXXX
XXXXXXXXXX
XXXXXXXXXX
XXXXXXXXXX
...

correct output
3025

user output
3025

Test 4

Group: 1, 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
10
FFFFFFFFFF
FFFFFCFFFF
FFFFFFJFFF
FFFFFFFFFF
...

correct output
12

user output
12

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: ACCEPTED

input
20
BBCBUBOUOBBCUUBBCOUO
BOUCOOCUBCOOOCOBOCUO
UCCUUUOBCOCBCBUBUCOO
BUOBUCUCUOOBCOOUBUOO
...

correct output
38724

user output
38724

Test 7

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
20
CBGLSHGZHYZDWBNDBJUG
SMUXOJQYPXZDTMJUIWOJ
XIDSTNBGHKRKOVUVMINB
MTQGCFRUHQKALXRNCQGS
...

correct output
8334

user output
8334

Test 8

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
20
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
...

correct output
44100

user output
44100

Test 9

Group: 2, 3, 4, 5, 6

Verdict: ACCEPTED

input
20
AAAAAAAAXAAAAAAAAAAA
AAAWAAAAAAAAAAAAAOAA
AAAAAAAAAAAAAAAAAPAA
AAAAAAAAKAAAAAAAAAAZ
...

correct output
18

user output
18

Test 10

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
50
GRGREEEGREGXRXXEGXXREXGRRRGRRR...

correct output
1584665

user output
1584665

Test 11

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
50
AITIISJUHCCRZNKSDCNQKYSQRINFWJ...

correct output
1077746

user output
1077746

Test 12

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
50
OOOOOOOOOOOOOOOOOOOOOOOOOOOOOO...

correct output
1625625

user output
1625625

Test 13

Group: 3, 4, 5, 6

Verdict: ACCEPTED

input
50
FFFFFFFFFFFFFFFFFFFFFFFFFFFFFF...

correct output
1680

user output
1680

Test 14

Group: 4, 5, 6

Verdict: ACCEPTED

input
100
NNCMDCDDCCNNNDNCMMNCDCDCCDCDNM...

correct output
25325366

user output
25325366

Test 15

Group: 4, 5, 6

Verdict: ACCEPTED

input
100
LIMQQIHASECROEVILNVULGWZJPPKOG...

correct output
22342463

user output
22342463

Test 16

Group: 4, 5, 6

Verdict: ACCEPTED

input
100
TTTTTTTTTTTTTTTTTTTTTTTTTTTTTT...

correct output
25502500

user output
25502500

Test 17

Group: 4, 5, 6

Verdict: ACCEPTED

input
100
QXQQQQQQQQQQQQQQQQQQQQQQQQQQQQ...

correct output
25650

user output
25650

Test 18

Group: 5, 6

Verdict: ACCEPTED

input
200
NAANANMMKNKKAKMKMAKNKMNKMMNNAA...

correct output
403292767

user output
403292767

Test 19

Group: 5, 6

Verdict: ACCEPTED

input
200
OMYWATTLURKQPTKEFMGGYAOONXWVSC...

correct output
388111321

user output
388111321

Test 20

Group: 5, 6

Verdict: ACCEPTED

input
200
CCCCCCCCCCCCCCCCCCCCCCCCCCCCCC...

correct output
404010000

user output
404010000

Test 21

Group: 5, 6

Verdict: ACCEPTED

input
200
LLLLLLLLLLLLLLLLLHLLLLLLLLLLLL...

correct output
14159445

user output
14159445

Test 22

Group: 6

Verdict: ACCEPTED

input
500
VVHWVUHVHUWWWVUUUWVUUHUUWHWUVW...

correct output
15683003812

user output
15683003812

Test 23

Group: 6

Verdict:

input
500
OIMZGEQSBMBDSDXSWRFNKSGFEBBTJE...

correct output
15575906951

user output
(empty)

Test 24

Group: 6

Verdict: ACCEPTED

input
500
IIIIIIIIIIIIIIIIIIIIIIIIIIIIII...

correct output
15687562500

user output
15687562500

Test 25

Group: 6

Verdict:

input
500
WWWWWWWWWWWWWWWWWWWWWWWWWWWWWW...

correct output
3058970930

user output
(empty)