每天一道Code 1372. 二叉树中的最长交错路径

在这里插入图片描述
代码思路
DFS
遍历每一个节点,它在父节点的交错路径上就代表它不可能作为根节点去dfs,而当前节点不在父节点的交错路径上就将它看做根节点dfs。
在代码中使用的是0,1指代下一步前进方向。
用全局变量存最大值。

另外,在做完之后去看了眼题解和大佬们的思路,看到一个双百的题解思路和我差不多,我咋跑不出来双百,好气,改了改遍历顺序,玄学出来了一个过题时间208ms后面再怎么改都慢不了了(ノ ̄▽ ̄),更有趣的是交相同代码都要慢十几ms,再也没跑出来208ms,真·玄学过题。在这里插入图片描述

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
    
    
public:
    int ans = 0;
    void dfs(TreeNode* root, bool dir,int k){
    
    
        //dir=0 ->left, dir=1 ->right
        if(!root) return ;
        ans = max(ans,k);       //更新最大值
        if(dir){
    
                        //当前结点是父节点的左节点
            dfs(root->left,1,1);   //下一步同向,作为根节点
            dfs(root->right,0,k+1);//下一步反向,在父节点交错路径上
        }
        else{
    
    
            dfs(root->left,1,k+1);
            dfs(root->right,0,1);
        }
        return;
    }
    int longestZigZag(TreeNode* root) {
    
    
        //最长交错路径,没算上root,因此ans不需要减1
        dfs(root->left,1,1);    //dfs左子树下一步向右走
        dfs(root->right,0,1);   //dfs右子树下一步向左走
        return ans;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_41746268/article/details/108346413