代码思路
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;
}
};