Skip to content

Commit 70ebaf1

Browse files
committed
add course schedule solution
1 parent 939986b commit 70ebaf1

File tree

1 file changed

+97
-0
lines changed

1 file changed

+97
-0
lines changed

course-schedule/Tessa1217.java

Lines changed: 97 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,97 @@
1+
import java.util.ArrayList;
2+
import java.util.LinkedList;
3+
import java.util.List;
4+
import java.util.Queue;
5+
6+
class Solution {
7+
public boolean canFinish(int numCourses, int[][] prerequisites) {
8+
// return BFS(numCourses, prerequisites);
9+
return DFS(numCourses, prerequisites);
10+
}
11+
12+
// BFS 방식으로 풀이
13+
private boolean BFS(int numCourses, int[][] prerequisites) {
14+
15+
// 연결 그래프 생성
16+
List<List<Integer>> graph = new ArrayList<>();
17+
for (int i = 0; i < numCourses; i++) {
18+
graph.add(new ArrayList<>());
19+
}
20+
21+
// 들어오는 간선의 수 배열
22+
int[] inDegree = new int[numCourses];
23+
24+
for (int[] pre : prerequisites) {
25+
int course = pre[0];
26+
int prereq = pre[1];
27+
inDegree[course]++;
28+
graph.get(prereq).add(course);
29+
}
30+
31+
// 들어오는 간선이 없는 과목 Queue에 삽입
32+
Queue<Integer> q = new LinkedList<>();
33+
for (int i = 0; i < inDegree.length; i++) {
34+
if (inDegree[i] == 0) {
35+
q.offer(i);
36+
}
37+
}
38+
39+
// 들은 과목 수
40+
int takeCourse = 0;
41+
while (!q.isEmpty()) {
42+
int course = q.poll();
43+
takeCourse++;
44+
45+
for (int connect : graph.get(course)) {
46+
inDegree[connect]--;
47+
if (inDegree[connect] == 0) {
48+
q.offer(connect);
49+
}
50+
}
51+
}
52+
53+
return takeCourse == numCourses;
54+
}
55+
56+
// DFS 방식 풀이
57+
private boolean DFS(int numCourses, int[][] prerequisites) {
58+
List<List<Integer>> graph = new ArrayList<>();
59+
for (int i = 0; i < numCourses; i++) {
60+
graph.add(new ArrayList<>());
61+
}
62+
63+
for (int[] pre : prerequisites) {
64+
int course = pre[0];
65+
int prereq = pre[1];
66+
graph.get(prereq).add(course);
67+
}
68+
69+
// 처리 상태 배열
70+
int[] state = new int[numCourses];
71+
72+
for (int i = 0; i < numCourses; i++) {
73+
if (hasCycle(graph, state, i)) {
74+
return false;
75+
}
76+
}
77+
78+
return true;
79+
}
80+
81+
// 사이클 여부 확인
82+
private boolean hasCycle(List<List<Integer>> graph, int[] state, int course) {
83+
84+
if (state[course] == 1) return true; // 사이클이 있다면
85+
if (state[course] == 2) return false; // 이미 처리 완료되었으므로 return
86+
87+
state[course] = 1;
88+
for (int connect : graph.get(course)) {
89+
if (hasCycle(graph, state, connect)) {
90+
return true;
91+
}
92+
}
93+
state[course] = 2; // 처리 완료
94+
return false;
95+
}
96+
}
97+

0 commit comments

Comments
 (0)