Update house_robber_iii.py

yingl · web-flow · commit 6908cd26dbe9 · 2017-04-25T18:22:44.000+08:00
diff --git a/house_robber_iii.py b/house_robber_iii.py
@@ -1,4 +1,4 @@
-# -*- coding: utf-8 -*-
+# coding: utf-8
 
 class Solution:
     # @param {TreeNode} root, the root of binary tree.
@@ -15,4 +15,6 @@ def _houseRobber3(self, root):
         rob_right = self._houseRobber3(root.right)
         not_rob = rob_left[1] + rob_right[1] + root.val
         rob = max(rob_left) + max(rob_right)
-        return (not_rob, rob)
+        return (not_rob, rob)
+
+# medium: http://lintcode.com/zh-cn/problem/house-robber-iii/
