树上最长单色路径

问题描述：

对于一棵由黑白点组成的二叉树，我们需要找到其中最长的单色简单路径，其中简单路径的定义是从树上的某点开始沿树边走不重复的点到树上的另一点结束而形成的路径，而路径的长度就是经过的点的数量(包括起点和终点)。而这里我们所说的单色路径自然就是只经过一种颜色的点的路径。你需要找到这棵树上最长的单色路径。

给定一棵二叉树的根节点(树的点数小于等于300，请做到O(n)的复杂度)，请返回最长单色路径的长度。这里的节点颜色由点上的权值表示，权值为1的是黑点，为0的是白点。

问题求解：

/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/

class LongestPath {
public:
int maxLen=0;
int findPath(TreeNode* root) {
if(root==NULL) return 0;
MaxlenPath(root);
return maxLen;
}
int MaxlenPath(TreeNode* root)
{
if(root==NULL) return 0;//(1)root为空时返回0
int leftLen=0, rightLen=0;
int tmp;
if(root->left != NULL)
{//(2)搜索左子树
tmp = MaxlenPath(root->left);//左子树的最大长度
//如果当前节点和左子树节点颜色一样，才更新
if(root->left->val == root->val)
leftLen = tmp;
}
if(root->right != NULL)
{//(3)搜索右子树
tmp = MaxlenPath(root->right);//右子树的最大长度
//如果当前节点和右子树节点颜色一样，才更新
if(root->right->val == root->val)
rightLen = tmp;
}
//(4)更新最长的单色简单路径长度
maxLen=(leftLen+rightLen+1)>maxLen?(leftLen+rightLen+1):maxLen;
//(5)返回当前节点左右子树中较长的路径长度
return leftLen>rightLen?leftLen+1:rightLen+1;
}
};