알고리즘

백준 1940번 주몽 (JAVA)

박카스마시며코딩 2023. 11. 30. 19:17

https://www.acmicpc.net/problem/1940

 

1940번: 주몽

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

www.acmicpc.net

 

저는 투포인트를 통해 문제를 해결하였습니다.

먼저 입력을 정렬하고, 처음과 끝을 포인터로 가리키고, 합이 m과 같다면 처음 포인터를 늘리고 끝 포인트를 줄이고, m보다 작다면 처음 포인터를 늘리고, m보다 크다면 끝 포인트를 줄여 합이 m과 같아지는 것이 몇 쌍인지를 판단하였습니다.

 

package BOJ.twopoint;

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

public class BOJ_1940_2 {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        Function<String,Integer> stoi = Integer::parseInt;
        int n = stoi.apply(br.readLine());
        int m = stoi.apply(br.readLine());
        int[] input = new int[n];
        StringTokenizer st = new StringTokenizer(br.readLine()," ");
        for(int i = 0 ; i < n ; i++){
            input[i] = stoi.apply(st.nextToken());
        }

        System.out.println(binarySearch(input,n,m));
    }

    private static int binarySearch(int[] input, int n, int m) {
        Arrays.sort(input);
        int startIndex = 0;
        int endIndex = n-1;
        int cnt = 0;
        while(startIndex < endIndex){
            int sum = input[startIndex] + input[endIndex];
            if(sum == m){
                cnt++;
                startIndex++;
                endIndex--;
            }
            if(sum < m){
                startIndex++;
            }
            if(sum > m){
                endIndex--;
            }
        }
        return cnt;
    }
}