10-g0rnn

[g0rnn]

🔗 문제 링크

결혼식

✔️ 소요된 시간

30m

✨ 수도 코드

간단한 그래프 문제 가져왔습니다. 문제에서 결혼식엔 친구와 친구의 친구까지만 초대한다고 했으므로 해당하는 두 종류의 사람들만 세면 됩니다.
친구는 상근이와 직접적인 친구이고 친구의 친구는 상근이의 친구와 연관이 있는 사람입니다.

1. 입력 받으며 상근이의 친구일 경우 친구 세기
2. 상근이의 친구 중 상근이의 친구가 아닌 친구 세기

중복을 포함하지 않고 검색하는데 빠른 unordered_set을 사용하였습니다. 그리 어려운 문제는 아니라 쉽게 풀립니다 ;)

(+추가)
이 문제를 그냥 2차원 배열에 저장하여 dfs나 bfs로도 풀 수 있습니다. 그래프와 bfs, dfs를 한번에 익히기 좋을 것 같아요

📚 새롭게 알게된 내용

unordered_map을 좀 더 알아봐야겠습니다.. 그새 또 까먹었어요

[kangrae-jo]

그래프는 오랜만이네요..
제 코드와 균호님 코드와 비교했을때 거의 비슷합니다

저도 중복이 허용되지않는 set과 unordered_set이 떠올랐고,
순서가 필요하지 않으니 unordered_set을 선택해서 구현했습니다.
(처음에는 그냥 vector 로 invited를 설정했는데 메모리초과...)

차이점이라면 그래프를 구현한 방식인데 저는 2차원 배열처럼 구현을 했고,
균호님은 vector 안에 set을 넣으셨군요

c++에서 vector또는 list로 그래프로 구현을 하는걸로 알고있는데
set을 사용하는 특별한 이유가 있었는지 궁금합니다!!

CPP CODE

#include <iostream>
#include <vector>
#include <unordered_set>

using namespace std;

int main() {
    int n, m, cnt = 0;
    cin >> n >> m;

    vector<vector<int>> graph(n + 1);
    for (int i = 0; i < m; i++) {
        int a, b;
        cin >> a >> b;
        graph[a].push_back(b);
        graph[b].push_back(a);
    }

    unordered_set<int> invited;
    invited.insert(1);
    for (int myFriend : graph[1]) {
        invited.insert(myFriend);       
        for (int yourFriend : graph[myFriend]) {
            invited.insert(yourFriend);
        }
    }

    cout << invited.size() - 1 << endl;

    return 0;
}

[g0rnn]

그래프는 오랜만이네요.. 제 코드와 균호님 코드와 비교했을때 거의 비슷합니다

저도 중복이 허용되지않는 set과 unordered_set이 떠올랐고, 순서가 필요하지 않으니 unordered_set을 선택해서 구현했습니다. (처음에는 그냥 vector 로 invited를 설정했는데 메모리초과...)

차이점이라면 그래프를 구현한 방식인데 저는 2차원 배열처럼 구현을 했고, 균호님은 vector 안에 set을 넣으셨군요

c++에서 vector또는 list로 그래프로 구현을 하는걸로 알고있는데
set을 사용하는 특별한 이유가 있었는지 궁금합니다!!

처음 문제를 읽고 친구리스트를 검색하는 기능이 필요할 줄 알아서 unorderd_set으로 했습니다. 이제 다시 보니 해당 부분은 필요없어서 vector로 하는게 더 좋아보이네요 😅

[kokeunho]

그래프와 bfs에 대해서 공부할 수 있는 문제였습니다.
이차원 리스트로 그래프를 구현하긴 했는데
그새 bfs를 까먹어서 검색 찬스 사용했습니다..
bfs, dfs 문제를 다시 풀어봐야겠습니다.
좋은 문제 감사합니다 수고하셨습니다!

[wnsmir]

최근에 풀어본 그래프문제 모두 n*m의 틀이 주어지는 문제였어서 인접리스트로 푸는 그래프문제는 오랜만이라 시간이 조금 더 걸렸습니다.

파이썬에는 defaultdict라는 딕셔너리의 서브클래스가 존재합니다.
키가 존재하지 않을 때 기본값을 자동으로 설정해주어 키에러를 방지하는 라이브러리 입니다.

예를들어 빈 딕셔너리를 생성했을때

d = {}
d['a']를 하면 키에러가 발생하지만, defaultdict를 사용하면 기본값 0이 출력되게 됩니다.

따라서 기본값을 미리 주지 않고 만들면서 추가해나갈때 유용한 딕셔너리라 그래프문제를 풀때 저는 상상 Import해두고 구상을 하는 편입니다.

그래프를 먼저 defaultdict로 리스트형식으로 만들어줍니다.

이후 m번 반복되어 나오는 인접리스트관계를 그래프 append를 통해 양측에 서로 넣어줍니다.(무방향)

이렇게 노드가 완성되었다면, 친구의 친구 즉 2단계까지만 초대가 허용되기때문에 한레벨에 한층식 즉
BFS를 활용해줍니다.

BFS함수를 정의해주고 시작할 노드 즉 1번을 큐에 추가해줍니다.
중복방지를 위해 방문한 노드를 visited에 기록해주고 결과 즉 초대할 친구를 invited에 추가해줍니다.
visited와 invited에 set을 이용하는 이유도 중복을 방지하기 위함입니다.

BFS를 한번 돌떼마다 레벨이 하나 증가하기 때문에 첫 level은 0으로 두고 while문으로 bfs를 돌기 시작합니다.

큐에 원소가 남아있고, 레벨이 2 미만 즉 친구까지라면 친구의 친구까지 가능하므로 계속 진행시킵니다.
큐에 남아있는 원소만큼 for문을 돌려주고, 큐에서 젤 먼저 나온 원소를 현재큐로 잡은 뒤 현재큐의 이웃드를 방문하지 않은 노드라면 방문에 추가하고, 초대할 친구set에 추가해주고, 큐에 그 친구를 추가해줍니다.

한 레벨이 또 종료했다면 레벨을 하나 높여주고 다음 while문으로 넘어갑니다

from collections import defaultdict, deque
graph = defaultdict(list)

n = int(input())
m = int(input())

for _ in range(m):
    a, b = map(int, input().split())
    graph[a].append(b)
    graph[b].append(a)

def bfs(start, graph):
    queue = deque([start])
    visited = set([start])
    invited = set()
    level = 0

    while queue and level < 2:
        for _ in range(len(queue)):
            current = queue.popleft()
            for neighbor in graph[current]:
                if neighbor not in visited:
                    visited.add(neighbor)
                    invited.add(neighbor)
                    queue.append(neighbor)
        level += 1
    return invited

invited_friends = bfs(1, graph)

print(len(invited_friends))