题目:
思路:
如果直接计算会超时,因此需要建立邻接矩阵,将计算好的结果存在邻接矩阵中。
步骤如下:
1.建立邻接矩阵,将所有课程的前置课程放在动态数组中。
2. 建立条件判断矩阵,将已经访问的节点置为true,从而减少遍历时间。
3. 深度遍历,先判断节点是否访问,再进行下一步。
4. 对每个访问的节点判断其邻接矩阵中的点,进行深度遍历,同时将访问的节点与前置节点直接的条件置为true,同时判断间接的先决条件。
代码:
class Solution {
public List<Boolean> checkIfPrerequisite(int numCourses, int[][] prerequisites, int[][] queries) {
ArrayList<ArrayList<Integer>> arrayList = new ArrayList<>();
for (int i = 0; i < numCourses; i++) {
arrayList.add(new ArrayList<>());
}
for (int i = 0; i < prerequisites.length; i++) {
int before = prerequisites[i][0];
int behind = prerequisites[i][1];
arrayList.get(before).add(behind);
}
//建立邻接矩阵判断所有点之间关系
boolean[] visited = new boolean[numCourses];
boolean[][] isPre = new boolean[numCourses][numCourses];
for (int i = 0; i < numCourses; i++) {
dfs(visited,isPre,i,arrayList);
}
List<Boolean> result = new ArrayList<>();
for(int[] q:queries){
result.add(isPre[q[0]][q[1]]);
}
return result;
}
public void dfs(boolean[] visited,boolean[][] isPre,int i,ArrayList<ArrayList<Integer>> arrayList){
if(visited[i]){
return;
}
visited[i] = true;
ArrayList<Integer> arrayList1 = arrayList.get(i);
for(int j=0;j<arrayList1.size();j++){
int behind = arrayList1.get(j);
dfs(visited,isPre,behind,arrayList);
isPre[i][behind] = true;
for(int k=0;k<isPre.length;k++){
isPre[i][k] = isPre[i][k] | isPre[behind][k];
}
}
}
}