CSES - DatatÃ¤hti 2016 alku - Results
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#10.08 s1details
#20.08 s1details
#30.08 s1details
#40.09 s1details
#50.08 s1details
#60.08 s2details
#70.09 s2details
#80.08 s2details
#90.09 s2details
#100.08 s2details
#110.09 s3details
#120.10 s3details
#130.09 s3details
#140.10 s3details
#150.08 s3details

### Code

```import math
from array import array

# Parse input
line = input().split()
height = int(line[0])
width = int(line[1])
desired_trees = int(line[2])

field = []
for y in range(height):
line = input()
field.extend([x == '*' for x in line])

# debug
if desired_trees > math.sqrt(height*width):
a = 1/0

# Create index
index = []
for i in range(width*height):
this = 0
y, x = divmod(i, width)
if field[i]:
this = 1
if y > 0:
this += index[i-width]
if x > 0:
this += index[i-1]
if y > 0 and x > 0:
this -= index[i-width-1]
index.append(this)
#index = array('I', index) # use array instead of list for sake of speed

# Calculate
def trees_in(x1, y1, size):
x2 = x1 + size - 1
y2 = y1 + size - 1
if x1 == 0 and y1 == 0:
return index[y2*width + x2]
elif x1 == 0:
return index[y2*width + x2] - index[(y1-1)*width + x2]
elif y1 == 0:
return index[y2*width + x2] - index[y2*width + (x1-1)]
else:
return (index[y2*width + x2] - index[(y1-1)*width + x2] -
index[y2*width + (x1-1)] + index[(y1-1)*width + (x1-1)])

options = 0
min_min_size = math.ceil(math.sqrt(desired_trees))
for y in range(height):
min_size = min_min_size
for x in range(width):
max_size = min(height-y, width-x)
a = min_size
for size in range(a, max_size+1):
trees = trees_in(x, y, size)
if trees == desired_trees:
options += 1
elif trees > desired_trees:
# bigger area can't have less trees
break
else:
min_size = size - 1

print(options)
```

### Test details

#### Test 1

Group: 1

Verdict:

input
BBBAABBBAAAABBAAAABAABAABBBBBB...

correct output
2554

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'BBBAABBBAAAABBAAAABAABAABBBBBBAABBBBBAAABAAABABAABBABBAAABABABABBABBBBABABAABBAAABBBBABBBBAABBAABAAA'```

#### Test 2

Group: 1

Verdict:

input
GDFVYWQCZAFGICSXOSWBZMGPDBSSVL...

correct output
299

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'GDFVYWQCZAFGICSXOSWBZMGPDBSSVLFMEYWRESVQFYWMOLSCEADYZHZIIYEJVIPCLSFMYNKXTOWYSRDPDVNLQNLFPSXVHYKKWHZW'```

#### Test 3

Group: 1

Verdict:

input
AAAAAAAAAAAAAAAAAAAAAAAAAZAAAA...

correct output
4314

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'AAAAAAAAAAAAAAAAAAAAAAAAAZAAAAZAZAZAAAAAAAAAAAAAAAAAAAAAZZAAAAAAAAAAAZAAAAAZAAAAAAAAAAAAAAAAAAAAAAAA'```

#### Test 4

Group: 1

Verdict:

input
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA...

correct output
4231

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAZZAAAAAAZAAAAZAAAAAZAAAAAAAAZAAAAAAAAAAAZAAAZAAAAAAAAAAAAAZAAAA'```

#### Test 5

Group: 1

Verdict:

input
QQQQQQQQQQQQQQQQQQQQQQQQQQQQQQ...

correct output
5050

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'QQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQ'```

#### Test 6

Group: 2

Verdict:

input
BBABABBBABBAABBABBABAABAAABABA...

correct output
6253029

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'BBABABBBABBAABBABBABAABAAABABAABBBAABAABBBBBBBAABBABBBABBABBAAABBAABABABBABBBBBABBBBBBBABABABBBAABBABABAAABBBAABABBABBBBABBBABBBAABBBABBBBBBBAAABAAAABAABBBBAAABABBBABAABBBBABABAABAAABAABBABABAAAABBAA```

#### Test 7

Group: 2

Verdict:

input
RBKJMLDVQMKHYKCNDIVVKOMFUXTFMG...

correct output
485173

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'RBKJMLDVQMKHYKCNDIVVKOMFUXTFMGICWVUWKXCYRWJQMRSLJNGZTCGMDCQVTPWAYSXXVTCYJEYQCUUPTWKJBHZXVVGBEJOOYDSHTLKXHWCMKCHREGXSGDYECFJMMWEGCWSHWWWDIXJEGJSRLNMTQEJDAVMAFQPOZSLNWVWUDQWTKNMGIUFTDVDMLZIHVEMQHVCWJSE```

#### Test 8

Group: 2

Verdict:

input
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA...

correct output
12427725

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAZAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA```

#### Test 9

Group: 2

Verdict:

input
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA...

correct output
12467549

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA```

#### Test 10

Group: 2

Verdict:

input
QQQQQQQQQQQQQQQQQQQQQQQQQQQQQQ...

correct output
12502500

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'QQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQ```

#### Test 11

Group: 3

Verdict:

input
BAAAAABABBABAABAABABABBBABBAAB...

correct output
2500051369

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'BAAAAABABBABAABAABABABBBABBAABBBABBBABBBBBABBBBAAAAAABABAAAAAAABAABAABABBBAAABABAAAABABABAABBAAAABABAAABABBBBABBBBABABAAAAAABAAABBBBBBABBAAAABABABAABBABABAAABBBAABBAABABBABABAABABABBBBBABBBABAABAABAB```

#### Test 12

Group: 3

Verdict:

input
ABBURXDRVXAYBPXXOQZNYHLWGUEEWR...

correct output
192407124

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'ABBURXDRVXAYBPXXOQZNYHLWGUEEWRNOMCJGFEMRRSCURKRFGMOLHWBPJQSREDUVUHRESNQNYKFMBSNIUTFFZWALMYHECUOTYXVDYMLZYZCYDMYIKGCYYXXTNGJDCZPYFGHDBCPZCFWSJOCGXMJAFNIKHPBMNUZCDHYNEGXSPHYGIUXWIKQFKLYCXKNHDYCGUECOFTP```

#### Test 13

Group: 3

Verdict:

input
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA...

correct output
4998050400

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA```

#### Test 14

Group: 3

Verdict:

input
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA...

correct output
4998850144

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA```

#### Test 15

Group: 3

Verdict:

input
QQQQQQQQQQQQQQQQQQQQQQQQQQQQQQ...

correct output
5000050000

user output
(empty)

Error:
```Traceback (most recent call last):
File "input/code.py", line 6, in <module>
height = int(line[0])
ValueError: invalid literal for int() with base 10: 'QQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQQ```