본문 바로가기
Develop/Python + SWEA

[SW Expert Academy] 4871. 그래프 경로

by Tarra 2022. 2. 23.

4871. 그래프 경로


문제)

 

V개 이내의 노드를 E개의 간선으로 연결한 방향성 그래프에 대한 정보가 주어질 때, 특정한 두 개의 노드에 경로가 존재하는지 확인하는 프로그램을 만드시오.

두 개의 노드에 대해 경로가 있으면 1, 없으면 0을 출력한다.
 
예를 들어 다음과 같은 그래프에서 1에서 6으로 가는 경로를 찾는 경우, 경로가 존재하므로 1을 출력한다.

 

 

노드번호는 1번부터 존재하며, V개의 노드 중에는 간선으로 연결되지 않은 경우도 있을 수 있다.

 
 

[입력]
 

첫 줄에 테스트 케이스 개수 T가 주어진다.  1≤T≤50
 
다음 줄부터 테스트 케이스의 첫 줄에 V와 E가 주어진다. 5≤V≤50, 4≤E≤1000

테스트케이스의 둘째 줄부터 E개의 줄에 걸쳐, 출발 도착 노드로 간선 정보가 주어진다.

E개의 줄 이후에는 경로의 존재를 확인할 출발 노드 S와 도착노드 G가 주어진다.

 

[출력]

각 줄마다 "#T" (T는 테스트 케이스 번호)를 출력한 뒤, 답을 출력한다.

 

 

 

 

 

풀이)

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
def dfs(n, x):
    global visited
    stack = [n]
    while stack:
        n = stack.pop()
        if not visited[n]:
            visited[n] = 1
            # print(f"방문 노드:{n}, 방문체크: {visited}")
            if n == x and visited[n]:
                return 1
            for i in range(1, v + 1):
                if g[n][i] == 1 and visited[i] == 0:
                    stack.append(i)
    else:
        return 0
 
= int(input())
for _ in range(T):
    v, e = map(int, input().split())
    visited = [0* (v + 1)
 
    g = [[0* (v + 1for _ in range(v + 1)]
 
    for i in range(e):
        n, m = map(int, input().split())
        g[n][m] = 1
 
    s, x = map(int, input().split())
    print(f"#{_ + 1} {dfs(s, x)}")
cs

문제 출처 : https://swexpertacademy.com/main/main.do

 

SW Expert Academy

SW 프로그래밍 역량 강화에 도움이 되는 다양한 학습 컨텐츠를 확인하세요!

swexpertacademy.com

※ SW Expert 아카데미는 원칙적으로 문제를 무단 복제하는 것을 금지합니다.

학습용으로 문제를 가져왔으나, 문제가 될 시 수정 및 삭제하겠습니다.