글 읽기 - 파이썬 런타임 오류 ...ㅜㅜ 고수님들 제발 도와주세요 시험이 코앞입니다

mike1201 5년 전 0

알고리즘은 맞게 짠 것 같은데.... 질문 리스트에 언급된 반례 케이스도 다 되고요... 왜 안돌아갈까요 ㅜㅜ?

코드 간단하게 짰으니 고수님들 보시고 답변 부탁드려요 ㅜㅜ

def dfs(n):
    global num_ans
    if visited[n] == 1:
        for idx_, n_ in enumerate(array):
            if n_ == n:
                for idx__ in range(idx_, len(array)):
                    team[ array[idx__] ] = 1
                break
            team[ array[idx_] ] = -1
            num_ans += 1
        return
    if team[n] == 1:
        for n_ in array:
            team[n_] = -1; num_ans += 1
        return
    elif team[n] == -1:
        for n_ in array:
            team[n_] = -1; num_ans += 1
        return

    visited[n] = 1
    array.append(n)
    dfs( graph[n]-1 )

from collections import deque
import sys
T = int(sys.stdin.readline().strip())
for t in range(T):
    N = int(sys.stdin.readline().strip())
    if N == 1:
        tmp = list(map(int, sys.stdin.readline().strip().split()))
        print(0)
    else:
        graph = list(map(int, sys.stdin.readline().strip().split()))
        team = [0]*N
        num_ans = 0
        visited = [0]*N
        for n in range(N):
            if graph[n]-1 == n:
                team[n] = 1
                continue
            if team[n] == 1:
                continue
            if team[n] == -1:
                continue
            array = []
            dfs(n)
            for n_ in array:
                visited[n_] = 0
        print(num_ans)

heumsi 5년 전 0

일단

import sys

sys.setrecursionlimit(10**8)

를 상단에 추가해주세요.

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

문제

문제

출처

ICPC

파이썬 런타임 오류 ...ㅜㅜ 고수님들 제발 도와주세요 시험이 코앞입니다

mike1201 5년 전 0

heumsi 5년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말