nowcoder 树上最长单色路径(最长路径,二叉树+dp)
2016-04-20 14:07
309 查看
题目来源:
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数组)
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; };
相关文章推荐
- android 自己的思路写的订单管理界面
- nginx重启
- android studio+Droid4x(海马玩)
- block没那么难(三):block和对象的内存管理
- 手机页面自适应
- Java BIO、NIO、AIO、同步、异步
- Android Scrollview 内部组件android:layout_height="match_parent"无效的解决办法
- hdoj 5666 Segment(欧拉函数)
- 关于maven的依赖分析
- html 新消息title闪动实现
- transient详解
- [安卓开发] 真正获取本地图片的宽高
- couchbase笔记
- Ubuntu 14.04 配置 gitosis 和 gitweb
- 分布式之《保证分布式系统数据一致性的6种解决方案》
- 二维数组中的查找
- 二叉树节点路径查询
- [置顶] MapReduce源码分析之LocatedFileStatusFetcher
- 前端学习
- Timus 1060. Flip Game