알고리즘
백준 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;
}
}