글 읽기 - PyPy, python3 - 런타임 오류 관련

gudrhks2 3년 전 0

파이썬으로 알고리즘을 안풀어본지 오래 되어,

한번 풀어보았는데

pypy 31%에서 런타임 에러가 발생합니다.

그런데 맞은사람에서 python3, pypy 정답자가 없는것을 보고 질문 드립니다.

import sys

NO_ANSWER_POINT = 9


def solution(N, number):
    clump_N = N
    dp = [{}]
    count = 1
    while count < NO_ANSWER_POINT :
        cur_set = set({clump_N})
 
        for idx in range(1, count//2+1):
            for i in dp[count-idx]:
                for j in dp[idx]:
                    cur_set.add(i + j)
                    cur_set.add(i * j)
 
                    # 교환법칙
                    cur_set.add(abs(i - j))
                    if i != 0 and j != 0:
                        cur_set.add(i // j)
                        cur_set.add(j // i)
 
        if number in cur_set :
            break
        dp.append(cur_set)
        clump_N = clump_N * 10 + N
        count +=1
    answer = count if count < NO_ANSWER_POINT else "NO"
    return answer
 
N = int(sys.stdin.readline().rstrip())
K = int(sys.stdin.readline().rstrip())
numbers=[sys.stdin.readline() for i in range(K)]
results = [solution(N, int(number)) for number in list(map(int,numbers))]
print("\n".join(map(str, results)))

gudrhks2 3년 전 0

P.S. 같은 알고리즘으로 자바의 경우에는 통과하였습니다.

import java.io.*;
import java.util.*;

class boj_2087 {
    private static final int NO_ANSWER_POINT = 9;

    public static void main(final String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb =new StringBuilder();
        int N = stoi(br.readLine());
        int K = stoi(br.readLine());
        int tCase = 1;
        while(tCase <= K){
            int number = stoi(br.readLine());
            int answer = solution(N, number);
            sb.append( (answer != -1 )? answer : "NO");
            sb.append("\n");
            tCase++;
        }
        System.out.print(sb);
    }

    public static int solution(final int N, final int number) {
        List<Set<Integer>> dp = new ArrayList<>(9);
        dp.add(new HashSet<Integer>());
        
        int clump_N = N;
        int count = 1;
        
        while (count < NO_ANSWER_POINT){
            Set<Integer> curSet = new HashSet<>();
            curSet.add(clump_N);

            for(int idx = 1; idx < count/2 +1; idx++ ){
                for(int i : dp.get(idx)){
                    for(int j : dp.get(count-idx)){
                        if(i == 0 || j == 0) continue;
                        curSet.add(i+j);
                        curSet.add(i*j);
                        
                        //교환법칙
                        curSet.add(Math.abs(i-j));
                        if(i!=0 && j!=0)
                            curSet.add(i/j);
                            curSet.add(j/i);
                    }
                }
            }
            if(curSet.contains(number))
                break;
            dp.add(curSet);
            clump_N = clump_N*10 + N;
            count++;
        }

        return (count < NO_ANSWER_POINT)? count : -1;
    }

    private static int stoi(final String readLine) {
        return Integer.parseInt(readLine);
    }
}

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

문제

문제

출처

ICPC

PyPy, python3 - 런타임 오류 관련

gudrhks2 3년 전 0

gudrhks2 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말