유레카 이론

문제

---

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

10448번: 유레카 이론

 

 

나의 풀이

---

사실 처음엔 itertools 모듈의 중복조합을 사용해 구했다. 물론 답은 맞았지만 조금 더 nice해 보이는 코드가 있어 첨부한다. 이 문제는 브루트 포스 문제로 그냥 가능한 경우를 모두 카운트 해주면 된다.

문제에서 주어진 조건 중 1000이하의 자연수만 입력으로 주어진다고 했기 때문에 1000 이하의 모든 유레카 수를 구해두면 된다. eureka라는 flag 배열을 사용해 해당 index의 수가 유레카 수라면 1로 flag-up 해준다. 삼중 for문을 사용해 세 개의 수를 더해서 삼각수가 되는 1000 이하의 수들에 대해 1로 flag-up 시켜주고 입력으로 수를 받아 이들을 반환해주면 된다.

 

코드

---

triangle = [n*(n+1)//2 for n in range(1, 46)]
eureka = [0] * 1001

#미리 1000이하의 모든 유레카 수를 구한다
for i in triangle:
    for j in triangle:
        for k in triangle:
            if i+j+k <= 1000:
                eureka[i+j+k] = 1

T = int(input())
for _ in range(T):
    print(eureka[int(input())])