🐜 Study: 시소 짝꿍

Author: ing-eoking

시소 짝꿍.py

@@ -0,0 +1,9 @@
+def solution(weights):
+    answer = 0
+    m = [0] * 2001
+    for i in weights:
+        answer += (i % 3 == 0) * (m[i * 2 // 3] + m[i * 4 // 3]) \
+                + (i % 2 == 0) * (m[i * 3 // 2] + m[i // 2]) \
+                + (i % 4 == 0) * m[i * 3 // 4] + m[i * 2] \
+                + (m.__setitem__(i, m[i] + 1) or m[i] - 1)
+    return answer