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<Employee*> 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<Employee*> &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);
}
}
}
};