링크 : https://www.acmicpc.net/problem/10451
순열 사이클
문제
1부터 N까지 정수 N개로 이루어진 순열을 나타내는 방법은 여러 가지가 있다. 예를 들어, 8개의 수로 이루어진 순열 (3, 2, 7, 8, 1, 4, 5, 6)을 배열을 이용해 표현하면 \(\begin{pmatrix} 1 & 2 &3&4&5&6&7&8 \\ 3& 2&7&8&1&4&5&6 \end{pmatrix}\) 와 같다. 또는, Figure 1과 같이 방향 그래프로 나타낼 수도 있다.
순열을 배열을 이용해 \(\begin{pmatrix} 1 & \dots & i & \dots &n \\ \pi_1& \dots& \pi_i & \dots & \pi_n \end{pmatrix}\) 로 나타냈다면, i에서 πi로 간선을 이어 그래프로 만들 수 있다.
Figure 1에 나와있는 것 처럼, 순열 그래프 (3, 2, 7, 8, 1, 4, 5, 6) 에는 총 3개의 사이클이 있다. 이러한 사이클을 "순열 사이클" 이라고 한다.
N개의 정수로 이루어진 순열이 주어졌을 때, 순열 사이클의 개수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 테스트 케이스의 개수 T가 주어진다. 각 테스트 케이스의 첫째 줄에는 순열의 크기 N (2 ≤ N ≤ 1,000)이 주어진다. 둘째 줄에는 순열이 주어지며, 각 정수는 공백으로 구분되어 있다.
출력
각 테스트 케이스마다, 입력으로 주어진 순열에 존재하는 순열 사이클의 개수를 출력한다.
풀이
n = int(input())
# 바로 바꾼 버젼
for _ in range(n):
t = int(input())
adj_lst = [0] + list(map(int,input().strip().split()))
count = 0
for i in range(1,t+1):
# 순환하는 대로 가고 싶으니 현재 값을 지정함
now_num = i
if adj_lst[i]: # 값이 0이라면 순환하면 안 됨
while True: # 순환 함
next_num = adj_lst[now_num] # 다음 값을 가져오고
adj_lst[now_num] = 0 # 현재 값은 0이 됨
if next_num == 0: # 만약 다음값이 0 이라면 순환 끝 종료
count += 1
break
# 만약 순환 종료 되지 않으면 다음 값이 현재 값이 되고 순환함
now_num = next_num
print(count)n = int(input())
# visited를 기록하는 버젼
# 이게 더 빠르다
for _ in range(n):
t = int(input())
adj_lst = [0] + list(map(int,input().strip().split()))
visited = [0] * (t+1)
count = 0
for i in range(1,t+1):
# 순환하는 대로 가고 싶으니 현재 값을 지정함
now_num = i
if visited[i] == 0:
while True: # 방문하지 않았다면 순환함
next_num = adj_lst[now_num] # 방문할 장소 기록
visited[now_num] = 1 # 현재는 방문한 걸로 기록
if visited[next_num] == 1:
count += 1
break
# 만약 순환 종료 되지 않으면 다음 값이 현재 값이 되고 순환함
now_num = next_num
print(count)'Python > BaekJoon' 카테고리의 다른 글
| [python] 백준 알고리즘 32942번 : 그래프와 그래프 - 인접 행렬, 인접 리스트 (0) | 2025.02.19 |
|---|---|
| [python] 백준 알고리즘 11279번 : 최대 힙 - heapq 모듈 사용하기 (0) | 2025.02.17 |
| [python] 백준 알고리즘 1417번: 국회의원 선거 - 그리디 알고리즘 (0) | 2025.02.16 |
| [python] 백준 알고리즘 2537번: k진수 정수의 자릿수 나누기 - python에서 n진법으로 숫자 바꾸기 (0) | 2025.01.31 |
