알고리즘

[백준 1940] 주몽 Java 풀이

빠르게 핵심만 2024. 4. 22. 15:27
 

1940번: 주몽

첫째 줄에는 재료의 개수 N(1 ≤ N ≤ 15,000)이 주어진다. 그리고 두 번째 줄에는 갑옷을 만드는데 필요한 수 M(1 ≤ M ≤ 10,000,000) 주어진다. 그리고 마지막으로 셋째 줄에는 N개의 재료들이 가진 고

www.acmicpc.net

 

접근

n이 최대 15,000이므로 이중 for 문을 이용하여 문제를 풀 경우 시간 복잡도가 O(n^2)이 될 것으로 예상됩니다. 따라서 투 포인터를 이용한다면 시간 복잡도를 O(n)으로 줄일 수 있습니다.

 

코드

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

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

		// 입력 처리
		int n = Integer.parseInt(br.readLine());
		int m = 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());
		}

		// 1. 배열 arr를 오름차순으로 정렬한다.
		Arrays.sort(arr);

		// 2. 투 포인터를 이용하여 만들 수 있는 갑옷의 개수를 구한다.
		int i = 0, j = n - 1;
		int cnt = 0;
		while (i < j) {
			if (arr[i] + arr[j] < m) {
				i++;
			} else if (arr[i] + arr[j] > m) {
				j--;
			} else {
				cnt++;
				i++;
				j--;
			}
		}

		// 결과 출력
		System.out.println(cnt);
	}
}