95차 3번 문제풀이

Author: Yujin-Baek

live9/test95/문제3/백유진.py

@@ -0,0 +1,13 @@
+from itertools import combinations
+def solution(brown, yellow):
+    answer = []
+    
+    total = brown + yellow
+    
+    for h in range(1, total+1):
+        if total%h == 0:
+            w = total//h
+            if (w-2)*(h-2) == yellow:
+                return [w, h]
+    
+    return answer