690 leetcode employee importance

690 leetcode employee imortance, 难度easy

由于上周老师讲的是BFS,所以这周就做了这道BFS相关的题。
因为给的数据里包含一个结构体,所以先把它转换成矩阵比较方便,然后再用BFS进行解决。V为顶点的数量,E为边的数量,时间复杂度的计算应该要分成两部分来看,转换成矩阵的部分为O(E),BFS的时间复杂度为O(V+E),所以时间复杂度应为O(V+E)。
以下是代码:

class Solution {
public:
    int getImportance(vector employees, int id) {       
        vector<vector<int>> matrix(employees.size() + 1);
        vector<int> value(employees.size() + 1);
        bool visited[employees.size() + 1] = {false};
        int res = 0;

        getMatrix(employees, matrix, value);
        queue<int> q;
        q.push(id);
        while(!q.empty()) {
            int temp = q.front();
            res += value[temp];
            visited[temp] = true;
            q.pop();
            for(auto i:  matrix[temp]) {
                q.push(i);
            }
        }
        return res;
    }

    void getMatrix(vector &employees, vector<vector<int>>& matrix, vector<int> &value) {
        for(auto i : employees) {
            value[i->id] = i->importance;
            for(auto j: i->subordinates) {
                matrix[i->id].push_back(j);
            }
        }
    }
};

你可能感兴趣的:(690 leetcode employee importance)