[suwi] week 14

binary-tree-level-order-traversal/sungjinwi.py

@@ -0,0 +1,36 @@
+"""
+    풀이 :
+        level을 인자로 함께 넘겨서 level에 해당하는 인덱스에 append
+
+    node 개수 N
+
+    TC : O(N)
+        모든 node에 대해 dfs호출
+
+    SC : O(N)
+        dfs 함수 호출 스택이 노드 개수 N만큼
+"""
+
+# Definition for a binary tree node.
+# class TreeNode:
+#     def __init__(self, val=0, left=None, right=None):
+#         self.val = val
+#         self.left = left
+#         self.right = right
+class Solution:
+    def levelOrder(self, root: Optional[TreeNode]) -> List[List[int]]:
+        ans = []
+
+        def dfs(node: Optional[TreeNode], level: int) -> None:
+            if not node :
+                return
+            if len(ans) < level + 1 :
+                ans.append([node.val])
+            else:
+                ans[level].append(node.val)
+            dfs(node.left, level + 1)
+            dfs(node.right, level + 1)
+
+        dfs(root, 0)
+
+        return ans

counting-bits/sungjinwi.py

@@ -0,0 +1,18 @@
+"""
+    풀이 :
+        가장 낮은 자리의 비트를 제외한 dp의 성분 + 가장 낮은 자리 비트를 통해 dp 구현
+
+    n의 크기 : N
+
+    TC : O(N)
+
+    SC : O(N)
+"""
+class Solution:
+    def countBits(self, n: int) -> List[int]:
+        dp = [0] * (n + 1)
+
+        for num in range(1, n + 1):
+            dp[num] = dp[num >> 1] + (num & 1)
+
+        return dp

house-robber-ii/sungjinwi.py

@@ -0,0 +1,22 @@
+"""
+    풀이 : 
+        start를 포함하는 경우와 end를 포함하는 경우를 나눠서 dp 진행
+
+    nums의 갯수 N
+
+    TC : O(N)
+
+    SC : O(1)
+"""
+
+class Solution:
+    def rob(self, nums: List[int]) -> int:
+        if len(nums) == 1:
+            return nums[0]
+        prv1, cur1 = 0, 0
+        prv2, cur2 = 0, 0
+        for i in range(1, len(nums)):
+            prv1, cur1 = cur1, max(cur1, prv1 + nums[i])
+        for i in range(len(nums) - 1):
+            prv2, cur2 = cur2, max(cur2, prv2 + nums[i])
+        return max(cur1, cur2)