백준 10448 유레카 이론

문제 링크: https://www.acmicpc.net/problem/10448

10448번: 유레카 이론

삼각수 = 1, 3, 6, 10, 15, .... 

n번째 삼각수 = n * (n + 1) / 2

45번째 삼각수 = 45 * 46 / 2 = 1035

3중 for문을 각각 45번째 삼각수까지 돌면서 세개의 삼각수의 합이 입력받은 수와 같은경우가 있는지 확인한다.

#include <bits/stdc++.h>
using namespace std;
 
int triNum(int x) {
    return x * (x + 1) / 2;
}
 
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
 
    int t;
    cin >> t;
    while(t--) {
        int n;
        cin >> n;
 
        bool ok = false;
        for(int i = 1; i <= 45; i++) {
            for(int j = 1; j <= 45; j++) {
                for(int k = 1; k <= 45; k++) {
                    if(triNum(i) + triNum(j) + triNum(k) == n) ok = true;
                }
            }
        }
 
        if(ok) cout << 1 << '\n';
        else cout << 0 << '\n';
    }
    return 0;
}