题目出处:https://leetcode-cn.com/problems/find-eventual-safe-states/
思路:用三种颜色表示每个结点的状态,未访问过(0表示)、已访问过(1表示)、结点是安全的(2表示)。使用bool形式的递归判断一个结点是否为安全的。
递归:1. 搞清f(n)和f(n-1)之间的关系。 2.搞清递归的终止条件。
递归关系:按这题来说,就是如果结点node是安全的,则它所有的邻居都必须是安全的;如果结点node是不安全的,则它其中有一个邻居是不安全的。
递归终止条件:碰到一个已经访问过得结点。如果该节点已经访问过,则返回false;如果该节点是安全的,则返回true
class Solution {
public:
bool safe(vector<vector<int>>& graph, int node, vector<int>& color){ //结点node是否安全
if(color[node] != 0){ //递归终止条件,碰到一个已经访问过的结点
return color[node] == 2;
}
color[node] = 1; //结点被访问过
for(auto & neibor : graph[node]){ // node结点的邻居
if(!safe(graph, neibor, color)){ //有一个邻居结点不安全的话,那么该node结点就是不安全的
return false;
}
}
color[node] = 2; //结点是安全的
return true; //所有的子节点都是安全的,则该node结点是安全的
}
vector<int> eventualSafeNodes(vector<vector<int>>& graph) {
int n = graph.size(); // 节点个数
vector<int> ans;
vector<int> color(n, 0);
for(int i = 0; i < n; ++i){
if(safe(graph, i, color)){
ans.push_back(i);
}
}
return ans;
}
};