CSES - Datatähti 2016 alku - Results
Submission details
Task:Tontti
Sender:ollpu
Submission time:2015-10-05 21:46:06 +0300
Language:Python2
Status:READY
Result:14
Feedback
groupverdictscore
#1ACCEPTED14
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.06 s1details
#2ACCEPTED0.05 s1details
#3ACCEPTED0.07 s1details
#4ACCEPTED0.06 s1details
#5ACCEPTED0.07 s1details
#6--2details
#7--2details
#8--2details
#9--2details
#10--2details
#11--3details
#12--3details
#13--3details
#14--3details
#15--3details

Code

#!/usr/bin/env python
# -*- coding: utf-8 -*-
from sys import stdin
# import profile
def readline():
return stdin.readline().strip()
def get_vars():
global n, m, x
row1 = readline().split(' ')
n = int(row1[0]) # Korkeus
m = int(row1[1]) # Leveys
x = int(row1[2]) # Tavoite
get_vars()
def isqrt(n):
x = n
y = (x + 1) // 2
while y < x:
x = y
y = (x + n // x) // 2
return x
def is_star(char):
return char == '*'
def run():
metsa = []
for i in xrange(n):
metsa.append(map(is_star, readline()))
vaihtoehtoja = 0
sqrt_x = isqrt(x) - 1
first_w = max(1, sqrt_x)
min_dim = min(n, m)
block_w = 0
for row in metsa[:first_w]:
block_w += row[:first_w].count(True)
block = block_w
# Generate first bars
bars_x = []
for y2 in xrange(m):
bar = 0
for x2 in xrange(first_w):
if metsa[x2][y2]:
bar += 1
bars_x.append(bar)
bars_x_w = list(bars_x)
bars_x_w_ = list(bars_x)
bars_y = []
for x2 in xrange(n):
bar = 0
for y2 in xrange(first_w):
if metsa[x2][y2]:
bar += 1
bars_y.append(bar)
for w in xrange(first_w, min_dim):
wi = w - 1
block_x = block
for x1 in xrange(n-wi):
for y1 in xrange(m-wi):
if block == x:
vaihtoehtoja += 1
if not y1 == m-wi-1:
block += bars_x[y1+w]
block -= bars_x[y1]
# Recalculate block(s) & move x-bars
if not x1 == n-wi-1:
block_x += bars_y[x1+w]
block_x -= bars_y[x1]
block = block_x
# Recalculate x-bars
for y2 in xrange(m):
diff = 0
if metsa[x1][y2]:
diff -= 1
if metsa[x1+w][y2]:
diff += 1
if x1 is 0:
bars_x_w[y2] += 1
bars_x[y2] += diff
# Expand starting square
if not w == min_dim-1:
block_w += bars_x_w_[w]
block_w += bars_y[w]
# Corner
if metsa[w][w]:
block_w += 1
block = block_w
bars_x_w_ = list(bars_x_w)
# Calc new starting bars
# for y2 in xrange(m):
# if metsa[w][y2]:
# bars_x_w[y2] += 1
bars_x = list(bars_x_w)
for x2 in xrange(n):
if metsa[x2][w]:
bars_y[x2] += 1
print(vaihtoehtoja)
# profile.run('run()')
run()

Test details

Test 1

Group: 1

Verdict: ACCEPTED

input
10 10 1
......*...
.......*..
*..*....*.
*....*....
...

correct output
94

user output
94

Test 2

Group: 1

Verdict: ACCEPTED

input
10 10 5
**********
**********
**********
**********
...

correct output
0

user output
0

Test 3

Group: 1

Verdict: ACCEPTED

input
10 10 10
**...*...*
*..*.**.*.
...**.*..*
*...**.*..
...

correct output
4

user output
4

Test 4

Group: 1

Verdict: ACCEPTED

input
10 10 5
****......
*.*.**..**
....*.*..*
...*.***..
...

correct output
16

user output
16

Test 5

Group: 1

Verdict: ACCEPTED

input
10 10 2
**.***..*.
...*.*....
.***.*...*
***.***..*
...

correct output
30

user output
30

Test 6

Group: 2

Verdict:

input
500 500 1
.................................

correct output
9552040

user output
(empty)

Test 7

Group: 2

Verdict:

input
500 500 5
.................................

correct output
1536063

user output
(empty)

Test 8

Group: 2

Verdict:

input
500 500 25000
**...*...**..*.*..*.**.*..*.*....

correct output
288

user output
(empty)

Test 9

Group: 2

Verdict:

input
500 500 12500
**.**.*..*...*.**...*.***........

correct output
786

user output
(empty)

Test 10

Group: 2

Verdict:

input
500 500 5000
.*.*.**..*.*.**.**..*..**...*....

correct output
1763

user output
(empty)

Test 11

Group: 3

Verdict:

input
2000 2000 1
.................................

correct output
489611392

user output
(empty)

Test 12

Group: 3

Verdict:

input
2000 2000 5
.................................

correct output
120725884

user output
(empty)

Test 13

Group: 3

Verdict:

input
2000 2000 400000
..*..**.**.**.*.***...**.*..**...

correct output
1849

user output
(empty)

Test 14

Group: 3

Verdict:

input
2000 2000 200000
***.*....*.*..*....**..*..*.*....

correct output
2665

user output
(empty)

Test 15

Group: 3

Verdict:

input
2000 2000 80000
**.**...*.***.**....**.*....*....

correct output
5587

user output
(empty)