글 읽기 - 이거 왜 메모리 초과나나요ㄷㄷ

blucky8649 2년 전 0

같은 로직을 자바로 작성하였을 때는 통과되는데, 코틀린으로 작성하면 메모리초과가 나오네요.

import java.lang.StringBuilder

private lateinit var parents : Array<Int>
fun main() = with(System.`in`.bufferedReader()) {
    val t = readLine().toInt()
    val sb = StringBuilder()
    for(tc in 1 .. t) {
        sb.append("Scenario $tc: \n")
        val n = readLine().toInt()
        val k = readLine().toInt()
        parents = Array(n + 1) {i -> i}

        repeat(k) {
            val (a, b) = readLine().split(" ").map {it.toInt()}
            union(a, b)
        }
        repeat(readLine().toInt()) {
            val(a, b) = readLine().split(" ").map { it.toInt() }
            sb.append( if (find(a) == find(b)) 1 else 0 ).append('\n')
        }
        if (tc < t) sb.append('\n')
    }
    println(sb)
}
fun find(x : Int) : Int {
    if (parents[x] == x) return x
    parents[x] = find(parents[x])
    return parents[x]
}

fun union(a: Int, b: Int) {
    val parentA = find(a)
    val parentB = find(b)

    if (parentB == parentA) return
    parents[parentB] = parentA
}

blucky8649 2년 전 0

통과했던 자바코드도 올려드립니다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

public class Main {
    static int[] parents;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int t = Integer.parseInt(br.readLine());
        StringBuilder sb = new StringBuilder();
        for (int tc = 0; tc < t; tc++) {
            sb.append("Scenario ").append(tc + 1).append(":").append('\n');
            int n = Integer.parseInt(br.readLine()); //유저의 수
            int k = Integer.parseInt(br.readLine()); //친구 관계의 수
            StringTokenizer st;
            parents = new int[n + 1];

            for (int i = 0; i <= n; i++) {
                parents[i] = i;
            }

            for (int i = 0; i < k; i++) {
                st = new StringTokenizer(br.readLine(), " ");
                int start = Integer.parseInt(st.nextToken());
                int end = Integer.parseInt(st.nextToken());
                union(start, end);
            }

            // 이제 특정 친구관계가 존재하는지 알아봐야함
            int m = Integer.parseInt(br.readLine());
            for (int i = 0; i < m; i++) {
                st = new StringTokenizer(br.readLine(), " ");
                int a = Integer.parseInt(st.nextToken());
                int b = Integer.parseInt(st.nextToken());
                sb.append(find(a) == find(b) ? 1 : 0).append('\n');
            }
            if (tc < t - 1)
                sb.append('\n');
        }
        System.out.print(sb);
    }

    static int find(int x) {
        if (parents[x] == x) {
            return x;
        }
        return parents[x] = find(parents[x]);
    }

    static void union(int a, int b) {
        a = find(a);
        b = find(b);

        if (a == b) return;
        parents[b] = a;
    }
}

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

문제

문제

출처

ICPC

이거 왜 메모리 초과나나요ㄷㄷ

blucky8649 2년 전 0

blucky8649 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말