카펫(42842) - Lv.2

Author: DSMInhyeKang

Algorithm/ProgrammersCodingTest/PythonCodingTest.py

@@ -3524,4 +3524,14 @@ def solution(n, lighthouse):
         if onoff[now_leaf] == 1: continue
         onoff[parent] = 1
 
-    return sum(onoff) 
+    return sum(onoff) 
+
+
+
+# 카펫(42842) - Lv.2
+def solution(brown, yellow):    
+    for bh in range(1, brown//2 + 1):
+        bw = (brown - 2 * bh + 4) // 2
+        yw, yh = bw - 2, bh - 2
+        
+        if yellow == yw * yh and yellow + brown == bw * bh: return [bw, bh]