【算法分析与设计】【第六周】690. Employee Importance

题目来源：https://leetcode.com/problems/employee-importance/description/

小题大做。

题目大意

有一种员工信息的数据结构，其中包括员工的唯一ID，他的重要度和他的直系下属的ID。
现给出员工id，计算该员工及其所有下属的重要度之和。

// 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<int> subordinates;
};

例：
Input: [[1, 5, [2, 3]], [2, 3, []], [3, 3, []]], 1
Output: 11
解释:
员工 1 的重要度为 5 , 并拥有直系下属员工 2 和员工 3；员工 2 的重要度为 3 , 不拥有直系下属；员工 3 的重要度为 3 , 不拥有直系下属。
计算员工 1 及其所有下属的重要度之和 = 。

思路

1.把图建起来

容易看出，这是一个有向图。这里我们用邻接矩阵表示。
主对角线存员工重要度值。并用有向边表示上下属关系。

2.进行DFS

将得到的员工id作为DFS起点，进行一遍DFS。
与普通DFS不同的是，这里增加一个count变量，用于计算重要度之和。
一遍DFS可找到包含该起点的连通分量。连通分量中的每一个顶点，可在图（邻接矩阵）中找到这些顶点的重要度，并叠加。
最终得到的count即该员工及其所有下属的重要度之和。

代码

注：

• 由于题目没有说明员工id是否可以为0，所以数组下标都是直接按员工id来处理的，并没有进行类似（id-1）的操作。
• 为防止数组越界，将行列最大值设置为比题目陈述的最大员工数2000还大1的2001。

#define MAX 2001
int visited[MAX];        

class Solution {
public:
    int getImportance(vector<Employee*> employees, int id) {
        for (int i = 0; i < MAX; i++) visited[i] = 0;
        int size = employees.size();  
        vector<int> mm(MAX, 0);
        vector<vector<int>> M(MAX, mm);    
        // initialization
        for (int i = 0; i < MAX; i++)
            for (int j = 0; j < MAX; j++)
                M[i][j] = 0;
        // construct the graph
        for (int i = 0; i < size; i++) {
            int index = employees[i]->id;
            M[index][index] = employees[i]->importance;   // store the importance
            for (int j = 0; j < (employees[i]->subordinates).size(); j++) {
                M[index][(employees[i]->subordinates)[j]] = 1;   // sign the relationship
            }
        }   
        int sum = DFS(M, MAX, MAX, id);
        return sum;   
    }

    int DFS(vector<vector<int>> M, int MRowSize, int MColSize, int v) {
        int count = M[v][v];
        visited[v] = 1;  // begin with v
        stack<int> mys;
        mys.push(v);
        while (!mys.empty()) {
            int temp = mys.top();
            mys.pop();
            for (int i = 0; i < MRowSize; i++) {
                if (M[temp][i] != 0 && visited[i] == 0) {
                    count += M[i][i];
                    mys.push(i);
                    visited[i] = 1;
                }
            }
        }
        return count;
    } 
};

遇到的问题

总是提示“Runtime Error ”。但是反复查验并没有发现问题。
经测试，类似 M[i][j] = 0; 这样的赋值语句通不过leetcode的检测。
但是类似 M[0][0] = 0; 这样的赋值语句却可以行得通。
我猜想，可能是leetcode不支持二维数组的行列号为变量这种方式的赋值？
我也搞不明白，还要和同学进行交流，继续跟进这道题目。

其它解法

利用HashMap映射

wjx_sky对这道题进行了深入的探索，提供了“用hashMap将id与其对应的数据结构映射起来，找到id对应的员工信息并将其加入到一个队列中”这一思路。

更简洁的dfs写法

当然，对于dfs的思路，还有更简洁的写法。