알고리즘 문제 풀이/삼성 SW Expert Academy
SWEA 2814: 최장 경로 (C++)
Hugxy
2019. 9. 25. 01:20
모든 정점을 시작점으로 dfs를 돌려주면 된다.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | #include<iostream> using namespace std; int grp[11][11]; //1-indexed bool vis[11]; //방문처리 int n, m, Max = 1; void dfs(int len, int cur) { vis[cur] = true; if (Max < len) Max = len; for (int i = 1; i <= n; i++) { if (grp[cur][i] == 1 && !vis[i]) dfs(len + 1, i); } vis[cur] = false; } void initGrp() { for (int i = 1; i <= n; i++) for (int j = 1; j <= n; j++) grp[i][j] = 0; } int main(void) { int T; cin >> T; for (int tc = 1; tc <= T; tc++) { cin >> n >> m; for (int i = 0; i < m; i++) { int src, dst; cin >> src >> dst; grp[src][dst] = 1; grp[dst][src] = 1; } for (int i = 1; i <= n; i++) dfs(1, i); // i번 정점에서 길이 printf("#%d %d\n", tc, Max); Max = 1; initGrp(); } return 0; } | cs |
방문 처리를 조금 더 직관적으로 하면 다음과 같이 구현할 수 있다.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | #include<iostream> using namespace std; int grp[11][11]; //1-indexed bool vis[11]; //방문처리 int n, m, Max = 1; void dfs(int len, int cur) { //vis[cur] = true; if (Max < len) Max = len; for (int i = 1; i <= n; i++) { if (grp[cur][i] == 1 && !vis[i]) { vis[i] = true; dfs(len + 1, i); vis[i] = false; } } //vis[cur] = false; //확인이 끝난 점은 방문을 해제 } void initGrp() { for (int i = 1; i <= n; i++) for (int j = 1; j <= n; j++) grp[i][j] = 0; } int main(void) { int T; cin >> T; for (int tc = 1; tc <= T; tc++) { cin >> n >> m; for (int i = 0; i < m; i++) { int src, dst; cin >> src >> dst; grp[src][dst] = 1; grp[dst][src] = 1; } for (int i = 1; i <= n; i++) { vis[i] = true; dfs(1, i); // i번 정점에서 길이 vis[i] = false; } printf("#%d %d\n", tc, Max); Max = 1; initGrp(); } return 0; } | cs |