CSES - Datatähti 2025 alku - Results
Submission details
Task:Niitty
Sender:raikasdev
Submission time:2024-11-06 16:40:53 +0200
Language:Python3 (PyPy3)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
#50
#60
Test results
testverdicttimegroup
#10.05 s1, 2, 3, 4, 5, 6details
#20.05 s1, 2, 3, 4, 5, 6details
#30.05 s1, 2, 3, 4, 5, 6details
#4ACCEPTED0.05 s1, 2, 3, 4, 5, 6details
#5ACCEPTED0.04 s1, 2, 3, 4, 5, 6details
#60.07 s2, 3, 4, 5, 6details
#70.08 s2, 3, 4, 5, 6details
#80.06 s2, 3, 4, 5, 6details
#90.07 s2, 3, 4, 5, 6details
#100.09 s3, 4, 5, 6details
#110.09 s3, 4, 5, 6details
#120.07 s3, 4, 5, 6details
#130.10 s3, 4, 5, 6details
#140.21 s4, 5, 6details
#150.21 s4, 5, 6details
#160.14 s4, 5, 6details
#170.21 s4, 5, 6details
#18--5, 6details
#19--5, 6details
#20--5, 6details
#21--5, 6details
#22--6details
#23--6details
#24--6details
#25--6details

Code

# rows = [
#   ["A","A","A","K"],
#   ["A","B","A","A"],
#   ["A","B","A","A"],
#   ["A","A","A","A"]
# ]
rows = []
for _ in range(int(input())):
  rows.append(list(input()))
#print(rows)
plants = set(sum(rows, []))
 
possible = 0 # Ulkoreuna on aina yksi aitaus
 
# Niitty koostuu n x n ruudusta, eli se on neliö
width = len(rows)
height = len(rows)
x = 0
y = 0

def validArea(x = 0, y = 0, width = 1, height = 1):
  temp_rows = rows[y:(height+y)]
  for i, row in enumerate(temp_rows):
    temp_rows[i] = row[x:(width+x)]
  return set(sum(temp_rows, [])) == plants

# Etsitään kaikkien esiintyvien kasvien ensimmäinen sijanti,
# ja rakennetaan siitä suorakulmio
def superSearch():
    left = list(plants)
    values = []
    for y, row in enumerate(rows):
        for x, plant in enumerate(row):
            if plant in left:
                left.remove(plant)
                values.append([x,y])
                if len(left) == 0:
                    break
    box_x = min([item[0] for item in values])
    box_width = max([item[0] for item in values]) + 1
    box_y = min([item[1] for item in values])
    box_height = max([item[1] for item in values]) + 1
    return [box_x, box_y, box_width, box_height]

possibles = 0

box = superSearch()
box_x = box[0]
box_y = box[1]
box_width = box[2]
box_height = box[3]

# Vasemmalta
for _ in range(width):
    if validArea(box_x + 1, box_y, box_width - 1, box_height):
        box_x += 1
        box_width -= 1
    else:
        break
# Ylhäältä
for _ in range(height - box_height):
    if validArea(box_x, box_y + 1, box_width, box_height - 1):
        box_y += 1
        box_height -= 1
    else:
        break
# Oikealta
for _ in range(1, box_width):
    if validArea(box_x, box_y, box_width - 1, box_height):
        box_width -= 1
    else:
        break
# Alhaalta
for _ in range(1, box_height):
    if validArea(box_x, box_y, box_width, box_height - 1):
        box_height -= 1
    else:
        break

# Kaikki tätä laatikkoa suuremmat arvot ovat mahdollisia (x pienemmät ja width->(leveys-x))

free_space_left = box_x
free_space_right = width-box_width-x
free_space_up = box_y
free_space_down = height-box_height-y

def areaIncluded(x1, y1, w1, h1, x2, y2, w2, h2):
    if x2 > x1 or w2 < (w1 + x1 - x2):
        return False
    if y2 > y1 or h2 < (h1 + y1 - y2):
        return False
    return True
    
# Joo tässä varmaan saisi jotenkin hienosti tarkastettua, mutta tehdääs laiskasti vaan taas kaikki :D
for x in range(len(rows)):
  for y in range(len(rows)):
    for width in range(1, len(rows) - x + 1):
      for height in range(1, len(rows) - y + 1):
        if areaIncluded(box_x, box_y, box_width, box_height, x, y, width, height):
            possibles += 1
print(possibles)

Test details

Test 1

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

Verdict:

