Leetcode：1372. 二叉树中的最长交错路径

@[TOC] Leetcode：1372. 二叉树中的最长交错路径
官方解答思路

思路分析

题目要求我们找到树当中一条最长的左右交错的路径。这道题目可以从动态规划BFS和深度优先搜索DFS这两种思路下手。

动态规划BFS

记当u为左孩子时，用f(u)表示其父节点的交错路径长度；当u为右孩子时，用g(u)表示其父节点的交错路径长度。因为我们需要一条「交错」的路，如果 u 是 左孩子，那么其父节点的交错路径长度f(u)一定是它爸爸的爸爸的右儿子的交错路径长度，而这个老头的交错路径长度即是g(father(u))。同理，如果u是右孩子，则它老爸的交错路径长度是g(u)，它老爸的老爸的交错路径长度是 f(father(u))。按此思想，我们得出以下公式：
当u是左孩子时：f[u] = g[father(u)] + 1;
当u是右孩子时：g[u] = f[father(u)] + 1;
是的，说白了最最重要的思想就是“老爷子+1”就是老爸的交错路径长度。

代码如下：

class Solution {
public:
    unordered_map <TreeNode*, int> f, g;
    queue <pair <TreeNode *, TreeNode *>> q; 
    void dp(TreeNode *o) {
        f[o] = g[o] = 0;
        q.push({o, nullptr});
        while (!q.empty()) {
            auto y = q.front(); q.pop();
            auto x = y.second, u = y.first;
            f[u] = g[u] = 0;
            if (x) {
                if (x->left == u) f[u] = g[x] + 1;
                if (x->right == u) g[u] = f[x] + 1;
            }
            if (u->left) q.push({u->left, u});
            if (u->right) q.push({u->right, u});
        }
    }
    int longestZigZag(TreeNode* root) {
        dp(root);
        int maxAns = 0;
        for (const auto &u: f) maxAns = max(maxAns, max(u.second, g[u.first]));
        return maxAns;
    }
};

深度优先搜索DFS

记len为当前最长交错路径的长度，标志符号dir表示为方向（0向左，1向右）。我们从根结点出发，如果当前结点应该向左且可以向左的时候，那么就让其向左，且len+1；但如果当前结点应该向左，但却没有左孩子的时候，则只能向右，此时设置len=1；如果左右孩子都没有，则停止。以此类推，右边也是一样。需要注意的是，一开始从根结点root出发的时候，len要设为0，而不是1。因为root结点没有父节点，走过的路径为0。
还是递归大法香啊！

代码如下：

class Solution {
public:
    int maxAns;

    /* 0 => left, 1 => right */
    void dfs(TreeNode* o, bool dir, int len) {
        maxAns = max(maxAns, len);
        if (!dir) {
            if (o->left) dfs(o->left, 1, len + 1);
            if (o->right) dfs(o->right, 0, 1);
        } else {
            if (o->right) dfs(o->right, 0, len + 1);
            if (o->left) dfs(o->left, 1, 1);
        }
    } 

    int longestZigZag(TreeNode* root) {
        if (!root) return 0;
        maxAns = 0;
        dfs(root, 0, 0);
        dfs(root, 1, 0);
        return maxAns;
    }
};