글 읽기 - 이 코드가 무엇이 잘못되었을까용 ㅠ

unioc1970 1년 전 0

뭐가 잘못되었는지 모르겠습니다 ㅠㅠ 도와주시면 감사하겠습니다.

from collections import deque
import sys
sys.stdin=open("input.txt","r")
sys.setrecursionlimit(int(1e5))

x=[-1,0,1,0]
y=[0,1,0,-1]
n=int(input())
board=[]
for i in range(n):
    board.append(str(input()))
Q=deque()
ch=[[0]*n for _ in range(n)]
cnt=0
li=[]
for i in range(n):
    for j in range(n):
        if board[i][j]=='1' and ch[i][j]==0:
            Q.append((i,j))
            cnt+=1
            ch[i][j]=1
            k=1
            while Q:
                tmp=Q.popleft()
                for i in range(4):
                    xx=tmp[0]+x[i]
                    yy=tmp[1]+y[i]
                    if 0<=xx<n and 0<=yy<n and ch[xx][yy]==0 and board[xx][yy]=='1':
                        ch[xx][yy]=1
                        Q.append((xx,yy))
                        k+=1
            li.append(k)
print(len(li))
for i in li:
    print(i)

zenith82114 1년 전 1

오름차순으로 정렬하라고 문제에 써 있습니다.

unioc1970 1년 전 0

zenith82114님 감사합니다!

rhsek123 1년 전 1

정리해서 왔습니다..

from sys import stdin 
input = stdin.readline

#...
#...

for j in range(n):
    board.append(list(input().rstrip('\n')))

Q=deque()
cnt=0
li=[]

for l in range(n):
    for m in range(n):
        if board[l][m]=='1':
            Q.append((l,m))
            board[l][m]='0'
            cnt+=1
            k=1
            while Q:
                tmp=Q.popleft()
                for q in range(4):
                    yy=tmp[0]+y[q]
                    xx=tmp[1]+x[q]
                    if 0<=xx<n and 0<=yy<n and board[yy][xx]=='1':
                            Q.append((yy,xx))
                            board[yy][xx]='0'
                            k+=1
            li.append(k)
print(cnt)
li.sort()

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

문제

문제

출처

ICPC

이 코드가 무엇이 잘못되었을까용 ㅠ

unioc1970 1년 전 0

zenith82114 1년 전 1

unioc1970 1년 전 0

zenith82114님 감사합니다!

rhsek123 1년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말