글 읽기 - 큐가 빌 때까지 다 돌리기 vs K를 찾으면 멈추기

miramira227 1년 전 0

둘 중에 후자가 좋다고 생각하고 코드를 작성했는데 너무 지저분해져서요...

전자로 짜면 if문을 없애고 마지막 출력할 때 K에 해당하는 인덱스만 뽑으면 되서 더 가독성이 좋을거같은데

혹시 이 비루한 코드를 수정할 수 있는 좋은 팁을 부탁드려도 될까요...?

감사합니다!

def BFS(start, K):
    queue = deque([start])
    visited[start][0] = 1
    visited[start][1] = 1
    while queue:
        x = queue.popleft()
        if K in (x+1, x-1, 2*x):
            time = visited[x][0]
            # time이 같은 애들 고르기 
            same_time = list(filter(lambda k: visited[k][0] == time, range(0, len(visited))))

            case = 0
            # 앞에서부터 하나씩 더하기 
            for num in same_time:   # 아예 처음부터 돌면서 다시 계산
                if K in (num+1, num-1, 2*num):
                    if visited[K][0] == 0:    # 처음에는 무조건 처음나옴!
                        case = visited[num][1]
                        visited[K][0] = visited[num][0]
                    else:
                        case += visited[num][1]
            return time, case
        else:
            for next_N in (x+1, x-1, 2*x):
                if next_N >= 0 and next_N <= 100000:
                    if visited[next_N][0] == 0:    # 한번도 방문하지 않은 노드만 queue에 넣기
                        visited[next_N][0] = visited[x][0] + 1
                        visited[next_N][1] = visited[x][1]
                        queue.append(next_N)
                    else:
                        if visited[next_N][0] - 1 == visited[x][0]:
                            visited[next_N][1] += visited[x][1]

if N == K:
    print(0)
    print(1)
else:
    time_, case_ = BFS(N, K)
    print(f'{time_}\n{case_}')

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

문제

문제

출처

ICPC

큐가 빌 때까지 다 돌리기 vs K를 찾으면 멈추기

miramira227 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말