글 읽기 - 100% 에서 틀리는 이유가 있을까요

imonlyone 1년 전 0

다른 예제가 다 잘 맞는데 어디서 문제가 생긴지 잘 모르겠어요...

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

public class Main {
    static BufferedReader br;
    static StringBuilder sb;
    static int N;
    static Arr[] arr;

    public static void main(String[] args) throws IOException {
        input();
        logic();
    }

    static void input() throws IOException {
        br = new BufferedReader(new InputStreamReader(System.in));
        sb = new StringBuilder();
        N = Integer.parseInt(br.readLine());
        arr = new Arr[N];
        for (int i = 0; i < N; i++) {
            String str = br.readLine();
            arr[i] = new Arr(str);
        }
    }

    static void logic() {
        Arrays.sort(arr);

        int r = 0;

        for (int i = 0; i < N - 1; i++) {
            if (!arr[i].str.equals(arr[i + 1].str)) {
                r++;
                sb.append(arr[i].str).append('\n');
                if (r == N - 1)
                    sb.append(arr[i + 1].str);
            } else {
                r++;
                if (arr[i].str.equals(arr[i + 1].str) && r == N - 1)
                    sb.append(arr[i].str);
            }
        }

        System.out.println(sb);
    }

    static class Arr implements Comparable<Arr> {
        public String str;

        public Arr(String str) {
            this.str = str;
        }

        @Override
        public int compareTo(Arr other) {
            if (this.str.length() == other.str.length()) {
                return this.str.compareTo(other.str);
            }
            return Integer.compare(this.str.length(), other.str.length());
        }
    }
}

siyamaki 1년 전 0

2개 이상을 가정하고 푸셨는데 N은 1개도 들어올 수 있습니다.

1
im
output: 없음
answer: im

siyamaki 1년 전 0

꼼수를 하나 드리자면 set은 중복을 허용하지 않아 아래와 같이 쓸 수도 있습니다

HashSet<String> set = new HashSet<>();
for(int i = 0; i < N; i++) {
    set.add(br.readLine());
}

ArrayList<String> list = new ArrayList<>(set);
Collections.sort(list, (String s1, String s2) -> {
    if(s1.length() == s2.length()) {
        return s1.compareTo(s2);
    }
    return s1.length() - s2.length();
});

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

문제

문제

출처

ICPC

100% 에서 틀리는 이유가 있을까요

imonlyone 1년 전 0

siyamaki 1년 전 0

siyamaki 1년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말