20.8.23 周赛 5480. 可以到达所有点的最少点数目 中等

题目

给你一个 有向无环图 ， n 个节点编号为 0 到 n-1 ，以及一个边数组 edges ，其中 edges[i] = [fromi, toi] 表示一条从点  fromi 到点 toi 的有向边。

找到最小的点集使得从这些点出发能到达图中所有点。题目保证解存在且唯一。

你可以以任意顺序返回这些节点编号。

示例 1：

输入：n = 6, edges = [[0,1],[0,2],[2,5],[3,4],[4,2]]
输出：[0,3]
解释：从单个节点出发无法到达所有节点。从 0 出发我们可以到达 [0,1,2,5] 。从 3 出发我们可以到达 [3,4,2,5] 。所以我们输出 [0,3] 。
示例 2：

输入：n = 5, edges = [[0,1],[2,1],[3,1],[1,4],[2,4]]
输出：[0,2,3]
解释：注意到节点 0，3 和 2 无法从其他节点到达，所以我们必须将它们包含在结果点集中，这些点都能到达节点 1 和 4 。

提示：

2 <= n <= 10^5
1 <= edges.length <= min(10^5, n * (n - 1) / 2)
edges[i].length == 2
0 <= fromi, toi < n
所有点对 (fromi, toi) 互不相同。

思路

1. 做的时候有自己思路，不过提交的时候有很多细节漏了，然后改了很多代码，和原来的思路有很大出入
2. 不总结了，看下面的题解思路

看题解后思路

1. 统计入度为0的结点，即没有出现在edges[i][1]的结点
2. 入度为0的结点就是结果了

代码

class Solution {
public:
    vector<int> findSmallestSetOfVertices(int n, vector<vector<int>>& edges) {
        vector<bool> judge(n, true);
        vector<bool> saveRes(n, false);
        vector<int> res;
        int count = 0;
        int i = 0;
        while(i < edges.size()){
            if(judge[edges[i][0]]) {
                judge[edges[i][0]]=false;
                saveRes[edges[i][0]] = true;
                count++;
            }
            if(saveRes[edges[i][1]]) saveRes[edges[i][1]] = false;
            if(judge[edges[i][1]]){
                judge[edges[i][1]]=false;
                count++;
            }
            i++;
        }
        for(int i = 0; i < n; i++){
            if(saveRes[i]) res.push_back(i);
        }
        return res;
    }
};