알고리즘

백준 1940 주몽

킨글 2024. 9. 18. 22:02

설명

  • 간단하게 이중 포문을 통해 문제 해결을 하려 했지만, 시간초과가 발생했다.
  • 투포인터를 사용해서 M보다 크면 end를 --해주고 M보다 작으면 start++을 해주었다.

코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.StringTokenizer;

public class Main {

    // 1940
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int N = Integer.parseInt(br.readLine());
        int M = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());
        List<Integer> list = new ArrayList<>();

        for (int i = 0; i < N; i++) {
            int num = Integer.parseInt(st.nextToken());
            list.add(num);
        }
        Collections.sort(list);

        int result = 0;
        int start = 0;
        int end = list.size() - 1;
        while(start < end) {
            int sum = list.get(start) + list.get(end);
            if(sum == M) {
                start++;
                // start++대신 end--로 해도 된다. 
                result++;
            } else if(sum > M) {
                end--;
            } else {
                start++;
            }
        }

        bw.write(result + "");
        bw.flush();
        bw.close();
    }
}