알고리즘/위상정렬
BOJ - 1766 ) 문제집
개발자가될수있을까?
2020. 11. 8. 14:09


1766번: 문제집
첫째 줄에 문제의 수 N(1 ≤ N ≤ 32,000)과 먼저 푸는 것이 좋은 문제에 대한 정보의 개수 M(1 ≤ M ≤ 100,000)이 주어진다. 둘째 줄부터 M개의 줄에 걸쳐 두 정수의 순서쌍 A,B가 빈칸을 사이에 두고 주
www.acmicpc.net
문제집에서 순서가 존재하기때문에 차수를 고려한 위상정렬을 진행해준다.
다만, 문제 번호가 1-N 까지 문제중에서 차수가 0 이되었을때에 우선순위를 부여해야한다 ( 쉬운문제 부터 푼다 )
따라서, 우선순위큐에 문제번호가 낮은 차수가 0 인 노드부터 처리해주자
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.StringTokenizer;
public class 문제집 {
public static void main(String[] args) throws Exception {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
int n = Integer.parseInt(st.nextToken());
int m = Integer.parseInt(st.nextToken());
ArrayList<Integer> node[] = new ArrayList[n+1];
for(int i= 1 ; i<= n ; i++) {
node[i] = new ArrayList<>();
}
int degree[] = new int[n+1];
for(int i =0; i < m ; i ++) {
st=new StringTokenizer(br.readLine());
int s = Integer.parseInt(st.nextToken());
int e = Integer.parseInt(st.nextToken());
node[s].add(e);
degree[e] +=1;
}
PriorityQueue<Integer> pq = new PriorityQueue<>(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o1-o2;
}
});
for(int i=1 ; i <=n;i++) {
if(degree[i]==0) {
pq.add(i);
}
}
StringBuilder sb = new StringBuilder();
while(!pq.isEmpty()) {
int now = pq.poll();
sb.append(now +" ");
int size = node[now].size();
for(int i=0 ;i<size; i++) {
int next = node[now].get(i);
degree[next]--;
if(degree[next]==0) {
pq.add(next);
}
}
}
System.out.println(sb);
}
}
