여름의 서재

[SWEA] 1210_Ladder1 본문

알고리즘/SWEA

[SWEA] 1210_Ladder1

엉아_ 2021. 8. 13. 00:37
728x90
import sys

sys.stdin = open('input.txt')

for _ in range(10):
    T = int(input())
    matrix = [list(map(int, input().split())) for i in range(100)]
    line = list(filter(lambda x: matrix[0][x] == 1, range(100))) # 세로축 찾기

    target = matrix[-1].index(2) # 시작 세로축의 위치 찾기
    target_index = line.index(target)

    i = -2
    while i > -100: # 아래에서부터 올라감
        if target > 1 and matrix[i][target-1] == 1: # 왼쪽으로 먼저 가지가 뻗어있으면 왼쪽 세로축으로 이동
            target_index -= 1
            target = line[target_index]
        elif target < 99 and matrix[i][target+1] == 1: # 오른쪽으로 먼저 가지가 뻗어있으면 오른쪽 세로축으로 이동
            target_index += 1
            target = line[target_index]
        i -= 1

    print('#{0} {1}'.format(T, target))

'알고리즘 > SWEA' 카테고리의 다른 글

[SWEA] 4864_문자열 비교  (0) 2021.08.17
[SWEA] 1221_GNS  (0) 2021.08.17
[SWEA] 2001_파리 퇴치  (0) 2021.08.13
[SWEA] 4836_색칠하기  (0) 2021.08.13
[SWEA] 4837_부분집합의 합  (0) 2021.08.13
Comments