CSES - Datatähti 2016 alku - Results
Submission details
Task:Tontti
Sender:ollpu
Submission time:2015-10-11 11:06:36
Language:Python2
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#10.06 s1details
#2ACCEPTED0.05 s1details
#3ACCEPTED0.05 s1details
#40.06 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
    def calculate_diagonal(dx, dy):
        start = 0
        end = -1
        block = 0
        vaihtoehtoja = 0
        
        end_finished = False
        while True: # Repeat direction swap
            while True: # +end (adds to block)
                end += 1
                if not (dx+end < n and dy+end < m):
                    end_finished = True
                    end -= 1
                    break
                
                # Calculate diff to block
                for x2 in xrange(dx+start, dx+end+1):
                    if metsa[x2][dy+end]:
                        block += 1
                for spot in metsa[dx+end][dy+start:dy+end]:
                    if spot:
                        block += 1
                
                if block == x:
                    vaihtoehtoja += 1
                elif block > x:
                    break
            while True: # +start (substracts from block)
                start += 1
                if start > end:
                    start -= 1
                    break # Reached minimun size
                
                # Calculate diff to block
                for x2 in xrange(dx+start-1, dx+end+1):
                    if metsa[x2][dy+start-1]:
                        block -= 1
                for spot in metsa[dx+start-1][dy+start:dy+end+1]:
                    if spot:
                        block -= 1
                
                
                if block == x:
                    vaihtoehtoja += 1
                elif block < x:
                    break
                
            if end_finished:
                break
        return vaihtoehtoja
    
    # Calculate diagonals starting from the top (topleft corner included)
    for y1 in xrange(m):
        vaihtoehtoja += calculate_diagonal(0, y1)
    # Calculate diagonals starting from the left
    for x1 in xrange(1, n):
        vaihtoehtoja += calculate_diagonal(x1, 0)
    print(vaihtoehtoja)

# import profile
# profile.run('run()')
run()

Test details

Test 1

Group: 1

Verdict:

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

correct output
94

user output
65

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:

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

correct output
16

user output
15

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)