백준 알고리즘 17362번 수학은 체육과목 입니다 2(python)

단순 수학 문제이다.

n = int(input())
if n <= 5:
    print(n)
else:
    if (n-5)//4 % 2 == 0:
        print(5 - (n-5)%4)
    else:
        print(1 + (n-5)%4)