99클럽 코테 스터디 14일차 TIL , 숫자 카드 2

문제

풀이

제출 1 - 오답(시간초과)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
    static int[] numberCards;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        numberCards = new int[n];
        StringTokenizer numberToken = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            numberCards[i] = Integer.parseInt(numberToken.nextToken());
        }
        Arrays.sort(numberCards);

        StringBuilder answer = new StringBuilder();

        int m = Integer.parseInt(br.readLine());
        StringTokenizer checkToken = new StringTokenizer(br.readLine());
        for (int i = 0; i < m; i++) {
            int x = Integer.parseInt(checkToken.nextToken());
            answer.append(findAndCount(x, 0, numberCards.length-1))
                    .append(" ");
        }

        System.out.println(answer.toString().trim());

    }

    private static int findAndCount(int x, int lt, int rt) {
        int cnt = 0;
        while(rt >= lt){
            int center = (rt+lt) / 2;
            if(x > numberCards[center]){
                lt = center + 1;
            }else if(x < numberCards[center]){
                rt = center - 1;
            }else{
                cnt += 1;
                cnt += findAndCount(x, lt, center-1);
                cnt += findAndCount(x, center+1, rt);
                break;
            }
        }
        return cnt;
    }

}

 

x와 같은 값을 찾았을 때 재귀를 사용해 앞뒤로 같은 값을 가진 것이 더 있는지 확인하는 로직인데 재귀 때문에 시간초과가 발생했다.

제출 2 - 정답

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
    static int[] numberCards;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        numberCards = new int[n];
        StringTokenizer numberToken = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            numberCards[i] = Integer.parseInt(numberToken.nextToken());
        }
        Arrays.sort(numberCards);

        StringBuilder answer = new StringBuilder();

        int m = Integer.parseInt(br.readLine());
        StringTokenizer checkToken = new StringTokenizer(br.readLine());
        for (int i = 0; i < m; i++) {
            int x = Integer.parseInt(checkToken.nextToken());
            answer.append(upperFirst(x) - sameFirst(x))
                    .append(" ");
        }

        System.out.println(answer.toString().trim());

    }

    private static int sameFirst(int x) {
        int lt = 0;
        int rt = numberCards.length;
        while(lt < rt){
            int center = (lt+rt)/2;
            if(numberCards[center] >= x){
                rt = center;
            }else{
                lt = center + 1;
            }
        }
        return lt;
    }

    private static int upperFirst(int x) {
        int lt = 0;
        int rt = numberCards.length;
        while(lt < rt){
            int center = (lt+rt)/2;
            if(numberCards[center] > x){
                rt = center;
            }else{
                lt = center + 1;
            }
        }
        return lt;
    }


}

 

방법을 바꾸어서 x와 같은 첫 인덱스를 찾고 x보다 큰 값을 가지는 첫번째 인덱스를 찾아서 빼주어서 개수를 찾는 방식이다.

이분탐색을 두번만 사용하니 더 빠르다.

TIL

• 이분탐색을 통해 개수를 구하는 방법을 알았다.