【AtCoder】ABC 335 B - Tetrahedral Number | 茶コーダーが解くAtCoder

atcoder.jp

実行時間制限: 2 sec / メモリ制限: 1024 MB / Difficulty: 52

問題概要

整数 $N$ が与えられる。非負整数の組 $(x, y, z)$ であって $x + y + z \leq N$ を満たすものを辞書順で小さい方から順に全て出力せよ。

制約

$0 \leq N \leq 21$

考察

$N$ の制約が非常に小さいので、3重ループを用いて $(x, y, z)$ の組を全探索し、条件を満たすものを順に出力していこう。

コード

#include <bits/stdc++.h>
using namespace std;

#define rep(i, start, end) for (auto i = (start); (i) < (end); (i)++)

// ======================================== //

int main()
{
    int N;
    cin >> N;

    rep(x, 0, N + 1) rep(y, 0, N + 1) rep(z, 0, N + 1)
    {
        if (x + y + z <= N)
        {
            cout << x << " " << y << " " << z << endl;
        }
    }