Merge pull request #43 from bsa0322/sua_dfs&bfs_assign

ZERO-black · web-flow · commit 67d027855230 · 2022-05-01T01:57:06.000+09:00
[DFS &amp; BFS] 04월 15일 (선택.cpp)
diff --git a/04월 15일 - DFS & BFS/과제/10026.cpp b/04월 15일 - DFS & BFS/과제/10026.cpp
@@ -0,0 +1,91 @@
+#include <iostream>
+#include <vector>
+#include <queue>
+
+using namespace std;
+typedef pair<int, int> ci;
+
+void greenToRed(int n, vector<vector<char>> &board) {
+    for (int i = 0; i < n; i++) {
+        for (int j = 0; j < n; j++) {
+            if (board[i][j] == 'G') {
+                board[i][j] = 'R';
+            }
+        }
+    }
+}
+
+void bfs(int n, int r, int c, vector<vector<bool>> &visited, vector<vector<char>> &board) {
+    //상, 하, 좌, 우
+    int dr[4] = {-1, 1, 0, 0};
+    int dc[4] = {0, 0, -1, 1};
+
+    queue<ci> q;
+
+    //큐 초기화
+    q.push({r, c});
+    visited[r][c] = true;
+    while (!q.empty()) {
+        int cr = q.front().first;
+        int cc = q.front().second;
+        q.pop();
+
+        for (int i = 0; i < 4; i++) {
+            int nr = cr + dr[i];
+            int nc = cc + dc[i];
+            if (nr < 0 || nr >= n || nc < 0 || nc >= n || visited[nr][nc]) {
+                continue;
+            }
+            if (board[nr][nc] != board[cr][cc]) {
+                continue;
+            }
+            q.push({nr, nc});
+            visited[nr][nc] = true;
+        }
+    }
+}
+
+int countArea(int n, vector<vector<char>> &board) {
+    vector<vector<bool>> visited(n, vector<bool>(n, false));
+    int area = 0;
+
+    for (int i = 0; i < n; i++) {
+        for (int j = 0; j < n; j++) {
+            if (visited[i][j]) {
+                continue;
+            }
+            bfs(n, i, j, visited, board);
+            area++;
+        }
+    }
+    return area;
+}
+
+/**
+ * [적록색약]
+ *
+ * 그림의 색을 기준으로 구역을 나누는 문제
+ * 1. 적록색약이 아닌 사람 기준으로 구역을 나눈다.
+ * 2. 그림의 초록을 모두 빨강으로 바꾼 후, 적록색약인 사람 기준으로 구역을 나눈다.
+ */
+
+int main() {
+    int n;
+    string s;
+
+    //입력
+    cin >> n;
+    vector<vector<char>> board(n, vector<char>(n, ' '));
+    for (int i = 0; i < n; i++) {
+        cin >> s;
+        for (int j = 0; j < n; j++) {
+            board[i][j] = s[j];
+        }
+    }
+
+    //연산 & 출력
+    cout << countArea(n, board) << ' ';
+    greenToRed(n, board);
+    cout << countArea(n, board);
+    return 0;
+}
diff --git a/04월 15일 - DFS & BFS/과제/11725.cpp b/04월 15일 - DFS & BFS/과제/11725.cpp
@@ -0,0 +1,47 @@
+#include <iostream>
+#include <vector>
+
+using namespace std;
+
+vector<vector<int>> adj_list;
+vector<int> parent;
+
+void dfs_recursion(int prev, int curr) {
+    if (parent[curr]) {
+        return;
+    }
+    parent[curr] = prev;
+    for (int i = 0; i < adj_list[curr].size(); i++) {
+        dfs_recursion(curr, adj_list[curr][i]);
+    }
+}
+
+/**
+ * [트리의 부모 찾기]
+ *
+ * 1번 노드에서 출발해서 탐색
+ * 루트 노드에서 출발 했기 때문에, 현재 노드의 부모는 이전 노드가 된다.
+ */
+
+int main() {
+    int n, a, b;
+
+    //입력
+    cin >> n;
+    adj_list.assign(n + 1, vector<int>(0));
+    parent.assign(n + 1, 0);
+    for (int i = 1; i < n; i++) {
+        cin >> a >> b;
+        adj_list[a].push_back(b);
+        adj_list[b].push_back(a);
+    }
+
+    //연산
+    dfs_recursion(1, 1); //1번 노드는 루트노드이므로, 부모를 자기 자신으로 설정
+
+    //출력
+    for (int i = 2; i <= n; i++) {
+        cout << parent[i] << '\n';
+    }
+    return 0;
+}
diff --git a/04월 15일 - DFS & BFS/과제/1697.cpp b/04월 15일 - DFS & BFS/과제/1697.cpp
@@ -0,0 +1,50 @@
+#include <iostream>
+#include <queue>
+
+using namespace std;
+const int SIZE = 1e5;
+
+int bfs(int n, int k) {
+    vector<int> check(SIZE + 1, 0);
+    queue<int> q; //큐에 좌표값 저장
+    int ans = 0;
+
+    check[n] = 1; //시작 노드 방문체크 + 시간 초기화
+    q.push(n);
+    while (!q.empty()) {
+        int x = q.front();
+        q.pop();
+
+        if (x == k) {
+            ans = check[x] - 1;
+            break;
+        }
+
+        vector<int> child = {x - 1, x + 1, x * 2}; //자식 노드
+        for (int i = 0; i < 3; i++) {
+            if (child[i] >= 0 && child[i] <= SIZE && !check[child[i]]) {
+                check[child[i]] = check[x] + 1;
+                q.push(child[i]);
+            }
+        }
+    }
+    return ans;
+}
+
+/**
+ * [숨바꼭질]
+ *
+ * x좌표 위에서 x-1, x+1, 2*x의 위치로 계속 이동하며 탐색
+ * 가장 빠른 시간을 찾아야 하므로 주변 노드부터 탐색하는 풀이로 풀어서 k에 도달하면 바로 탐색 종료 (bfs)
+ */
+
+int main() {
+    int n, k;
+
+    //입력
+    cin >> n >> k;
+
+    //연산 & 출력
+    cout << bfs(n, k);
+    return 0;
+}
diff --git a/04월 15일 - DFS & BFS/과제/19538.cpp b/04월 15일 - DFS & BFS/과제/19538.cpp
@@ -0,0 +1,85 @@
+#include <iostream>
+#include <vector>
+#include <queue>
+
+using namespace std;
+
+vector<int> bfs(vector<vector<int>> &people, vector<int> &init, int n) {
+    vector<int> rest(n + 1, 0); //각 사람이 루머를 믿기 위해 필요한 최소 주변인 수
+    vector<int> ans(n + 1, -1); //몇 분 후에 믿는지
+    queue<int> q;
+
+    //시작 정점 초기화
+    for (int i = 0; i < init.size(); i++) {
+        ans[init[i]] = 0;
+        q.push(init[i]);
+    }
+
+    //루머 믿어야 하는 주변인 수 초기화
+    for (int i = 1; i <= n; i++) {
+        rest[i] = (people[i].size() + 1) / 2;
+    }
+
+    while (!q.empty()) {
+        int curr = q.front(); //현재 사람
+        int t = ans[curr]; //루머 믿은 시간
+        q.pop();
+
+        for (int i = 0; i < people[curr].size(); i++) {
+            int next = people[curr][i];
+            if (ans[next] > -1) { //이미 루머를 믿는 다면
+                continue;
+            }
+            rest[next]--;
+            if (!rest[next]) { //주변인들 중 절반 이상이 루머를 믿으면
+                ans[next] = t + 1;
+                q.push(next);
+            }
+        }
+    }
+    return ans;
+}
+
+/**
+ * [루머]
+ *
+ * 주변인들이 얼마나 믿는지를 배열을 통해 관리해야 함
+ * 방문 체크 배열을 루머를 믿는 시간을 저장하는 배열로 대체함
+ *    ans[i] = -1 일 경우, i는 루머를 믿지 않는다
+ *    ans[i] = t인 경우, i는 t분 부터 루머를 믿기 시작함
+ * 각자가 루머를 믿기 위해 주변인의 절반 이상이 루머를 믿어야 하므로, 각 사람이 루머를 믿기까지 루머를 믿는 주변인 몇명이 남았는지를 리스트에 기록한다.
+ * 남은 사람이 0인 순간, 해당 사람은 루머를 믿기 시작
+ */
+
+int main() {
+    int n, m, input;
+
+    //루머 퍼뜨리는 관계 입력
+    cin >> n;
+    vector<vector<int>> people(n + 1, vector<int>());
+    for (int i = 1; i <= n; i++) {
+        while (true) {
+            cin >> input;
+            if (!input) {
+                break;
+            }
+            people[i].push_back(input);
+        }
+    }
+
+    //최초 유포자 입력
+    cin >> m;
+    vector<int> init(m, 0);
+    for (int i = 0; i < m; i++) {
+        cin >> init[i];
+    }
+
+    //연산
+    vector<int> ans = bfs(people, init, n);
+
+    //출력
+    for (int i = 1; i <= n; i++) {
+        cout << ans[i] << ' ';
+    }
+    return 0;
+}
diff --git a/04월 15일 - DFS & BFS/과제/2667_bfs.cpp b/04월 15일 - DFS & BFS/과제/2667_bfs.cpp
@@ -0,0 +1,86 @@
+#include <iostream>
+#include <vector>
+#include <queue>
+#include <algorithm>
+
+using namespace std;
+typedef pair<int, int> ci;
+
+//단지 내 집 개수 탐색하는 bfs
+int bfs(int n, int r, int c, vector<vector<bool>> &board) {
+    int dr[4] = {-1, 1, 0, 0};
+    int dc[4] = {0, 0, -1, 1};
+
+    queue<ci> q;
+
+    //큐 초기화
+    q.push({r, c});
+    board[r][c] = false;
+    int cnt = 1;
+    while (!q.empty()) {
+        int cr = q.front().first;
+        int cc = q.front().second;
+        q.pop();
+
+        for (int i = 0; i < 4; i++) {
+            int nr = cr + dr[i];
+            int nc = cc + dc[i];
+            if (nr < 0 || nr >= n || nc < 0 || nc >= n || !board[nr][nc]) {
+                continue;
+            }
+            q.push({nr, nc});
+            board[nr][nc] = false;
+            cnt++;
+        }
+    }
+    return cnt;
+}
+
+//단지 수와 단지 내 집 개수 구하는 함수
+vector<int> cntComplex(int n, vector<vector<bool>> &board) {
+    vector<int> complex;
+    for (int i = 0; i < n; i++) {
+        for (int j = 0; j < n; j++) {
+            if (board[i][j]) {
+                int cnt = bfs(n, i, j, board);
+                complex.push_back(cnt);
+            }
+        }
+    }
+    return complex;
+}
+
+/**
+ * [단지 번호 붙이기] - bfs
+ *
+ * 단순 탐색
+ * (주의) 단지 내 집의 수를 "오름차순"으로 출력
+ *
+ * 이 풀이에서는 방문체크 배열을 따로 사용하지 않고, 처음 지도에 표기를 1 -> 0으로 바꾸어 중복으로 탐색하지 않도록 함
+ */
+
+int main() {
+    int n;
+    string s;
+
+    //입력
+    cin >> n;
+    vector<vector<bool>> board(n, vector<bool>(n, false));
+    for (int i = 0; i < n; i++) {
+        cin >> s;
+        for (int j = 0; j < n; j++) {
+            board[i][j] = s[j] - '0';
+        }
+    }
+
+    //연산
+    vector<int> ans = cntComplex(n, board);
+    sort(ans.begin(), ans.end());
+
+    //출력
+    cout << ans.size() << '\n';
+    for (int i = 0; i < ans.size(); i++) {
+        cout << ans[i] << '\n';
+    }
+    return 0;
+}
diff --git a/04월 15일 - DFS & BFS/과제/2667_dfs.cpp b/04월 15일 - DFS & BFS/과제/2667_dfs.cpp
