結果

問題 No.3506 All Distance is Square Number
コンテスト
ユーザー takuma saito
提出日時 2026-04-19 16:55:47
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
WA  
実行時間 -
コード長 2,527 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,642 ms
コンパイル使用メモリ 339,648 KB
実行使用メモリ 6,400 KB
最終ジャッジ日時 2026-04-19 16:56:06
合計ジャッジ時間 17,153 ms
ジャッジサーバーID
（参考情報） 		judge2_1 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 3 WA * 26
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>

using namespace std;

bool s(int n) {
    if (n < 0) return false;
    int r = round(sqrt(n));
    return r * r == n;
}

int main() {
    int n;
    cin >> n;
    if (n < 2) return 0;
    vector<int> a(n + 1), b(n + 1);
    int w;
    for (w = 1; w <= 200; ++w) {
        vector<bool> u(201, 0);
        u[w] = 1;
        bool ok = true;
        for (int i = 3; i <= n; ++i) {
            bool fi = false;
            for (int va = 1; va <= 200; ++va) {
                if (u[va]) continue;
                for (int vb = 1; vb <= 200; ++vb) {
                    if (va == vb || u[vb]) continue;
                    if (!(s(va) || s(w + vb))) continue;
                    if (!(s(vb) || s(w + va))) continue;
                    bool okj = true;
                    for (int j = 3; j < i; ++j) {
                        if (!s(va + a[j]) && !s(vb + b[j]) && !s(va + w + b[j]) && !s(vb + w + a[j])) {
                            okj = false;
                            break;
                        }
                    }
                    if (okj) {
                        a[i] = va;
                        b[i] = vb;
                        u[va] = u[vb] = 1;
                        fi = true;
                        break;
                    }
                }
                if (fi) break;
            }
            if (!fi) {
                ok = false;
                break;
            }
        }
        if (ok) break;
    }
    int m = (n == 2) ? 1 : 2 * n - 3;
    cout << m << endl;
    cout << "1 2 " << w << endl;
    for (int i = 3; i <= n; ++i) cout << "1 " << i << " " << a[i] << endl;
    for (int i = 3; i <= n; ++i) cout << "2 " << i << " " << b[i] << endl;
    for (int i = 1; i < n; ++i) {
        for (int j = i + 1; j <= n; ++j) {
            vector<int> q;
            if (i == 1 && j == 2) {
                q = {1};
            } else if (i == 1) {
                if (s(a[j])) q = {j - 1};
                else q = {1, n + j - 3};
            } else if (i == 2) {
                if (s(b[j])) q = {n + j - 3};
                else q = {1, j - 1};
            } else {
                if (s(a[i] + a[j])) q = {i - 1, j - 1};
                else if (s(b[i] + b[j])) q = {n + i - 3, n + j - 3};
                else if (s(a[i] + w + b[j])) q = {i - 1, 1, n + j - 3};
                else q = {n + i - 3, 1, j - 1};
            }
            cout << q.size();
            for (int v : q) cout << " " << v;
            cout << endl;
        }
    }
    return 0;
}
0