题目:对于一棵由黑白点组成的二叉树,我们需要找到其中最长的单色简单路径,其中简单路径的定义是从树上的某点开始沿树边走不重复的点到树上的另一点结束而形成的路径,而路径的长度就是经过的点的数量(包括起点和终点)。而这里我们所说的单色路径自然就是只经过一种颜色的点的路径。你需要找到这棵树上最长的单色路径。
给定一棵二叉树的根节点(树的点数小于等于300,请做到O(n)的复杂度),请返回最长单色路径的长度。这里的节点颜色由点上的权值表示,权值为1的是黑点,为0的是白点。
本题有个很重要的解题技巧,就是void path(TreeNode* root,int& white,int& black)里面的引用,因为原来每个节点里只存储了颜色值,没有包含关于左右子树单色路径长度的信息,通过引用就可以把单色路径值从树的底部一步步传上来,非常方便,值得借鉴。
代码如下:
class LongestPath {
public:
int res=0;
void path(TreeNode* root,int& white,int& black){
if(root->left==NULL&&root->right==NULL){
if(root->val==1)
white=0,black=1;
else
black=0,white=1;
}
else{
int lw0=0,lb1=0,rw0=0,rb1=0;
if(root->left){
path(root->left,lw0,lb1);
}
if(root->right){
path(root->right,rw0,rb1);
}
if(root->val==1){
if(lb1+rb1+1>res){
res=lb1+rb1+1;
}
white=0;
black=lb1>rb1?lb1+1:rb1+1;
}
else{
if(rw0+lw0+1>res)
res=rw0+lw0+1;
black=0;
white=rw0>lw0?rw0+1:lw0+1;
}
}
}
int findPath(TreeNode* root) {
if(root==NULL)return 0;
int white=0,black=0;
path(root,white,black);
return res;
}
};