[3018] 캠프파이어

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

3018번: 캠프파이어

<문제>

입력이 작으므로, 효율성보다는 정확하게 구현하면 된다.

bool [n][idx]를, n번째 사람이 idx번째 노래를 아는 경우에 true를 갖도록 해주었고

캠프파이어에 참여한 인원은 vector<int>v에 입력받았다.

이후에 선영이가 존재하는 경우에는 v를 한번만 순회하며 새로운 노래를 아는 상태로 만들고

그렇지 않은 경우에는 v를 2중으로 순회하고, 현재까지 아는 노래만큼 추가로 순회하며

총 3중for로 노래를 공유하는 부분을 구현해주었다.

 

<소스코드>

#include <bits/stdc++.h>
using namespace std;
bool chk[101][51];
int n, e, idx = 1;
int main(void) {
    cin >> n >> e;
    int i, j;
    for (i = 0; i < e; i++) {
        int k;
        cin >> k;
        vector<int> v;
        bool flag = false;
        for (j = 0; j < k; j++) {
            int x;
            cin >> x;
            if (x == 1) flag = true;
            v.push_back(x);
        }
        if (flag == true) {
            for (auto& i : v) {
                chk[i][idx] = true;
            }
            idx++;
        } else {
            for (auto& i : v)
                for (auto& j : v) {
                    for (int X = 1; X < idx; X++) {
                        if (chk[j][X] == true) chk[i][X] = true;
                    }
                }
        }
    }
    for (i = 1; i <= n; i++) {
        bool flag = true;
        for (j = 1; j < idx; j++) {
            if (chk[i][j] == false) {
                flag = false;
                break;
            }
        }
        if (flag == true) cout << i << '\n';
    }
    return 0;
}