Submission #73388199

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;

int main() {

    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    // ---------- INPUT ----------

    int N, K;
    cin >> N >> K;

    // labs[i] will store all themes of lab i
    vector<vector<string>> labs(N);

    for(int i = 0; i < N; i++) {

        int Mi;
        cin >> Mi;

        labs[i].resize(Mi);

        for(int j = 0; j < Mi; j++) {
            cin >> labs[i][j];
        }
    }

    // ---------- SOLUTION ----------

    long long answer = 0;

    for(int i=0;i<N-1;i++){
        for(int j=i+1;j<N;j++){
            int common=0;
            unordered_set<string> themes(labs[i].begin(), labs[i].end());
            for(auto &st: labs[j]){
                if(themes.count(st)){
                    common++;
                }
            }
            if(common>=K){
                answer++;
            }   
        }
    }

    // ---------- OUTPUT ----------

    cout << answer << '\n';

    return 0;
}

Submission Info

Judge Result