이진탐색 부품찾기

jaegeunsong97·2023년 3월 12일
0
post-thumbnail

4가지 방법인데 이걸 이렇게 풀수 있다는 것에 매우 놀랐다. ㅎㅅㅎ..... 난 이진탐색만 계속 생각했는데..... 쩝....

🔔 Scanner + 이진탐색

import java.util.*;

public class Main {

    // 이진 탐색 소스코드 구현(반복문)
    public static int binarySearch(int[] arr, int target, int start, int end) {
        while (start <= end) {
            int mid = (start + end) / 2;
            // 찾은 경우 중간점 인덱스 반환
            if (arr[mid] == target) return mid;
            // 중간점의 값보다 찾고자 하는 값이 작은 경우 왼쪽 확인
            else if (arr[mid] > target) end = mid - 1;
            // 중간점의 값보다 찾고자 하는 값이 큰 경우 오른쪽 확인
            else start = mid + 1; 
        }
        return -1;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        
        // N(가게의 부품 개수)
        int n = sc.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = sc.nextInt();
        }
        
        // 이진 탐색을 수행하기 위해 사전에 정렬 수행
        Arrays.sort(arr);

        // M(손님이 확인 요청한 부품 개수)
        int m = sc.nextInt();
        int[] targets = new int[m];
        for (int i = 0; i < m; i++) {
            targets[i] = sc.nextInt();
        }

        // 손님이 확인 요청한 부품 번호를 하나씩 확인
        for (int i = 0; i < m; i++) {
            // 해당 부품이 존재하는지 확인
            int result = binarySearch(arr, targets[i], 0, n - 1);
            if (result != -1) {
                System.out.print("yes ");
            }
            else {
                System.out.print("no ");
            }
        }
    }

🔔 Buffered + 이진 탐색

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

public class Main {
  public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    
    int N = Integer.parseInt(br.readLine());
    int[] arr = new int[N];
    StringTokenizer st = new StringTokenizer(br.readLine());
    for (int i = 0; i < N; i++) arr[i] = Integer.parseInt(st.nextToken());

    Arrays.sort(arr);
    
    int M = Integer.parseInt(br.readLine());
    int[] find = new int[M];
    st = new StringTokenizer(br.readLine());
    for (int i = 0; i < M; i++) find[i] = Integer.parseInt(st.nextToken());

    for (int i = 0 ; i < M; i++) {
      int result = binarySearch(arr, find[i], 0, N - 1);
      
      if (result != -1) System.out.print("yes ");
      else System.out.print("no ");
    }
  }

  public static int binarySearch(int[] arr, int target, int start, int end) {
    while (start <= end) {
      int mid = (start + end) / 2;
      if (target == arr[mid]) return mid;
      else if (target < arr[mid]) end = mid - 1;
      else start = mid + 1;
    }
    
    return -1;
  }
}

🔔 계수 정렬

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

public class Main {

    public static void main(String[] args) thrwos IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        
        // N(가게의 부품 개수)
        int n = Integer.parseInt(br.readLine());
        int[] arr = new int[1000001];
        StringTokenizer st = new StringTokenizer(br.readLine());
        
        for (int i = 0; i < n; i++) {
            int x = Integer.parseInt(st.nextToken());
            arr[x] = 1;
        }

        // M(손님이 확인 요청한 부품 개수)
        int m = Integer.parseInt(br.readLine());
        int[] targets = new int[m];
        st = new StringTokenizer(br.readLine());
        
        for (int i = 0; i < m; i++) {
            targets[i] = Integer.parseInt(st.nextToken());
        }

        // 손님이 확인 요청한 부품 번호를 하나씩 확인
        for (int i = 0; i < m; i++) {
            // 해당 부품이 존재하는지 확인
            if (arr[targets[i]] == 1) System.out.print("yes ");
            else System.out.print("no ");
        }
    }
}

🔔 Set

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

public class Main {

    public static void main(String[] args) thrwos IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        
        // N(가게의 부품 개수)
        int n = Integer.parseInt(br.readLine());
        // 집합(Set) 정보를 처리하기 위한 HashSet 라이브러리
        HashSet<Integer> set = new HashSet<>();
        StringTokenizer st = new StringTokenizer(br.readLine());
        
        for (int i = 0; i < n; i++) {
            int x = Integer.parseInt(st.nextToken());
            set.add(x);
        }

        // M(손님이 확인 요청한 부품 개수)
        int m = Integer.parseInt(br.readLine());
        int[] targets = new int[m];
        st = new StringTokenizer(br.readLine());
        
        for (int i = 0; i < m; i++) {
            targets[i] = Integer.parseInt(st.nextToken());
        }

        // 손님이 확인 요청한 부품 번호를 하나씩 확인
        for (int i = 0; i < m; i++) {
            // 해당 부품이 존재하는지 확인
            if (set.contains(targets[i])) System.out.print("yes ");
            else System.out.print("no ");
        }
    }
}
profile
현재 블로그 : https://jasonsong97.tistory.com/

0개의 댓글