[BOJ] 백준 3273 - 두 수의 합

문제 링크

www.acmicpc.net/problem/3273

3273번: 두 수의 합

 

풀이

1) 각 수의 개수를 세줍니다.

2) 수 $i$마다, ($i$의 개수) * ($x - i$의 개수)를 정답에 더해줍니다.

3) 경우의 수가 두 번씩 중복되었으므로 정답을 2로 나눠준 뒤 출력합시다.

 

 

정답 코드

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

ll n, x;
map<int, int> p;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    cin >> n;
    for (int i = 0; i < n; i++) {
        int c;
        cin >> c;
        p[c]++;
    }
    cin >> x;
    int ans = 0;
    for (auto now: p) {
        auto[i, cnt] = now;
        if (p[x - i]) {
            ans += cnt * p[x - i];
        }
    }
    cout << ans / 2;
}