3273: 두 수의 합

• toc {:toc}

문제

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)

출력

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

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

풀이

제한시간 1초이기 때문에 대략 1억 연산을 할 수 있다고 생각.

수열의 크기 자체가 최대 10만이기 때문에 단순 브루트포스로는 시간초과가 발생함.

===시간 단축을 위해 used 배열을 통해 숫자가 거쳐갔는지 확인한다.

===이를 통해 x-arr[i]를 통한 인덱스 숫자가 있고 없음을 판단하고

===true일 경우 카운트 할 수 있다.

#include <bits/stdc++.h>
using namespace std;
 
int arr[1000005];
bool used[2000001];
 
int main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	int n, x, cnt;
	cin >> n;
	for(int i=1; i<=n; i++){
		cin >> arr[i];
	}
	cin >> x;
	
	cnt = 0;
	
	for(int i=1; i<=n; i++){
		if(x-arr[i]>0 && used[x-arr[i]])
			cnt++;
		used[arr[i]]=true;
	}
	cout << cnt << '\n';
	
	return 0;
}