10-janghw0126

[janghw0126]

🔗 문제 링크

백준 | 줄 세우기

✔️ 소요된 시간

1시간

✨ 수도 코드

이 문제는 두 학생의 키를 비교하면서 학생들을 위상 정렬을 이용하여 줄 세우는 문제였습니다. 문제의 지문에 따라, 어떤 학생이 다른 학생보다 앞에 서야 한다는 조건을 만족시키면서 모든 학생을 나열해야 하므로 순서가 정해져 있는 작업을 차례대로 수행해야 되는 위상정렬을 사용해야겠다는 생각을 하였습니다.

📝 문제 해결 과정

1. 총 학생 수와 비교 횟수를 입력받고, 각 학생 간의 비교 관계를 저장할 그래프와 진입 차수를 저장할 배열을 만듭니다.
2. 먼저, 진입 차수가 0인 학생(앞에 서야 하는 학생이 없는 학생)을 큐에 추가합니다. -> 위상정렬 진행!
3. 큐에서 학생을 하나씩 꺼내며, 해당 학생의 진입 차수를 0으로 만들고 그 학생의 뒤에 서야 하는 학생들의 진입 차수를 하나씩 줄여줍니다.
4. 만약 어떤 학생의 진입 차수가 0이 되면, 그 학생을 큐에 넣어 줄을 세우는 과정을 반복합니다.

🚨 핵심 로직

이 문제는 싸이클이 없기 때문에 위에서 말했듯이 위상 정렬을 통해 학생들을 정렬할 수 있습니다. 큐가 비어갈 때까지, 진입 차수가 0인 학생들을 차례로 처리해주면 순서대로 줄을 세울 수 있으므로 이러한 로직을 바탕으로 문제를 풀었습니다.

👩‍💻 최종 코드

import sys
from collections import deque

# 학생 수(n)와 비교 횟수(m)를 입력받음
n, m = map(int, sys.stdin.readline().split())

# 그래프와 진입 차수 리스트를 초기화함
graph = [[] for _ in range(n + 1)]
deg = [0] * (n + 1)
q = deque()
res = []

# 비교 관계 입력과 그래프를 구성함
for _ in range(m):
    u, v = map(int, sys.stdin.readline().rstrip().split())
    # u가 v 앞에 서야 함
    graph[u].append(v)
    # v의 진입 차수를 증가시킴
    deg[v] += 1

# 진입 차수가 0인 노드 큐에 삽입함
for i in range(1, n + 1):
    if deg[i] == 0:
        q.append(i)

# 위상 정렬을 시작함
while q:
    # 진입 차수가 0인 노드를 큐에서 꺼냄
    cur = q.popleft()
    # 결과 리스트에 추가함
    res.append(cur) 

    # 현재 노드 뒤에 서야 하는 노드들의 진입 차수를 감소시킴
    for next_node in graph[cur]:
        deg[next_node] -= 1
        # 진입 차수가 0이 되면 큐에 추가함
        if deg[next_node] == 0:
            q.append(next_node)

# 결과 출력
print(*res)

📚 새롭게 알게된 내용

위상정렬을 마스터해보고자 요즘 이 유형 문제만 죽도록 파고 있는데 역시 쉽지 않은 유형이군여,,,,
열심히 아자잣 해보겠습니닷😅😂😂

[jung0115]

저도 혜원 님 풀이랑 비슷하게 풀었습니다! 위상정렬 마스터는 쉽지 않은 것 같아요 😅 저도 따라서 열심히 하겠습니다! 이번 차시도 수고하셨어요

import java.io.BufferedReader
import java.io.InputStreamReader
import java.util.StringTokenizer
import java.util.LinkedList
import java.util.Queue

fun main() {
  val br = BufferedReader(InputStreamReader(System.`in`))
  var st = StringTokenizer(br.readLine())

  val N = st.nextToken().toInt()
  val M = st.nextToken().toInt()

  var graph = Array(N+1, {mutableListOf<Int>()})
  var edgeCnt = Array(N+1, {0})

  for(i: Int in 0..M-1) {
    st = StringTokenizer(br.readLine())
    val A = st.nextToken().toInt()
    val B = st.nextToken().toInt()

    graph[A].add(B)
    edgeCnt[B]++ // 진입차수 기록
  }

  var queue = LinkedList<Int>() as Queue<Int>
  var answer = StringBuilder()

  // 진입차수 0인 값을 큐에 = 본인보다 앞에 서야하는 사람이 없다
  for(i: Int in 1..N) {
    if(edgeCnt[i] == 0)
      queue.offer(i);
  }

  while(!queue.isEmpty()) {
    var student = queue.poll()

    // 남은 학생 중에 본인보다 앞에 서야하는 사람이 없으므로 바로 줄 세워줌
    answer.append(student).append(" ")

    for(i: Int in 0..graph[student].size - 1) {
      // 현재 학생보다 뒤에 서야했던 학생 => 앞에 세워야 했던 사람이 한 명 줄어듦
      var back = graph[student][i]
      edgeCnt[back]--
      // 한 명 줄어들면서 더이상 앞에 서야하는 사람이 없으면 큐에 추가
      if(edgeCnt[back] == 0) {
        queue.offer(back)
      }
    }
  }

  print(answer)
}