你这个学期必须选修 numCourses
门课程,记为 0
到 numCourses - 1
。
在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites
给出,其中 prerequisites[i] = [ai, bi]
,表示如果要学习课程 ai
则 必须 先学习课程 bi
。
- 例如,先修课程对
[0, 1]
表示:想要学习课程0
,你需要先完成课程1
。
请你判断是否可能完成所有课程的学习?如果可以,返回 true
;否则,返回 false
。
即图中不能有环。
注意邻接表和邻接矩阵的转换。
自己的解法:拓扑排序,找出每个节点前面节点的个数,不断遍历将值为零的访问。
类似广度优先搜索,不过不完全一样,未使用队列,时间复杂度较高。
class Solution {
public:
bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
if(prerequisites.size()==0)return true;
vector<long long> pre(numCourses,0);
vector<int> visited(numCourses,0);
for(int i=0;i<prerequisites.size();i++)
{
int a=prerequisites[i][0],b=prerequisites[i][1];
pre[a]++;
}
int maxcou=numCourses;
while(maxcou--)
{
for(int i=0;i<numCourses;i++)
{
if(visited[i]==0&&pre[i]==0)
{
visited[i]=1;
for(int j=0;j<prerequisites.size();j++)
{
if(prerequisites[j][1]==i)
pre[prerequisites[j][0]]--;
}
}
}
}
for(int i=0;i<numCourses;i++)
{
if(visited[i]==0)return false;
}
return true;
}
};
注意:在while循环中判断条件为课程数量,即最多可能循环这么多次。考虑极端情况:
5->4->3->2->1,第一次到最后才找到5,第二次到最后才找到4,这样每次循环只能访问一个节点。需n次。若为1->2->3->4->5,则仅需一次即可访问全部节点。
题目中使用的是邻接矩阵,或者转换一下使用邻接表,会更方便。
官方题解:
1.深搜
class Solution {
private:
vector<vector<int>> edges;
vector<int> visited;
bool valid = true;
public:
void dfs(int u) {
visited[u] = 1;
for (int v: edges[u]) {
if (visited[v] == 0) {
dfs(v);
if (!valid) {
return;
}
}
else if (visited[v] == 1) {
valid = false;
return;
}
}
visited[u] = 2;
}
bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
edges.resize(numCourses);
visited.resize(numCourses);
for (const auto& info: prerequisites) {
edges[info[1]].push_back(info[0]);
}
for (int i = 0; i < numCourses && valid; ++i) {
if (!visited[i]) {
dfs(i);
}
}
return valid;
}
};
2.广搜
使用一个队列;将邻接矩阵转化为邻接表。
class Solution {
private:
vector<vector<int>> edges;
vector<int> indeg;
public:
bool canFinish(int numCourses, vector<vector<int>>& prerequisites) {
edges.resize(numCourses);
indeg.resize(numCourses);
for (const auto& info: prerequisites) {
edges[info[1]].push_back(info[0]);
++indeg[info[0]];
}
queue<int> q;
for (int i = 0; i < numCourses; ++i) {
if (indeg[i] == 0) {
q.push(i);
}
}
int visited = 0;
while (!q.empty()) {
++visited;
int u = q.front();
q.pop();
for (int v: edges[u]) {
--indeg[v];
if (indeg[v] == 0) {
q.push(v);
}
}
}
return visited == numCourses;
}
};