LINTCODE——二叉树具有相同值的最长路径

LINTCODE——二叉树具有相同值的最长路径

思路：这套题目做了挺久的，思路清楚了其实不难
首先，这道题目中的E数组（存储边数据的数组）我们需要处理一下，我用的方法是建立一个二维数组adj，adj[i]存储与i节点相连的节点编号；
其次：我们在进行深度优先搜索的时候，要用一个marked的容器，记录节点i是否被访问过，如果被访问过，直接跳过节点i,不然会进入死循环
最后：最长路径的数据处理和LINTCODE中二叉树的最大路径和处理方法一致，得到结果，成功AC
PS：由于这个花了我不少时间，所以代码部分个人觉得注释也比较详细；

class Solution {
private:
    //N为节点数
    //adj[i]存储与i相连接的节点编号
    int res = 0;
    int N;
    vector<vector<int>> adj;

public:
    /*
     * @param : as indicated in the description
     * @param : as indicated in the description
     * @return: Return the number of edges on the longest path with same value.
     */
    int LongestPathWithSameValue(vector<int> &A, vector<int> &E) {
        // write your code here

        N = A.size();
        adj = vector<vector<int>>(N);
        //输入数据处理,把边保存到adj里面
        for(int i = 0; i < N-1; i++)
        {
            //构造adj
            addEdge(E[i*2]-1, E[i*2+1]-1);
        }

        for(int i = 0 ; i < N ; i++)
        {
            //建立一个容器marked判断节点i是否被访问过
            vector<bool> marked(N, false);
            depth(A, marked, i);
        }
        return res;
    }
    int depth(vector<int> &A, vector<bool> &marked, int index)
    {

        int L = 0, R = 0;
        marked[index] = true;
        bool flag = true;
        //在与index相连接的节点里面寻找值相等并且没有被访问过的点，深度搜索
        for(int j : adj[index])
        {
            //判断节点值是否与index的值相等
           if(A[j] == A[index] && marked[j] == false)
           {
                //无法得知节点j是Index的左节点还是右节点
                //但是这里只求路径长度，可以假定第一个匹配到的为左，第二个为右；
                if(flag)
                {
                    L = depth(A, marked, j)+1;
                    flag = false;
                }
                else
                {
                    R = depth(A, marked, j)+1;
                }
            }
        }
        //这个处理过程和二叉树的最大路径和类似，这里就不再多说了
        res = max(res, L + R);
        return max(L,R);
    }
    void addEdge(int v, int w)
    {
        adj[v].push_back(w);
        adj[w].push_back(v); 
    }
};