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

给你一棵以 root 为根的二叉树，二叉树中的交错路径定义如下：

选择二叉树中 任意 节点和一个方向（左或者右）。 如果前进方向为右，那么移动到当前节点的的右子节点，否则移动到它的左子节点。
改变前进方向：左变右或者右变左。 重复第二步和第三步，直到你在树中无法继续移动。 交错路径的长度定义为：访问过的节点数目 -
1（单个节点的路径长度为 0 ）。

请你返回给定树中最长 交错路径 的长度。

示例 1：

输入：root = [1,null,1,1,1,null,null,1,1,null,1,null,null,null,1,null,1]
输出：3
解释：蓝色节点为树中最长交错路径（右 -> 左 -> 右）。

示例 2：

输入：root = [1,1,1,null,1,null,null,1,1,null,1]
输出：4
解释：蓝色节点为树中最长交错路径（左 -> 右 -> 左 -> 右）。

示例 3：

输入：root = [1]
输出：0

解题思路

写一个dfs函数，在dfs的过程中，不停的记录当前结点的方向l_or_r，然后再进入它的子树时，相反则路径长度+1，相同则重置路径长度，然后每遍历一次，就更新一次最长路径

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left),
 * right(right) {}
 * };
 */
class Solution {
    int max_res;//最长路径
    //结点，方向，路径长度
    void dfs(TreeNode* node, int l_or_r, int dis) { // l_or_r==0为右，==1为左
        if (!node) {
            return;
        }
        max_res = max(dis, max_res); // 更新最长的路径
        if (l_or_r == 1) {           // 如果向左
            dfs(node->right, 0, dis + 1);
            dfs(node->left, 1, 1);  // 不符合z，只得重置
        } else {                    // 如果向右
            dfs(node->right, 0, 1); // 不符合z，只得重置
            dfs(node->left, 1, dis + 1);
        }
    }

public:
    int longestZigZag(TreeNode* root) {
        dfs(root->left, 1, 1);
        dfs(root->right, 0, 1);
        return max_res;
    }
};