diff --git "a/\354\213\234\354\206\214 \354\247\235\352\277\215.py" "b/\354\213\234\354\206\214 \354\247\235\352\277\215.py" new file mode 100644 index 0000000..d8d3e43 --- /dev/null +++ "b/\354\213\234\354\206\214 \354\247\235\352\277\215.py" @@ -0,0 +1,6 @@ +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