7453번 합이 0인 네 정수

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

(A, B), (C, D)

2개의 그룹으로 나누어

각 그룹 내에서 나올 수 있는 모든 합의 경우의 수를 구해준다.

그리고 ab를 기준으로 오름차순 정렬을 한 다음에

lower_bound와 upper_bound를 활용하여 개수를 구해준다.

#include <bits/stdc++.h>
#include <unordered_map>
 
using namespace std;
 
vector<int> A, B, C, D, ab, cd;
int main(void)
{
    //freopen("C:\\Users\\park7\\Desktop\\sample_input.txt", "r", stdin);
 
    vector<int> v = { 1, 2, 3, 3, 3, 3, 4, 5 };
                    //0, 1, 2, 3, 4, 5, 6, 7 
    int idx = lower_bound(v.begin(), v.end(), 3) - v.begin(); // 2
    int idx2 = upper_bound(v.begin(), v.end(), 3) - v.begin(); // 6
 
    // 3의 개수 = 6 - 2 = 4
    printf("%d %d\n", idx, idx2);
 
    return 0;
}

#include <bits/stdc++.h>
 
using namespace std;
 
vector<int> A, B, C, D, ab, cd;
 
int main(void)
{
    //freopen("C:\\Users\\park7\\Desktop\\sample_input.txt", "r", stdin);
    int n, a, b, c, d;
    long long ans = 0;
    scanf("%d", &n);
 
    for (int i = 0; i < n; i++)
    {
        scanf("%d %d %d %d", &a, &b, &c, &d);
        A.push_back(a);
        B.push_back(b);
        C.push_back(c);
        D.push_back(d);
    }
 
    for (int i = 0; i < n; i++)
        for (int j = 0; j < n; j++)
        {
            ab.push_back(A[i] + B[j]);
            cd.push_back(C[i] + D[j]);
        }
 
    sort(ab.begin(), ab.end());
 
    for (auto i : cd)
    {
        int idx1 = lower_bound(ab.begin(), ab.end(), -i) - ab.begin();
        int idx2 = upper_bound(ab.begin(), ab.end(), -i) - ab.begin();
 
        if (idx1 == ab.size()) continue;
 
        ans += abs(idx1 - idx2);
    }
 
    printf("%lld\n", ans);
 
    return 0;
}