[编程题]树上最长单色路径

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

给定一棵二叉树的根节点(树的点数小于等于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 res=0;
    void path(TreeNode*root,int &black1,int &white0)
        {
        if(root->left==NULL&&root->right==NULL)
            {
            if(root->val==1)
                black1=1,white0=0;
            else black1=0,white0=1;
        }
        else
            {
            int lb1=0,lw0=0,rb1=0,rw0=0;
            if(root->left)
              path(root->left,lb1,lw0);
            if(root->right)
                path(root->right,rb1,rw0);
            if(root->val==1)
                {
                if(lb1+rb1+1>res)
                    res=lb1+rb1+1;
                white0=0;
                black1=lb1+1>rb1+1?lb1+1:rb1+1;
            }
            else
                {
                if(lw0+rw0+1>res)
                    res=lw0+rw0+1;
                black1=0;
                white0=lw0+1>rw0+1?lw0+1:rw0+1;
            }
        }
    }
    int findPath(TreeNode* root) {
        // write code here
        if(root==NULL)
            return 0;
        int black1=0,white0=0;
        path(root,black1,white0);
        return res;
    }
};