CSES - Datatähti 2016 alku - Results
Submission details
Task:Osajono
Sender:felixbade
Submission time:2015-10-11 11:29:05 +0300
Language:Python3
Status:READY
Result:0
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