CSES - Datatähti 2016 alku - Results
Submission details
Task:Tontti
Sender:ollpu
Submission time:2015-10-02 21:05:35
Language:Python2
Status:READY
Result:14
Feedback
groupverdictscore
#1ACCEPTED14
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.06 s1details
#2ACCEPTED0.06 s1details
#3ACCEPTED0.04 s1details
#4ACCEPTED0.05 s1details
#5ACCEPTED0.06 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


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
    for w in xrange(first_w, min_dim):
        wi = w - 1
        block_x = block
        
        # Generate bars
        bars = []
        for y2 in xrange(m):
            bar = 0
            for x2 in xrange(w):
                if metsa[x2][y2]:
                    bar += 1
            bars.append(bar)
        
        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[y1+w]
                    block -= bars[y1]
                    
            if not x1 == n-wi-1:
                diff = 0
                for y2 in xrange(w):
                    if metsa[x1+w][y2]:
                        diff += 1
                        
                    if metsa[x1][y2]:
                        diff -= 1
                block_x += diff
                block = block_x
                
                # Recalculate bars
                for y2 in xrange(m):
                    diff = 0
                    if metsa[x1][y2]:
                        diff -= 1
                    if metsa[x1+w][y2]:
                        diff += 1
                    bars[y2] += diff
        if not w == min_dim-1:
            for w2 in xrange(w+1):
                # Calc x-bar
                if metsa[w2][w]:
                    block_w += 1
                # Calc y-bar
                if w2 < w:
                    if metsa[w][w2]:
                        block_w += 1
            block = block_w
    print(vaihtoehtoja)
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)