1193번 - 분수찾기

1193번: 분수찾기

문제

무한히 큰 배열에 다음과 같이 분수들이 적혀있다.

1/1	1/2	1/3	1/4	1/5	…
2/1	2/2	2/3	2/4	…	…
3/1	3/2	3/3	…	…	…
4/1	4/2	…	…	…	…
5/1	…	…	…	…	…
…	…	…	…	…	…

이와 같이 나열된 분수들을 1/1 -> 1/2 -> 2/1 -> 3/1 -> 2/2 -> … 과 같은 지그재그 순서로 차례대로 1번, 2번, 3번, 4번, 5번, … 분수라고 하자.

X가 주어졌을 때, X번째 분수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 X(1 ≤ X ≤ 10,000,000)가 주어진다.

출력

첫째 줄에 분수를 출력한다.

나의 제출 - Python

input을 받아 각 층을 계산한 후, 해당 층에서 input을 빼거나 더하면 분모가 나온다. 만약 층이 홀수라면 그대로 분모로 사용하면 되는데, 층이 짝수라면 지금 나온 숫자가 분모가 아니라 분자로 들어간다. 출력은 분수의 형태로 format을 지정해준다.

X = int(input())
step = 1
x = 1

while True:
    if x >= X :
        break
    step += 1
    x += step 

if step % 2 == 1:
    m = step-(x-X)
    s = (x-X) +1

else:
    m = (x-X) +1
    s = step-(x-X)

print("{0}/{1}".format(s,m))