[오늘의 알고리즘] 시소 짝꿍 (#64)

Co-authored-by: unknown <Administrator@SKCC20N00129.SKCC.NET>

Author: yongjun-0903

yongjun-0903/시소 짝꿍.py

@@ -0,0 +1,17 @@
+from itertools import combinations
+from math import gcd
+
+def solution(weights):
+    answer = 0
+    combs = list(combinations(weights, 2))
+    siso = [2, 3, 4]
+
+    for comb in combs:
+        if comb[0] == comb[1]:
+            answer += 1
+        else:
+            x = gcd(comb[0], comb[1])
+            if comb[0] // x and comb[1] // x in siso:
+                answer += 1
+    
+    return answer