2368. 受限条件下可到达节点的数目-c语言记忆深度优先搜索

2368. 受限条件下可到达节点的数目-c语言记忆深度优先搜索

现有一棵由 n 个节点组成的无向树，节点编号从 0 到 n - 1 ，共有 n - 1 条边。

给你一个二维整数数组 edges ，长度为 n - 1 ，其中 edges[i] = [ai, bi] 表示树中节点 ai 和 bi 之间存在一条边。另给你一个整数数组 restricted 表示 受限 节点。

在不访问受限节点的前提下，返回你可以从节点 0 到达的 最多 节点数目。

注意，节点 0 不 会标记为受限节点。

示例 1：

输入：n = 7, edges = [[0,1],[1,2],[3,1],[4,0],[0,5],[5,6]], restricted = [4,5]
输出：4
解释：上图所示正是这棵树。
在不访问受限节点的前提下，只有节点 [0,1,2,3] 可以从节点 0 到达。

示例 2：

输入：n = 7, edges = [[0,1],[0,2],[0,5],[0,4],[3,2],[6,5]], restricted = [4,2,1]
输出：3
解释：上图所示正是这棵树。
在不访问受限节点的前提下，只有节点 [0,5,6] 可以从节点 0 到达。

这个图方法既不能使用二维矩阵，用空间换时间，否则空间复杂度不能通过，也不能采用深度优先，否则时间复杂度通过不了，博主这里建议使用c语言解题使用邻接表可以解决问题：

int count;
void dfs(int **edges,int edgesSize,int*restricted,int restrictedSize,int now,int *r){
    printf("now %d ",now);
    count++;
    for(int i=0;i<edgesSize;i++){
        if(r[i]==1){
            if(edges[i][0]==now){
                r[i]=0;

              
                int j;
                for( j=0;j<restrictedSize;j++){
                    if(edges[i][1]==restricted[j]){
                        break;
                    }
                }
                if(j==restrictedSize){
                    dfs(edges,edgesSize,restricted,restrictedSize,edges[i][1],r);

                }
              
                
            }
            if(edges[i][1]==now){
                r[i]=0;
                
                  int j;
                   for( j=0;j<restrictedSize;j++){
                    if(edges[i][0]==restricted[j]){
                       break;
                    }
                }
                if(j==restrictedSize){
                     dfs(edges,edgesSize,restricted,restrictedSize,edges[i][0],r);

                }
               
                
            }

        }
    }

}

int reachableNodes(int n, int** edges, int edgesSize, int* edgesColSize, int* restricted, int restrictedSize){
    int *r=(int *)malloc(sizeof(int)*edgesSize);
    for(int i=0;i<edgesSize;i++){
        r[i]=1;
    }
    count=0;

     dfs(edges,edgesSize,restricted,restrictedSize,0,r);
     return count;



}