글 읽기 - input()과 sys.stdin.readline().split()의 차이점

pyd6119 3년 전 0

아래 코드에서

MAP = [list(map(int, sys.stdin.readline().split())) for _ in range(N)] 이런 형식으로 2차원 배열을 입력하고자 했으나 list index 범위가 벗어난다는

error가 발생했습니다.

if MAP[i][j] == 1:

IndexError: list index out of range

반면 sys.stdin.readline().split() 대신 input()을 사용했더니 제대로 된 결과가 나왔습니다.

MAP = [list(map(int, input())) for _ in range(N)]

결론은 input()을 사용했을땐 되고 sys.stdin.readline().split()는 안되는 이유가 무엇인지 궁금합니다.

import sys

dy = [0, 0, 1, -1]
dx = [1, -1, 0, 0]

N = int(sys.stdin.readline().rstrip())

# MAP = [list(map(int, sys.stdin.readline().split())) for _ in range(N)] -> list index error
MAP = [list(map(int, input())) for _ in range(N)]

def DFS(y, x):
    global count
    count += 1
    MAP[y][x] = 0 

    for i in range(4):
        ny = y + dy[i]
        nx = x + dx[i]

        if ny < 0 or ny > N-1 or nx < 0 or nx > N-1:
            continue
        if MAP[ny][nx] == 1:
            DFS(ny, nx)

    return count

result = []

for i in range(N):
    for j in range(N):
        if MAP[i][j] == 1:
            count = 0
            result.append(DFS(i, j)) 

print(len(result))

for i in sorted(result): # 정렬된 상태에서 결과를 출력해야 한다.
    print(i)

pyd6119 3년 전 0

MAP = [list(map(int, sys.stdin.readline().rstrip('\n'))) for _ in range(N)]

이렇게 사용하니까 문제가 해결되긴 했습니다.

WeissBlume 3년 전 0

https://www.acmicpc.net/proble... 이 문제 설명이 도움이 될 것 같아요

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

input()과 sys.stdin.readline().split()의 차이점

pyd6119 3년 전 0

pyd6119 3년 전 0

WeissBlume 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말