백준 2230번: 수 고르기

문제 설명

• https://www.acmicpc.net/problem/2230
• 수열의 두 수를 뽑았을 때 그 차이가 M이상인 것 중에서 차이가 가장 적은 값을 고르는 문제입니다.

접근법

• 첫번째, 모든 수를 비교하면 최솟값을 찾을 수 있습니다. 하지만 O(N^2)의 시간으로 시간초과가 발생합니다.
• 두번째, 값을 정렬하면 조금 더 시간을 절약할 수 있습니다. 만약 {1,2,3,4,5,6,7}이 있을 때 1과 4를 비교해서 3차이가 난다는 걸 알았다면 뒤에 있는 5,6,7은 1과의 비교에서 반드시 3보다 큰 차이가 납니다.
  • if 'A[x] - A[y] > M' then 'A[x] - A[t] > M' (y<t, A is sorted)
  • 조건을 만족하지 못하는 최악의 경우에는 여전히 O(N^2)의 시간복잡도가 소요됩니다.
• 세번째, 투포인터를 이용합니다. 숫자를 오름차순으로 정렬한 뒤 M보다 크면 오른쪽 포인터를 한칸, M보다 작으면 왼쪽 포인터를 한칸 이동시키면 M과 가까운 값들만 빠르게 탐색할 수 있습니다.

투포인터

while(left < N){
	// 경험상 leftIdx를 +1하는 조건을 if문에 넣는게 조금 더 설계가 편리했습니다.
	if(now >= M){ 
    	값 갱신 
    	leftIdx++;
    }else if(rightIdx == N-1){ // 값을 갱신하지 않기 때문에 첫번째 조건과 분리했습니다.
    	leftIdx++;
    }else{
    	rightIdx++;
    }
}

정답

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

public class Main {
	public static void main(String[] args) throws Exception{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());		
		int N = Integer.parseInt(st.nextToken());
		int M = Integer.parseInt(st.nextToken());
		int[] arr = new int[N];
		for (int i = 0; i < arr.length; i++) {
			arr[i] = Integer.parseInt(br.readLine());
		}
		Arrays.sort(arr);
		int leftIdx = 0;
		int rightIdx = 0;
		int minVal = Integer.MAX_VALUE;
		while(leftIdx < N) {
			int val = arr[rightIdx] - arr[leftIdx];
			if(val>= M) {
				minVal = Math.min(minVal, val);
				leftIdx++;
			}else if(rightIdx == N-1){
				leftIdx++;
			}else {
				rightIdx++;
			}
			
		}
		System.out.println(minVal);
		
		
	}
}