LeetCode每日一题 1462. 课程表 IV

题目描述

给定 numCourses 门课程，以及一些课程之间的先决条件关系，然后给出一些查询，查询课程是否是其他课程的先决条件。

思路

这个问题可以使用图的拓扑排序来解决。首先，我们需要根据给定的先决条件关系构建课程之间的有向图。

然后，对于每个查询 [uj, vj]，我们可以检查课程 uj 是否是课程 vj 的先决条件。为了做到这一点，我们可以使用深度优先搜索（DFS）或广度优先搜索（BFS）来遍历图，并检查是否从课程 uj 可以到达课程 vj。如果可以到达，则 uj 是 vj 的先决条件，否则不是。

具体步骤如下：

1. 构建课程之间的有向图，可以使用邻接表或邻接矩阵表示。

2. 对于每个查询 [uj, vj]，使用 DFS 或 BFS 遍历图，检查是否从课程 uj 可以到达课程 vj。

3. 如果可以到达，则将查询的答案设置为 true，否则设置为 false。

代码

class Solution {
public:
    vector<bool> checkIfPrerequisite(int numCourses, vector<vector<int>>& prerequisites, vector<vector<int>>& queries) {
        // 构建课程之间的有向图
        vector<vector<int>> graph(numCourses);
        for (const vector<int>& pre : prerequisites) {
            int u = pre[0];
            int v = pre[1];
            graph[u].push_back(v);
        }
        
        vector<bool> answer;
        
        for (const vector<int>& query : queries) {
            int u = query[0];
            int v = query[1];
            // 使用 DFS 检查 u 是否是 v 的先决条件
            vector<bool> visited(numCourses, false);
            if (dfs(graph, u, v, visited)) {
                answer.push_back(true);
            } else {
                answer.push_back(false);
            }
        }
        
        return answer;
    }
    
    bool dfs(const vector<vector<int>>& graph, int u, int v, vector<bool>& visited) {
        if (u == v) {
            return true;
        }
        
        visited[u] = true;
        for (int neighbor : graph[u]) {
            if (!visited[neighbor] && dfs(graph, neighbor, v, visited)) {
                return true;
            }
        }
        
        return false;
    }
};