[C++] 백준(BOJ) - 7795 : 먹을 것인가 먹힐 것인가

30 Mar 2024 in Develop on C++, Algorithm

인프런에 있는 큰돌님의 강의 10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트를 듣고 정리한 필기입니다.

• 문제
• 입력
• 출력
• 알고리즘
• 풀이
• 코드
• 평가

문제

BOJ - 7795 : 먹을 것인가 먹힐 것인가(링크)

심해에는 두 종류의 생명체 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보다 큰 쌍의 개수를 출력한다.

알고리즘

1. 정렬
2. 이분탐색
3. 투포인터

풀이

• 개수만을 카운팅 => 정렬
• x라는 값 미만의 값이 몇개인가?
  • upper_bound, lower_bound
  • 정렬된 상태일 때, 인덱스의 값이 x미만의 값의 개수와 같다

코드

// BOJ - 7795 : 먹을 것인가 먹힐 것인가
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

int t, n, m, res;
vector<int> v1, v2;

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

    cin >> t;
    while (t--)
    {
        res = 0;
        v1.clear();
        v2.clear();

        cin >> n >> m;

        for (int i = 0; i < n; i++)
        {
            int a;
            cin >> a;
            v1.push_back(a);
        }

        for (int i = 0; i < m; i++)
        {
            int a;
            cin >> a;
            v2.push_back(a);
        }

        sort(v1.begin(), v1.end());
        sort(v2.begin(), v2.end());

        for (int i = 0; i < v1.size(); i++)
        {
            res += (int)(lower_bound(v2.begin(), v2.end(), v1[i]) - v2.begin());
        }

        cout << res << '\n';
    }

    return 0;
}

평가

• upper_bound와 lower_bound를 사용하기 전에 정렬하기
• 여러 개의 테스트케이스를 진행할 땐, 자료구조를 꼭 초기화하기