-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathBaekjoon_1025.py
62 lines (56 loc) · 1.67 KB
/
Baekjoon_1025.py
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
61
62
# 1025
import sys
input = sys.stdin.readline
n, m = map(int, input().split())
def getMap(n) :
inputs = []
for _ in range(n):
ip = []
[ip.append(int(num)) for num in input().rstrip()]
inputs.append(ip)
return inputs
def getBiggest(maps, rd, cd) :
results = [-1]
for i in range(n) :
for j in range(m) :
results.append(getResult(maps, i, j, rd, cd))
return max(results)
def getResult(maps, tx, ty, rd, cd) :
results = []
types = [(-1, -1), (-1, 1), (1, -1), (1, 1)]
if cd == 0 and rd == 0 :
return maps[tx][ty] if maps[tx][ty] ** 0.5 % 1 == 0 else -1
for t in types :
result = []
rs = 0
if rd == 0 :
i = tx
for j in range(ty, m if t[1] == 1 else -1, cd * t[1]) :
rs *= 10
rs += maps[i][j]
result.append(rs)
elif cd == 0 :
j = ty
for i in range(tx, n if t[0] == 1 else -1, rd * t[0]) :
rs *= 10
rs += maps[i][j]
result.append(rs)
else :
i, j = tx, ty
while 0 <= i < n and 0 <= j < m :
rs *= 10
rs += maps[i][j]
i += rd * t[0]; j += cd * t[1]
result.append(rs)
for rs in result :
results.append(rs if rs ** 0.5 % 1 == 0 else -1)
return max(results)
def solution(n, m) :
maps = getMap(n)
biggest = -1
for i in range(n + 1) :
for j in range(m + 1) :
result = getBiggest(maps, i, j)
biggest = result if biggest < result else biggest
return biggest
print(solution(n, m))