[JAVA] BOJ(백준) - 두 수의 합 - 3273

문제내용

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

3273번: 두 수의 합

n개의 서로 다른 양의 정수 a1, a2, ..., an으로 이루어진 수열이 있다. ai의 값은 1보다 크거나 같고, 1000000보다 작거나 같은 자연수이다. 자연수 x가 주어졌을 때, ai + aj = x (1 ≤ i < j ≤ n)을 만족하는 (ai, aj)쌍의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 수열의 크기 n이 주어진다. 다음 줄에는 수열에 포함되는 수가 주어진다. 셋째 줄에는 x가 주어진다. (1 ≤ n ≤ 100000, 1 ≤ x ≤ 2000000)

출력

문제의 조건을 만족하는 쌍의 개수를 출력한다.

---

문제풀이

풀이시간이 5분밖에 안걸렸을 만큼 방법만 알면 매우 쉬운문제고, 몰라도 충분히 풀수 있다.

먼저 필자가 예전에 했던 포스팅중에 투 포인터 알고리즘에 대한 포스팅이 있다. 이 문제도 마찬가지로 투포인터를 이용해 푸는 문제인데, 필자가 사용한 투 포인터에 간략하게 설명하면 다음과 같다.

 

1. start, end 두개의 포인터 변수를 만들어준다

2. start=0, end=마지막인덱스 로 설정하고, arr[start]+arr[end] 의 값과 x 값을 비교해준다.

3. x값이 더 크면 start++, x값이 더 작으면 end--, x값과 같아지면 카운트 후 start++ 해준다.

 

이처럼 투 포인터는 두개의 포인터를 이용해 빠르게 특정값을 찾으려고 할때 사용된다. 이때 배열or리스트 저장공간의 수열들은 이분탐색처럼 정렬이 돼있어야 한다는 조건이 있다.

 

코드를보자.

---

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
	static int n,x;
	static int[] arr;
	public static void main(String[] args) throws IOException{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		n = Integer.parseInt(br.readLine());
		arr = new int[n];
		
		StringTokenizer st = new StringTokenizer(br.readLine());
		for(int i=0; i<n; i++)
			arr[i] = Integer.parseInt(st.nextToken());
		
		x = Integer.parseInt(br.readLine());
		
		Arrays.sort(arr);//정렬
		int s=0 , e=n-1, ans=0;//start, end, 정답
		while(s<e) {
			int sum = arr[s]+arr[e];
			if(sum==x) {
				ans++;
				s++;
				continue;
			}
			else if(sum<x) s++;
			else e--;
		}
		
		System.out.println(ans);
	}
}

---

마치며

투 포인터의 아주 기본적인 문제다. 투포인터 알고리즘 문제를 잘 모른다면,,, 한번 알고리즘에 대해 검색해보고 다시 풀어보는걸 추천한다.