[백준] 3694번 : 로봇 프로젝트

문제

www.acmicpc.net/problem/3649

3649번: 로봇 프로젝트

문제해결방법

레고 조각 배열을 정렬하고 양 끝에서부터 구멍을 막을 수 있는 조각이 있는지 조사한다.

양 끝 조각의 합이 구멍보다 큰 경우 조각이 큰 쪽을 하나 감소, 작은 경우 조각이 작은 쪽을 하나 증가.

만약 구멍을 막을 수 있는 조각을 찾았다면 이후의 조각들은 구멍을 막을 수 있는 조각이 있더라고 차이의 절댓값이 더 작으므로 더 탐색하지 않아도 된다.

* 입력을 계속 받아야 하는 부분을 while이 아니라 if로 해서 한참을 못찾았다. 정신 차리자.

코드

package baekjoon;
 
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
 
public class Main3694_로봇프로젝트 {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String str;
        while((str = br.readLine())!=null) {
            int X = Integer.parseInt(str)*10000000;
            int N = Integer.parseInt(br.readLine());
            int[] legos = new int[N];
            for(int i=0; i<legos.length; i++) legos[i] = Integer.parseInt(br.readLine());
            Arrays.sort(legos);
            int s = 0, e = N-1;
            boolean flag = false;
            while(s<e) {
                int sum = legos[s]+legos[e];
                if(sum==X) {
                    flag = true;
                    break;
                }
                else if(sum<X) s++;
                else e--;
            }
            if(flag) System.out.println("yes "+legos[s]+" "+legos[e]);
            else System.out.println("danger");
        }
    }
}

코드에 대한 피드백이나 더 좋은 아이디어는 언제나 환영입니다.