CSES - Datatähti 2016 alku - Results
Submission details
Task:Lennot
Sender:Ilari
Submission time:2015-09-29 17:01:37 +0300
Language:Python3
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#10.08 s1details
#20.08 s1details
#30.09 s1details
#40.10 s1details
#50.07 s1details
#60.08 s2details
#70.08 s2details
#80.08 s2details
#90.08 s2details
#100.09 s2details
#110.08 s3details
#120.09 s3details
#130.09 s3details
#140.08 s3details
#150.10 s3details
#160.09 s3details
#170.07 s3details

Code

#!/usr/bin/python3
import math
n, m, x = map(int, input().split())
#print(n,m,x)
a=[]
for i in range(0,n):
	a.append(list(map(int, input().replace('.','0').replace('*','1'))))
#a=a.replace('.','0').replace('*','1')


def sumTrees(x,y):
	trees=0
	b=[]
	for i in range(0,y):
		i1=i-1
		b.append([])
		ai=list(a[i])
		for j in range(0,x):
			trees+=ai[j]
			if i>0:
				b[i].append(trees+b[i1][j])
			else:
				b[i].append(trees)
		trees=0
	return b

b=sumTrees(m,n)

#for i in b:
#	print(i)
#for i in range(0,n):
#	ai=a[i]
#	b.append([])
#
#	for j in range(0,m):
#		b[i]=ai[j]
#print(a)


#for i in range(1, limit+1):
#	pass
#@profile

#@profile
def countTrees(x,x1,y,y1,size,x0,y0):
	#trees=0
	if not (x0 or y0):
		trees=b[y+size][x+size]-b[y1][x+size]-b[y+size][x1]+b[y1][x1]
	elif x==0 and y==0:
		trees=b[size][size]
	elif x==0:
		trees=b[y+size][size]-b[y1][size]
	else:
		trees=b[size][x+size]-b[size][x1]

	return trees

def possibleSquares(ny,mx,sqrt):
	if ny<mx:
		limit=ny
	else:
		limit=mx
	return range(sqrt-1,limit)

#print(countTrees(2,0,3))
#answer=0
sqrt=int(math.sqrt(x))
#@profile
def calculate():
	answer=0
	for i in range(0, m):
		mx=m-i
		i1=i-1
		if i==0:
			i0=1
		else:
			i0=0
		for j in range(0,n):
			if j==0:
				j0=1
			else:
				i0=0
			j1=j-1
			#trees = 0
			for k in possibleSquares(n-j,mx, sqrt):
				trees = countTrees(i,i1,j,j1,k,i0,j0)
				if trees > x:
					break
				elif trees == x:
					answer+=1
	return answer
print(calculate())

Test details

Test 1

Group: 1

Verdict:

input
10 20
2 1 3
7 6 4
1 6 7
1 6 1
...

correct output
8

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 2

Group: 1

Verdict:

input
10 20
4 3 10
1 10 9
3 4 10
2 6 7
...

correct output
9

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 3

Group: 1

Verdict:

input
10 20
5 7 4
6 1 1
7 3 8
8 4 2
...

correct output
8

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 4

Group: 1

Verdict:

input
10 20
1 6 2
5 3 3
7 3 6
5 6 2
...

correct output
13

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 5

Group: 1

Verdict:

input
10 20
10 8 5
2 4 7
9 4 7
9 4 1
...

correct output
4

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 6

Group: 2

Verdict:

input
1000 2000
91 828 365044406
17 984 445675537
251 852 100987451
907 487 58830088
...

correct output
11893353673

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 7

Group: 2

Verdict:

input
1000 2000
722 939 530579090
404 606 268877348
133 750 760086153
506 46 582310443
...

correct output
30248963445

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 8

Group: 2

Verdict:

input
1000 2000
340 237 43690066
217 141 453160975
744 202 639037814
605 926 404985542
...

correct output
3126797692

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 9

Group: 2

Verdict:

input
1000 2000
88 312 190442306
480 402 411574469
29 901 397491243
636 459 323246996
...

correct output
18416073173

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 10

Group: 2

Verdict:

input
1000 2000
333 228 718389176
796 286 323493090
743 43 751876815
128 554 175625940
...

correct output
6399349335

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 11

Group: 3

Verdict:

input
100000 200000
28264 92686 186865663
92570 33956 925976418
87377 71249 644757113
16701 81203 922125505
...

correct output
518249578675

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 12

Group: 3

Verdict:

input
100000 200000
95740 71482 846654568
44131 16806 670712211
3967 49254 424174139
39369 53007 830346557
...

correct output
920862321580

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 13

Group: 3

Verdict:

input
100000 200000
79947 25489 71554257
59184 25577 328436360
82945 73554 4942918
22380 92385 874250042
...

correct output
399407698440

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 14

Group: 3

Verdict:

input
100000 200000
31139 12960 580545990
27744 95556 747296719
46969 42578 840321561
5638 28960 513805324
...

correct output
165235287505

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 15

Group: 3

Verdict:

input
99993 199980
1 3 1
3 2 1
1 4 1
4 2 1
...

correct output
2

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 16

Group: 3

Verdict:

input
100000 149994
93867 98509 1755709
85029 99843 1347591
10305 35305 6447
75638 80585 1829972
...

correct output
1124960

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack

Test 17

Group: 3

Verdict:

input
100000 200000
70413 71496 49
15963 40963 18635
81291 89420 1850028
8848 33848 17316
...

correct output
110298

user output
(empty)

Error:
Traceback (most recent call last):
  File "input/code.py", line 3, in <module>
    n, m, x = map(int, input().split())
ValueError: need more than 2 values to unpack