input
10
TNCTNPNTPC
NPPNTNTPTP
NTNTTCNTCT
NPCPNPPNTT
...

correct output
2035

user output
150

Test 2

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

Verdict:

input
10
NFWQLWNWYS
DZOQJVXFPJ
CNHXPXMCQD
QRTBVNLTQC
...

correct output
9

user output
4

Test 3

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

Verdict:

input
10
XXXXXXXXXX
XXXXXXXXXX
XXXXXXXXXX
XXXXXXXXXX
...

correct output
3025

user output
100

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:

input
20
BBCBUBOUOBBCUUBBCOUO
BOUCOOCUBCOOOCOBOCUO
UCCUUUOBCOCBCBUBUCOO
BUOBUCUCUOOBCOOUBUOO
...

correct output
38724

user output
840

Test 7

Group: 2, 3, 4, 5, 6

Verdict:

input
20
CBGLSHGZHYZDWBNDBJUG
SMUXOJQYPXZDTMJUIWOJ
XIDSTNBGHKRKOVUVMINB
MTQGCFRUHQKALXRNCQGS
...

correct output
8334

user output
1176

Test 8

Group: 2, 3, 4, 5, 6

Verdict:

input
20
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
KKKKKKKKKKKKKKKKKKKK
...

correct output
44100

user output
400

Test 9

Group: 2, 3, 4, 5, 6

Verdict:

input
20
AAAAAAAAXAAAAAAAAAAA
AAAWAAAAAAAAAAAAAOAA
AAAAAAAAAAAAAAAAAPAA
AAAAAAAAKAAAAAAAAAAZ
...

correct output
18

user output
12

Test 10

Group: 3, 4, 5, 6

Verdict:

input
50
GRGREEEGREGXRXXEGXXREXGRRRGRRR...

correct output
1584665

user output
17550

Test 11

Group: 3, 4, 5, 6

Verdict:

input
50
AITIISJUHCCRZNKSDCNQKYSQRINFWJ...

correct output
1077746

user output
4608

Test 12

Group: 3, 4, 5, 6

Verdict:

input
50
OOOOOOOOOOOOOOOOOOOOOOOOOOOOOO...

correct output
1625625

user output
2500

Test 13

Group: 3, 4, 5, 6

Verdict:

input
50
FFFFFFFFFFFFFFFFFFFFFFFFFFFFFF...

correct output
1680

user output
1296

Test 14

Group: 4, 5, 6

Verdict:

input
100
NNCMDCDDCCNNNDNCMMNCDCDCCDCDNM...

correct output
25325366

user output
19200

Test 15

Group: 4, 5, 6

Verdict:

input
100
LIMQQIHASECROEVILNVULGWZJPPKOG...

correct output
22342463

user output
15600

Test 16

Group: 4, 5, 6

Verdict:

input
100
TTTTTTTTTTTTTTTTTTTTTTTTTTTTTT...

correct output
25502500

user output
10000

Test 17

Group: 4, 5, 6

Verdict:

input
100
QXQQQQQQQQQQQQQQQQQQQQQQQQQQQQ...

correct output
25650

user output
5152

Test 18

Group: 5, 6

Verdict:

input
200
NAANANMMKNKKAKMKMAKNKMNKMMNNAA...

correct output
403292767

user output
(empty)

Test 19

Group: 5, 6

Verdict:

input
200
OMYWATTLURKQPTKEFMGGYAOONXWVSC...

correct output
388111321

user output
(empty)

Test 20

Group: 5, 6

Verdict:

input
200
CCCCCCCCCCCCCCCCCCCCCCCCCCCCCC...

correct output
404010000

user output
(empty)

Test 21

Group: 5, 6

Verdict:

input
200
LLLLLLLLLLLLLLLLLHLLLLLLLLLLLL...

correct output
14159445

user output
(empty)

Test 22

Group: 6

Verdict:

input
500
VVHWVUHVHUWWWVUUUWVUUHUUWHWUVW...

correct output
15683003812

user output
(empty)

Test 23

Group: 6

Verdict:

input
500
OIMZGEQSBMBDSDXSWRFNKSGFEBBTJE...

correct output
15575906951

user output
(empty)

Test 24

Group: 6

Verdict:

input
500
IIIIIIIIIIIIIIIIIIIIIIIIIIIIII...

correct output
15687562500

user output
(empty)

Test 25

Group: 6

Verdict:

input
500
WWWWWWWWWWWWWWWWWWWWWWWWWWWWWW...

correct output
3058970930

user output
(empty)