파이썬 초보입니다. 5×5로 입력을 받아서 가장 긴 막대기의 길이를 출력하는 코드
조회수 211회
5×5로 입력을 받아서 0은 빈칸 1은 막대기라고 했을 때
가장 긴 막대기의 길이를 출력하는 코드를 짜고 싶은데 자꾸 답이 안나와요.
입력 예시
0 0 1 1 0
1 1 0 1 0
0 1 1 1 0
0 0 0 1 0
0 0 0 0 0
출력 예시
4
이렇게 나오게 하고 싶어요!
1 답변
-
참고하세요.
- numpy 사용
import numpy as np input_str = """ 0 0 1 1 0 1 1 0 1 0 0 1 1 1 0 0 0 0 1 0 0 0 0 0 0 """ matrix = np.array(list(map(str.split, input_str.strip().splitlines()))).astype(int) matrix_transpose = np.transpose(matrix) max_count = max(set(map(np.count_nonzero, matrix)).union(set(map(np.count_nonzero, matrix_transpose)))) 4
댓글 입력