-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathswea_2805.py
More file actions
60 lines (51 loc) · 1.46 KB
/
Copy pathswea_2805.py
File metadata and controls
60 lines (51 loc) · 1.46 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
test = [[1, 4, 0, 5, 4], [4, 4, 2, 5, 0], [0, 2, 0, 3, 2], [5, 1, 2, 0, 4], [5, 2, 2, 1, 2]]
n = len(test)
# def counting_list(matrix):
# count_list = []
# for i in range(1, n + 1, 2):
# count_list.append(i)
# for i in range(n-2, 0, -2):
# count_list.append(i)
#
# return count_list
# my_list = counting_list(test)
# empty_matrix = [[0] * n for _ in range(n)]
# for j in my_list:
# for i in my_list:
# empty_matrix[n - j][n - i] = 1
#
# for row in empty_matrix:
# print(row)
# def farming(matrix):
# sum = 0
# center = (n - 1) / 2
# count_list = counting_list(matrix)
# for i in range(n):
# sum += matrix[i][center]
def make_check_matrix(n):
check_matrix = [[0] * n for _ in range(n)]
center = n // 2
for i in range(n):
distance = abs(i - center)
start = distance
end = n - distance
for j in range(start, end):
check_matrix[i][j] = 1
return check_matrix
def farm(matrix):
n = len(matrix)
check_matrix = make_check_matrix(n)
result = [[0] * n for _ in range(n)]
sum_value = 0
for i in range(n):
for j in range(n):
result[i][j] = check_matrix[i][j] * matrix[i][j]
for i in range(n):
sum_value += sum(result[i])
return sum_value
t = int(input())
for tc in range(1, t+1):
n = int(input())
target = [list(map(int, input())) for _ in range(n)]
result = farm(target)
print(f'#{tc} {result}')