BOJ 14572 : 스터디 그룹

• 문제 링크 : boj.kr/14572
• 난이도 : P5
• 태그 : 그리디, 투포인터, 정렬

---

코드

#include <bits/stdc++.h>

#define all(x) (x).begin(), (x).end()

#define INF 0x7FFFFFFF

using namespace std;

using ll = long long;
using ld = long double;
using pii = pair<int,int>;
using pll = pair<ll, ll>;

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);

    int n, k, d;
    cin >> n >> k >> d;

    vector<pair<int, vector<int>>> arr(n);
    for(int i = 0; i < n; i++) {
        int m;
        cin >> m >> arr[i].first;
        arr[i].second.resize(m);
        for(auto &u : arr[i].second) cin >> u;
    }
    sort(all(arr));

    vector<int> cnt(k+1);
    int l = 0;
    int ans = 0;
    for(int r = 0; r < n; r++) {
        for(auto u : arr[r].second) cnt[u]++;
        while(arr[r].first - arr[l].first > d) {
            for(auto u : arr[l].second) cnt[u]--;
            l++;
        }
        int a = 0, b = 0;
        int s = r - l + 1;
        for(int i = 1; i <= k; i++) {
            if(cnt[i]) a++;
            if(cnt[i] == s) b++;
        }
        ans = max(ans, (a-b)*s);
    }
    cout << ans;

    return 0;
}

---

풀이

실력 차이를 만족하는 구간이 있다면 이 구간 안의 모든 학생이 그룹에 포함되는 것이 최적이다.

(그룹 내의 학생들이 아는 모든 알고리즘의 수 - 그룹 내의 모든 학생들이 아는 알고리즘의 수)는 그룹원의 수가 늘어날 때마다 같거나 커진다.