[JAVA] 이.코.테 ( 탐색 ) - 부품 찾기

부품 찾기

문제

동빈이네 전자 매장에는 부품이 N개 있다. 각 부품은 정수 형태의 고유한 번호가 있다. 어느 날 손님이 M개의 종류의 부품을 대량으로 구매하겠다며 당일 날 견적서를 요청했다. 동빈이는 때를 놓치지 않고 손님이 문의한 부품 M개 종류를 모두 확인해서 견적서를 작성해야 한다. 이때 가게 안에 부품이 모두 있는지 확인하는 프로그램을 작성해보자.
예를 들어 가게의 부품이 총 5개일 때 부품 번호가 다음과 같다고 하자.

N=5 [8, 3, 7, 9, 2]

손님은 총 3개의 부품이 있는지 확인 요청했는데 부품 번호는 다음과 같다.

M=3 [5, 7, 9]

이때 손님이 요청한 부품 번호의 순서대로 부품을 확인해 부품이 있으면 yes를, 없으면 no를 출력한다. 구분은 공백으로 한다.

입력

• 첫째 줄에 정수 N이 주어진다. (1<=N<=1,000,000)
• 둘째 줄에는 공백으로 구분하여 N개의 정수가 주어진다. 이때 정수는 1보다 크고 1,000,000 이하이다.
• 셋째 줄에는 정수 M이 주어진다. (1<=M<=100,000)
• 넷째 줄에는 공백으로 구분하여 M개의 정수가 주어진다. 이때 정수는 1보다 크고 10억 이하이다.

출력

• 첫째 줄에 공백으로 구분하여 각 부품이 존재하면 yes를, 없으면 no를 출력한다.

입력 예시

5
8 3 7 9 2
3
5 7 9

출력 예시

no yes yes

 

 

 

 

나의 풀이

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

public class SearchPart {

	public static boolean search(int[] parts, int request_parts) {
		// 이진탐색을 하기위하여 찾을 데이터배열을 우선 정렬한다.
		Arrays.sort(parts);
		int start = 0;
		int end = parts.length - 1;

		while (start <= end) {
			int mid = (start + end) / 2;
			if (parts[mid] == request_parts) {// 찾은경우 중간점 반환
				return true;

			} else if (parts[mid] > request_parts) {// 중간점보다 작은경우 왼쪽 확인
				end = mid - 1;
			} else {// 오른쪽 확인
				start = mid + 1;
			}
		}
		return false;

	}

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st ;
		
		int n = Integer.parseInt(br.readLine()); //////// 부품 n개
		int[] parts = new int[n];
		
		st = new StringTokenizer(br.readLine());
		for (int i = 0; i < parts.length; i++) { //// 부품 번호
			parts[i] = Integer.parseInt(st.nextToken());
		}

		int m = Integer.parseInt(br.readLine());
		int[] request_parts = new int[m]; /////// 손님이 문의한 부품 n개
		
		st = new StringTokenizer(br.readLine());
		for (int i = 0; i < request_parts.length; i++) { //// 요청한 부품 번호
			request_parts[i] = Integer.parseInt(st.nextToken());
		}

		br.close();

		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

		for (int i = 0; i < request_parts.length; i++) {
			if (search(parts, request_parts[i])) {
				bw.write("yes ");
			} else {
				bw.write("no ");
			}
		}
		bw.close();
	}
}

 

이진 탐색을 사용하여 문제를 해결 하였습니다.

이진 탐색은 정렬이 우선적으로 되야 하므로 부품전체 데이터를 오름차순 정렬을 시키고나서

찾으려고 하는 데이터를 하나씩 가져와서 이진 탐색을 시켰습니다.

풀이시간 : 25분