[Python] 백준 1977번 완전제곱수

[Python] 백준 1977번 완전제곱수

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

문제

코드

m = int(input())
n = int(input())
i = 1
l = []
while True:
    if i*i >= m and i*i <= n:
        l.append(i*i)
    elif i*i > n:
        break
    i += 1
if len(l) == 0:
    print(-1)
else:
    print(sum(l))
    print(min(l))

리뷰

for문 사용해서 푸는게 비효율적이라고 느꼈다.

그냥 무한루프 돌린 다음에 i*i이 n을 넘어가는 순간 바로 탈출 시키면

훨씬 효율적으로 작동할 것 같다.

다른 풀이

m = int(input())
n = int(input())
x = 1
l = []
while x**2 <= n:
    if x**2 >= m:
        l.append(x**2)
    x += 1
if len(l) > 0:
    print(sum(l))
    print(min(l))
else:
    print(-1)