백준 1789: 수들의 합 (Python)

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

1789번: 수들의 합

 

import sys
s = int(sys.stdin.readline())
res = 0

for i in range(s+1):
    if s == 1:
        res = 1
        break
    if i * (i + 1) // 2 > s:
        res = i - 1
        break

print(res)

 

뭔가.. 중구난방함

 

 

import sys
s = int(sys.stdin.readline())
n = 1

while n * (n + 1) // 2 <= s:
    n += 1

print(n - 1)

 

코드 길이를 좀 줄여보았다