nowcoder 树上最长单色路径(最长路径，二叉树+dp)

题目来源：

http://www.nowcoder.com/test/1597148/summary 第三题

问题描述：

对于一棵由黑白点组成的二叉树，我们需要找到其中最长的单色简单路径，其中简单路径的定义是从树上的某点开始沿树边走不重复的点到树上的另一点结束而形成的路径，而路径的长度就是经过的点的数量(包括起点和终点)。而这里我们所说的单色路径自然就是只经过一种颜色的点的路径。你需要找到这棵树上最长的单色路径。
给定一棵二叉树的根节点(树的点数小于等于300，请做到O(n)的复杂度)，请返回最长单色路径的长度。这里的节点颜色由点上的权值表示，权值为1的是黑点，为0的是白点。

分析：

我自己的实现就是在树上采用递归，分别获取左右子树中连续单色路径，然后处理。但是这样要按左右子树是否存在分为四种情况讨论，代码量较大，索性最终A了此题。

比赛结束后看了别人代码，发现若增加个二维（两种颜色）数组用于存储中间值，代码量能减少很多。代码地址为：https://github.com/skyqinsc/ACM/blob/master/%E4%BA%8C%E5%8F%89%E6%A0%91%E5%90%8C%E8%89%B2%E8%B7%AF%E5%BE%84.cpp

但这种借助数组的方法有个缺陷，就是树的深度不能太深。

代码：

（采用dp数组）

class LongestPath {
public:
    LongestPath(){memset(dp,0,sizeof dp);ans=0;}
    int findPath(TreeNode* root){
    	// write code here
    	dfs(root,1);
    	return ans;
    }
    void dfs(TreeNode *root,int id){
    	if(root==NULL) return;
    	dfs(root->left,id<<1);dfs(root->right,id<<1|1);
    	if(root->val==0){
    		dp[id][0]=max(dp[id<<1][0],dp[id<<1|1][0])+1;
    		ans=max(ans,dp[id<<1][0]+dp[id<<1|1][0]+1);
    	}
    	else{
    		dp[id][1]=max(dp[id<<1][1],dp[id<<1|1][1])+1;
    		ans=max(ans,dp[id<<1][1]+dp[id<<1|1][1]+1);
    	}
    }
private:
	int dp[1010][2];
	int ans;
};