[python] SWEA - 5986. 새샘이와 세 소수

🤔문제 해결

• D3 | 소수, 완전탐색

💨 미리 소수를 구해놓는다.

💨 3중 포문으로 소수 3개를 더해서 N 이 나오면 True

 

💻소스 코드

prime = []
for i in range(2, 1000):
    for j in range(2, i):
        if i % j == 0:
            break
    else:
        prime.append(i)

for tc in range(int(input())):
    N = int(input())
    M = len(prime)
    cnt = 0
    for i in range(M):
        if prime[i] > N:
            break
        for j in range(i, M):
            if prime[j] > N:
                break
            for k in range(j, M):
                if prime[k] > N:
                    break

                if prime[i] + prime[j] + prime[k] == N:
                    cnt += 1

    print(f'#{tc + 1} {cnt}')
 

📕문제 확인

출처: SW Expert Academy

SW Expert Academy