[Java] 1940 주몽

문제

주몽은 철기군을 양성하기 위한 프로젝트에 나섰다. 그래서 야철대장을 통해 철기군이 입을 갑옷을 만들게 하였다. 야철대장은 주몽의 명에 따르기 위하여 연구에 착수하던 중 아래와 같은 사실을 발견하게 되었다.

갑옷을 만드는 재료들은 각각 고유한 번호를 가지고 있다. 갑옷은 두 개의 재료로 만드는데 두 재료의 고유한 번호를 합쳐서 M(1 ≤ M ≤ 10,000,000)이 되면 갑옷이 만들어 지게 된다. 야철대장은 자신이 만들고 있는 재료를 가지고 갑옷을 몇 개나 만들 수 있는지 궁금해졌다. 이러한 궁금증을 풀어 주기 위하여 N(1 ≤ N ≤ 15,000) 개의 재료와 M이 주어졌을 때 몇 개의 갑옷을 만들 수 있는지를 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에는 재료의 개수 N(1 ≤ N ≤ 15,000)이 주어진다. 그리고 두 번째 줄에는 갑옷을 만드는데 필요한 수 M(1 ≤ M ≤ 10,000,000) 주어진다. 그리고 마지막으로 셋째 줄에는 N개의 재료들이 가진 고유한 번호들이 공백을 사이에 두고 주어진다. 고유한 번호는 100,000보다 작거나 같은 자연수이다

 

출력

첫째 줄에 갑옷을 만들 수 있는 개수를 출력한다.

 

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

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

        int material = Integer.parseInt(bufferedReader.readLine());
        int Num = Integer.parseInt(bufferedReader.readLine());
        int[] materialNum = new int[material];

        StringTokenizer stringTokenizer = new StringTokenizer(bufferedReader.readLine());

        for(int i=0; i<material; i++){
            materialNum[i] = Integer.parseInt(stringTokenizer.nextToken());
        }
        Arrays.sort(materialNum);

        int count = 0;
        int i = 0;
        int j = material - 1;

        while(i < j){
            if(materialNum[i] + materialNum[j] < Num){
                i++;
            }
            else if(materialNum[i] + materialNum[j] > Num){
                j--;
            }
            else{
                count++;
                i++;
                j--;
            }
        }
        System.out.println(count);
        bufferedReader.close();
    }
}

▶ int material = Integer.parseInt(bufferedReader.readLine()) : 사용자로부터 입력받은 값을 Integer형으로 변환 후 material 변수에 저장함

 

▶ int Num = Integer.parseInt(bufferedReader.readLine()) : 사용자로부터 입력받은 값을 Integer형으로 변환 후 Num 변수에 저장함

 

▶ int[] materialNum = new int[material] : 재료들이 가진 고유한 번호들의 배열을 저장하기 위해 재료의 수만큼의 크기를 가진 배열 materialNum을 선언함

 

▶ for(int i=0; i<material; i++){ materialNum[i] = Integer.parseInt(stringTokenizer.nextToken()) } : 재료의 수만큼 반복문을 돌리며 materialNum 배열에 들어갈 값을 사용자로부터 입력받음

 

▶ Arrays.sort(materialNum) : sort() 메서드를 사용해 materialNum 배열에 저장된 값들을 오름차순으로 정렬함

 

▶ int j = material - 1 : i가 0번 index부터 시작하기 때문에 j는 재료의 수보다 하나 더 작은 수로 초기화 함

 

while(i < j){
    if(materialNum[i] + materialNum[j] < Num){
        i++;
    }
    else if(materialNum[i] + materialNum[j] > Num){
        j--;
    }
    else{
        count++;
        i++;
        j--;
    }
}

▶ 'j가 i보다 크다'는 조건이 참인 동안 반복함

    → materialNum[i] + materialNum[j]이 Num(갑옷이 되는 번호)보다 작은 경우 작은 번호 index 올림

    → materialNum[i] + materialNum[j]이 Num(갑옷이 되는 번호)보다 큰 경우 큰 번호 index를 하나 내림

    → materialNum[i] + materialNum[j]이 Num(갑옷이 되는 번호)과 같은 경우 count를 증가시키고 양쪽 포인터를 모두 이동시킴