inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

파이썬 알고리즘 문제풀이 입문(코딩테스트 대비)

12. 단지 번호 붙이기(DFS)

BFS 코드 질문

343

hyoseok BANG

작성한 질문수 6

0

안녕하세요 강사님

BFS를 이용해서 답안을 작성해보았는데요, 어딘가가 잘못된 것 같은데 잘 모르겠네요ㅜㅜ 제가 작성한 코드는 다음과 같습니다.

N = int(input())
apt = []
for _ in range(N):
    tmp = []
    for i in input():
        tmp.append(int(i))
    apt.append(tmp)
res = []
q = deque()
for r in range(N):
    for c in range(N):
        if apt[r][c] == 1:
            cnt = 1
            q.append((r,c))
            while q:
                x, y = q.popleft()  # (x,y) 좌표 방문
                apt[x][y] = 0  # 방문한 곳은 0으로 변환
                for i, j in ((x+1,y), (x,y+1), (x-1,y), (x,y-1)):
                    if i >= 0 and i <= N-1 and j >= 0 and j <= N-1:  # 좌표값이 제약조건을 만족할 경우
                        if apt[i][j] == 1:  # 문제 조건을 만족하는지 확인
                            cnt += 1
                            q.append((i,j))
            res.append(cnt)

print(len(res))
for i in sorted(res):
    print(i)

 

위와 같이 작성했고 1번 input으로 답을 확인해보면 단지 갯수는 4개, 각 단지의 아파트 수는 3, 5, 10, 28이 나옵니다. 그런데 답은 3, 5, 10, 22인걸 보니 마지막 단지만 어딘가 잘못 카운트 한 것 같네요. 다른 input들도 확인해보니 다 제일 큰 단지 값만 잘못 나옵니다. 어떤 부분이 잘못된걸까요?ㅜㅜ

bfs python 코테 준비 같이 해요!

답변 1

0

김태원

안녕하세요^^

체크하는 코드의 위치를 바꾸니 나오네요. 왜 그런지는 스스로 분석해보세요.

import sys
from collections import deque
sys.stdin=open("in1.txt", "r")
N = int(input())
apt = []
for _ in range(N):
    tmp = []
    for i in input():
        tmp.append(int(i))
    apt.append(tmp)
res = []
q = deque()
for r in range(N):
    for c in range(N):
        if apt[r][c] == 1:
            cnt = 1
            apt[r][c]=0
            q.append((r,c))
            while q:
                x, y = q.popleft()  # (x,y) 좌표 방문
                for i, j in ((x+1,y), (x,y+1), (x-1,y), (x,y-1)):
                    if i >= 0 and i <= N-1 and j >= 0 and j <= N-1:  # 좌표값이 제약조건을 만족할 경우
                        if apt[i][j] == 1:  # 문제 조건을 만족하는지 확인
                            cnt += 1
                            apt[i][j] = 0  # 방문한 곳은 0으로 변환
                            q.append((i,j))
            res.append(cnt)

print(len(res))
for i in sorted(res):
    print(i)

기존에 윈도우 10으로 잘 써왔는데 윈도우 11로 바꾸고 나서 채점이 안됩니다.

1

77

2

스택에서 ')'을 만나는 경우

0

78

3

문제가 어디있나요?

0

66

2

변수 or 함수명

0

61

1

침몰하는 타이타닉 문제 질문입니다

0

56

1

AA.py 책점 에러

0

57

1

오늘 구매했는데 파이썬 자료구조 궁금한거 있으면 답변이 잘 될까요.

0

111

2

5.동전분배하기 문제 밑에코드도 정답이될까요?

0

110

1

아나그램 비교 코드

0

116

2

AA.PY파일 복사 후 채점 진행할때 오류 발생합니다.

0

160

2

문제 링크가있나여?

0

147

2

채점기 Time Limit Exceeded 오류 문의

1

165

2

동적계획법은 사용하는 문제

0

126

2

제 코드 좀 봐주세요

0

148

1

예외가 존재할 가능성?

0

97

1

3번이 안풀립니다

0

93

0

5번 틀림

0

114

0

오류원인?

0

98

0

리스트 선언

0

106

1

침몰하는 타이타닉(그리디) 문제 질문

0

109

1

알고리즘

0

69

1

코딩테스트

0

92

1

DFS 순서 질문드립니다.

0

128

2

left, right를 사용한 풀이법에 대한 질문입니다

0

91

1