파이썬 알고리즘/BOJ
[Algorithm] 백준 #20125 쿠키의 신체 측정
walbe0528
2024. 7. 7. 10:59
728x90
반응형
https://www.acmicpc.net/problem/20125
import sys
input = sys.stdin.readline
N = int(input())
graph = [list(input().rstrip()) for _ in range(N)]
# print(graph)
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
# 심장위치 먼저 찾기 -> 상하좌우 확인해서 모두 *인 지점
for i in range(1, N-1):
for j in range(1, N-1):
flag = []
for m in range(4):
nx = i + dx[m]
ny = j + dy[m]
if graph[nx][ny] == '*':
flag.append([nx, ny])
continue
else:
break
if len(flag) == 4: # 심장인 지점
flag.sort()
a = flag[1][0]
b = flag[3][1]
break
# (1, 1)부터 시작이므로 좌표에 +1
print(a+1, b+1)
# 팔 길이 구하기
pos = []
for i in range(N):
if graph[a][i] == '*':
pos.append(i)
pos.sort()
# print(pos)
print(b-pos[0], pos[-1]-b, end=' ')
# 허리 구하기
pos = []
for i in range(N):
if graph[i][b] == '*':
pos.append(i)
pos.sort()
print(pos[-1]-pos[1], end=' ')
end_x, end_y = pos[-1], b # 허리 마지막 지점
# 다리 구하기
l_pos, r_pos = [], []
for i in range(N):
if 0 <= end_x+1+i < N and 0 <= end_y-1 < N:
if graph[end_x+1+i][end_y-1] == '*':
l_pos.append(0)
if 0 <= end_x+1+i < N and 0 <= end_y+1 < N:
if graph[end_x+1+i][end_y+1] == '*':
r_pos.append(0)
print(len(l_pos), len(r_pos))
- 기준이 되는 심장 먼저 찾기
- 상하좌우 확인해서 모두 *이 나오는 지점
- 문제에서 좌표가 (1,1)부터 시작하기에 좌표에 +1을 모두 해준다
728x90