[백준] 3273번 두 수의 합 (C++)

문제

https://www.acmicpc.net/problem/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)

출력

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

---

풀이

이 문제는 입력된 n개의 숫자들 중 두 수의 합이 x가 될 수 있는 모든 경우의 수를 구하는 문제이다.

 

O(n^2)으로 풀이하게 되면 시간 초과가 나오기 때문에 더 효율적인 방법으로 풀이해야한다.

 

시간 초과 풀이

#include <iostream>
#include <vector>

using namespace std;

int main() {
	int n, x, count = 0;

	cin >> n;
	vector <int> vec(n);
	for (int i = 0; i < n; i++) {
		cin >> vec[i];
	}
	cin >> x;

	for (int i = 0; i < n; i++) {
		int a = vec[i];
		for (int j = i + 1; j < n; j++) {
			int b = vec[j];

			if (a + b == x) count++;
		}
	}

	cout << count;

	return 0;
}

처음에 나도 O(n^2)으로 접근했다가 시간 초과가 나서 다른 방법을 생각했다.

 

 

투 포인터를 이용하여 해당 문제를 풀이했다.

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main() {
	int n, x, count = 0;

	cin >> n;
	vector <int> vec(n);
	for (int i = 0; i < n; i++) {
		cin >> vec[i];
	}
	cin >> x;

	sort(vec.begin(), vec.end());

	int left = 0, right = n - 1;
	while (left < right) {
		int value = vec[left] + vec[right];
		if (value > x) {
			right--;
		}
		else if (value < x) {
			left++;
		}
		else {
			count++;
			left++;
			right--;
		}
	}
	
	cout << count;
	return 0;
}