CSES - Datatähti 2016 alku - Results
Submission details
Task:Tontti
Sender:ollpu
Submission time:2015-10-05 21:46:06
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)