LeetCode 690. Employee Importance

LeetCode 690. Employee Importance_第1张图片

题意:给一个员工的数据结构,保存了1.id 2.重要值 3.直接管辖的员工id。要求给出一个id,返回他和他的所有直接/间接下属的重要值之和。

solution:hash。用排序+暴力+递归的方法应该也是很好做的,但是效率肯定会受影响。使用hash只需要至多遍历一遍数组即可。

/*
// Employee info
class Employee {
public:
    // It's the unique ID of each node.
    // unique id of this employee
    int id;
    // the importance value of this employee
    int importance;
    // the id of direct subordinates
    vector subordinates;
};
*/
class Solution {
public:
    /*
    访问一个元素首先在hash表中查找,没有就从i开始查找,并且把经过的元素都加入hash表。
    找到队首节点,将其sub添加进q,累加res,队首出列,循环直到q为空,返回结果。
    */
    int getImportance(vector employees, int id) {
        unordered_map hash; // id - index
        queue q;
        int i = 0;
        int res = 0;
        q.push(id);
        
        while ( !q.empty() ) {
            int curr_id = q.front();
            if ( hash.find(curr_id) != hash.end()  ) {
                for ( auto sub : employees[hash[curr_id]]-> subordinates) {
                    q.push(sub);
                }
                res += employees[hash[curr_id]]-> importance;
            }
            else {
                for ( ; i < employees.size(); i++ ) {
                    hash[employees[i]->id] = i;
                    if ( employees[i]->id == curr_id ) {
                        for ( auto sub : employees[i]-> subordinates) {
                            q.push(sub);
                        }
                        res += employees[i]-> importance;
                    }
                }                
            }
            q.pop();
        }
        return res;
    }
};

submission:

LeetCode 690. Employee Importance_第2张图片

你可能感兴趣的:(解题记录,leetcode,hash)