[Baekjoon/C++] 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)

출력

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

예제 입력 1 

9
5 12 7 10 9 1 2 3 11
13

예제 출력 1 

3

---

해결

먼저 수열의 크기와 수열, 합을 입력 받고, 수열을 오름차순으로 정렬한다.

투포인터를 이용해 원하는 합을 찾는다.

(두 수의 합이 x와 같으면 count를 증가시키고 시작점을 오른쪽으로 한 칸 이동한다. 두 수의 합이 x보다 크면 끝점을 왼쪽으로 한 칸 이동한다. 두 수의 합이 x보다 작으면 시작점을 오른쪽으로 한 칸 이동한다. 이 과정을 시작점이 끝점보다 작을 때 반복하며, 최종적으로 찾은 쌍의 개수를 출력한다.)

---

정답

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
  
    int n;
    cin >> n;
    vector<int> arr(n);
    for (int i = 0; i < n; i++)
        cin >> arr[i];
    int x;
    cin >> x;

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

    int count = 0;
    int start = 0;
    int end = n - 1;

    while (start < end) {
        int temp = arr[start] + arr[end];
        if (temp == x) {
            count++;
            start++;
        } else if (temp > x) {
            end--;
        } else {
            start++;
        }
    }

    cout << count;

    return 0;
}

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