[BOJ][C++] 백준 7795번: 먹을 것인가 먹힐 것인가

https://www.acmicpc.net/problem/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보다 큰 쌍의 개수를 출력한다.

 

풀이

투포인터를 이용하자

 

a와 b에 각각의 인덱스를 지정하고

조건에 맞는 쌍을 카운트할 수 있도록 각 인덱스를 줄여나가면 된다.

글보다는 코드로 보는게 이해가 쉽다

// 풀이 : https://whkakrkr.tistory.com

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

int main() {
    int t;
    cin >> t;
    while(t--) {
        // 입력
        int n,m;
        cin >> n >> m;
        vector<int>a(n);
        vector<int>b(m);
        for(int i=0; i<n; i++) {
            cin >> a[i];
        }
        for(int i=0; i<m; i++) {
            cin >> b[i];
        }
        
        // 풀이
        sort(a.begin(), a.end());
        sort(b.begin(), b.end());
        int indexA = n-1;
        int indexB = m-1;
        int ans = 0;
        while(indexA>=0 && indexB>=0) {
            while(a[indexA] <= b[indexB]) {
                indexB--;
            }
            ans += indexB+1;
            indexA--;
        }
        
        // 출력
        cout << ans << "\n";
    }
    
    return 0;
}