1462. 课程表 IV-深度优先遍历
你总共需要上 numCourses 门课,课程编号依次为 0 到 numCourses-1 。你会得到一个数组 prerequisite ,其中 prerequisites[i] = [ai, bi] 表示如果你想选 bi 课程,你 必须 先选 ai 课程。
有的课会有直接的先修课程,比如如果想上课程 1 ,你必须先上课程 0 ,那么会以 [0,1] 数对的形式给出先修课程数对。
先决条件也可以是 间接 的。如果课程 a 是课程 b 的先决条件,课程 b 是课程 c 的先决条件,那么课程 a 就是课程 c 的先决条件。
你也得到一个数组 queries ,其中 queries[j] = [uj, vj]。对于第 j 个查询,您应该回答课程 uj 是否是课程 vj 的先决条件。
返回一个布尔数组 answer ,其中 answer[j] 是第 j 个查询的答案。
示例 1:
输入:numCourses = 2, prerequisites = [[1,0]], queries = [[0,1],[1,0]]
输出:[false,true]
解释:课程 0 不是课程 1 的先修课程,但课程 1 是课程 0 的先修课程。
示例 2:
输入:numCourses = 2, prerequisites = [], queries = [[1,0],[0,1]]
输出:[false,false]
解释:没有先修课程对,所以每门课程之间是独立的。
示例 3:
输入:numCourses = 3, prerequisites = [[1,2],[1,0],[2,0]], queries = [[1,0],[1,2]]
输出:[true,true]
解题代码如下:
/**
* Note: The returned array must be malloced, assume caller calls free().
*/
bool find(int now,int end,int **pre,int *presize,int *r){
for(int i=0;i<presize[now];i++){
if(r[pre[now][i]]==1){
r[pre[now][i]]=0;
if(pre[now][i]==end){
return true;
}
if(find(pre[now][i],end,pre,presize,r)){
return true;
}
}
}
return false;
}
bool* checkIfPrerequisite(int numCourses, int** prerequisites, int prerequisitesSize, int* prerequisitesColSize, int** queries, int queriesSize, int* queriesColSize, int* returnSize){
int **pre=(int **)malloc(sizeof(int*)*numCourses);
int *presize=(int *)malloc(sizeof(int)*numCourses);
bool *re=(bool *)malloc(sizeof(bool)*queriesSize);
int *r=(int *)malloc(sizeof(int)*numCourses);
for(int i=0;i<numCourses;i++){
presize[i]=0;
pre[i]=(int *)malloc(sizeof(int)*numCourses);
}
for(int i=0;i<prerequisitesSize;i++){
pre[prerequisites[i][1]][presize[prerequisites[i][1]]]=prerequisites[i][0];
presize[prerequisites[i][1]]++;
}
for(int i=0;i<queriesSize;i++){
for(int i=0;i<numCourses;i++){
r[i]=1;
}
re[i]=find(queries[i][1],queries[i][0],pre,presize,r);
}
*returnSize=queriesSize;
return re;
}