[BOJ] 2097

https://www.acmicpc.net/problem/2097 

정사각형을 만드는게 가장 효율적이고
만든 다음에 남는 점의 개수가 있으면 무조건 +2를 해주면 됨.

다만 n이 1, 2인 경우에는 계산이 어려우므로 4를 반환하도록 조건을 넣어줘야 함.

import math
def solve(n):
    if n in [1, 2]:
        return 4
    a = int(math.sqrt(n))
    b = int(n / a)
    res = (a + b - 2) * 2
    if n != a * b:
        res += 2
    return res
print(solve(int(input())))