LeetCode 2039 网络空闲的时刻[bfs 数学] HERODING的LeetCode之路

Posted 2022-03-21 HERODING23

解题思路：
一道变形的广度优先题目，本质上还是使用BFS的模板，只不过需要对题目进行分析，对于每个访问到的点，分别讨论其未到重发时间的情况和已经重发过的情况，所需时间都可以用该式表示patience[next] * ((2 * dis - 1) / patience[next]) + 2 * dis + 1，广度优先遍历直到队列为0，获取最大的time即可，代码如下：

class Solution 
public:
    int networkBecomesIdle(vector<vector<int>>& edges, vector<int>& patience) 
        int n = patience.size();
        vector<vector<int>> adj(n);
        // 初始化距离矩阵
        for(auto& edge : edges) 
            adj[edge[0]].push_back(edge[1]);
            adj[edge[1]].push_back(edge[0]);
        
        vector<bool> visited(n, false);
        queue<int> q;
        q.push(0);
        visited[0] = true;
        int ans = 0;
        int dis = 1;
        while(!q.empty()) 
            int size = q.size();
            for(int i = 0; i < size; i ++) 
                int node = q.front();
                q.pop();
                for(auto& next : adj[node]) 
                    if(visited[next]) continue;
                    int time = patience[next] * ((2 * dis - 1) / patience[next]) + 2 * dis + 1;
                    ans = max(ans, time);
                    q.push(next);
                    visited[next] = true;
                
            
            dis ++;
        
        return ans;
    
;