[Python3] 파이썬 for-else 사용법 및 활용 예제 (제곱근 찾기)

본 문제에서는 자연수 5개가 주어집니다.

 

1. 숫자를 차례로 곱해 나온 수가 제곱수1가 되면 found를 출력
2. 모든 수를 곱해도 제곱수가 나오지 않았다면 not found를 출력

 

예를 들어, [2, 4, 2, 5, 1]의 정수를 차례대로 곱할 때, 수를 곱해나가면 2, 8, 16, 80, 80 이 나옵니다.

 

16은 4를 제곱해 나온 수이므로, 이 수는 제곱수입니다. 따라서 found를 출력합니다.

 

반대로, [5, 1, 2, 3, 1]의 정수를 차례대로 곱할 때, 수를 곱해나가면 5, 5, 10, 30, 30 이 나옵니다.

이중 어떤 수도 제곱 수가 아니므로 not found를 출력합니다.

 

코드를 작성해봅시다.

 

 

일반적인 구현 (for 반복문과 조건문 사용)

res = "not found"
tmp = 1

for i in range(5):
  a = int(input())
  
  if (((tmp*a) ** 0.5) % 1) == 0:
    res = "found"
  tmp*=a

print(res)

 

위 소스코드는 직접 작성한 코드입니다.

 

5번 반복하며 각각의 수를 곱한 결과를 비교하는 방식입니다.

 

파이썬으로 구현을 할 때에는 더 간단하게 작성할 수 있습니다. 바로 for-else문입니다.

 

 

for-else 사용하여 구현

import math

numbers = [int(input()) for _ in range(5)]
multiplied = 1
for number in numbers:
    multiplied *= number
    if math.sqrt(multiplied) == int(math.sqrt(multiplied)):
        print('found')
        break
else:
    print('not found')

 

for-else문을 사용하였습니다.

 

리스트 내 원소를 탐색하면서 조건문을 통해 검사하는 방식입니다.

 

반복문 내 조건문에서 조건에 해당하는 경우 발생 시 break문을 통해 반복문을 빠져나갑니다.

 

정상적으로 5회 반복을 마치면 else문에 해당하는 구문이 실행됩니다.