백준 7795번 C++ 풀이

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

7795번: 먹을 것인가 먹힐 것인가

시간 제한	메모리 제한	solved.ac 티어
1초	256MB	실버 3

문제

심해에는 두 종류의 생명체 A와 B가 존재한다. A는 B를 먹는다. A는 자기보다 크기가 작은 먹이만 먹을 수 있다. 예를 들어, A의 크기가 {8, 1, 7, 3, 1}이고, B의 크기가 {3, 6, 1}인 경우에 A가 B를 먹을 수 있는 쌍의 개수는 7가지가 있다. 8-3, 8-6, 8-1, 7-3, 7-6, 7-1, 3-1.

두 생명체 A와 B의 크기가 주어졌을 때, A의 크기가 B보다 큰 쌍이 몇 개나 있는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스의 첫째 줄에는 A의 수 N과 B의 수 M이 주어진다. 둘째 줄에는 A의 크기가 모두 주어지며, 셋째 줄에는 B의 크기가 모두 주어진다. 크기는 양의 정수이다. (1 ≤ N, M ≤ 20,000) 

출력

각 테스트 케이스마다, A가 B보다 큰 쌍의 개수를 출력한다.

---

전형적인 이분탐색문제이다. 브루트포스로 접근하면 O(nm)의 시간복잡도를 가지고, n과 m이 모두 최대 20000이기 때문에 시간제한에 걸리게 된다.

 

C로 하드코딩한건 여길 참고하자(https://codejin.tistory.com/108). C++에는 <algorithm>헤더에 이분탐색에 관한 함수가 정의되어있기 때문에 그걸 사용하도록 하자.

 

#include <bits/stdc++.h>
#define endl '\n'
using namespace std;

int nA, nB;
vector<int> a, b;

void input() {
    cin >> nA >> nB;
    a.clear(); b.clear();
    a.resize(nA); b.resize(nB);
    for(int& x : a) cin >> x;
    for(int& x : b) cin >> x;
    sort(a.begin(), a.end());
    sort(b.begin(), b.end());
}

void sol() {
    int result = 0;
    for(int& x : a) {
        result += lower_bound(b.begin(), b.end(), x) - b.begin();
    }
    cout << result << endl;
}

int main () {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    int t;
    cin >> t;
    while(t--) {
        input();
        sol();
    }
}