알고리즘

<백준> 2581번 파이썬 알고리즘

changha. 2021. 7. 4. 14:38
a = int(input())
b = int(input())
arr = []

while(a < b+1):
    if a == 2:
        arr.append(a)
    else:
        for i in range(2, a):
            if a % i == 0:
                break
            if i == a - 1:
                arr.append(a)
    a += 1

    
if len(arr) == 0:
    print(-1)
else: 
    print(sum(arr))
    print(arr[0])

a == 2 일때는 if a % i == 0 이부분에서 예외가 나니 

윗줄에 2일때 리스트에 그냥 더한다.