공부공간

BOJ - 1766 ) 문제집 본문

알고리즘/위상정렬

BOJ - 1766 ) 문제집

개발자가될수있을까? 2020. 11. 8. 14:09

 


www.acmicpc.net/problem/1766

 

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);
		
	}

}

 

'알고리즘 > 위상정렬' 카테고리의 다른 글

BOJ - 2056 ) 작업  (0) 2021.09.04
BOJ - 1005 ) ACM Craft  (0) 2020.05.11
BOJ - 2623 ) 음악프로그램  (2) 2020.03.24
BOJ - 2252 ) 줄 세우기  (0) 2020.03.23
Comments