結果

問題 No.5021 Addition Pyramid
ユーザー square1001
提出日時 2025-02-25 20:50:09
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 727 ms / 2,000 ms
コード長 2,599 bytes
コンパイル時間 1,360 ms
コンパイル使用メモリ 85,836 KB
実行使用メモリ 6,820 KB
スコア 233,061,677
最終ジャッジ日時 2025-02-25 20:50:47
合計ジャッジ時間 37,009 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 50
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <vector>
#include <cstdint>
#include <iostream>
#include <algorithm>
using namespace std;
namespace myrandom {
    uint64_t seed = 1234567891234567891;
    uint64_t xorshift64() {
        seed ^= seed << 13;
        seed ^= seed >> 7;
        seed ^= seed << 17;
        return seed;
    }
    int next_int(int l, int r) {
        return l + int(xorshift64() % (r - l));
    }
}
const int INF = 1012345678;
const int D = 100'000'000;
int get_score(int N, const vector<vector<int> >& A, const vector<int>& ans) {
    vector<vector<int> > B(N);
    B[N - 1] = ans;
    for (int i = N - 2; i >= 0; i--) {
        B[i].resize(i + 1);
        for (int j = 0; j <= i; j++) {
            B[i][j] = (B[i + 1][j] + B[i + 1][j + 1]) % D;
        }
    }
    int score = -INF;
    for (int i = 0; i < N; i++) {
        for (int j = 0; j <= i; j++) {
            int d = abs(B[i][j] - A[i][j]);
            d = min(d, D - d);
            score = max(score, d);
        }
    }
    return D / 2 - score;
}
vector<int> solve(int N, const vector<vector<int> >& A) {
    const int repeats = 20000;
    int best_score = 0;
    vector<int> opt;
    for (int id = 1; id <= repeats; id++) {
        vector<vector<int> > B(N);
        for (int i = 0; i < N; i++) {
            B[i].resize(i + 1);
        }
        int error = D / 2 - best_score;
        B[N - 1][0] = (A[N - 1][0] + myrandom::next_int(-error / 2, +error / 2 + 1) + D) % D;
        bool fail = false;
        for (int i = 1; i < N; i++) {
            vector<int> v(i + 1);
            int target = 0;
            for (int j = i; j >= 0; j--) {
                if (j != i) {
                    target = (target + B[(N - i) + j][j]) % D;
                }
                v[j] = (A[(N - i - 1) + j][j] - target + D) % D;
            }
            sort(v.begin(), v.end());
            v.push_back(v[0] + D);
            pair<int, int> opt = {-INF, -1};
            for (int j = 0; j <= i; j++) {
                opt = max(opt, {v[j + 1] - v[j], j});
            }
            if (opt.first / 2 < best_score) {
                fail = true;
                break;
            }
            int margin = (opt.first / 2 - best_score) / 2;
            int center = ((v[opt.second] + v[opt.second + 1]) / 2 + D / 2 + myrandom::next_int(-margin, +margin + 1) + D) % D;
            B[N - 1][i] = center;
            for (int j = i - 1; j >= 0; j--) {
                B[(N - i - 1) + j][j] = (B[(N - i) + j][j] + B[(N - i) + j][j + 1]) % D;
            }
        }
        int score = get_score(N, A, B[N - 1]);
        if (!fail) {
            cerr << id << ": " << score << endl;
            best_score = score;
            opt = B[N - 1];
        }
    }
    return opt;
}
int main() {
    int N;
    cin >> N;
    vector<vector<int> > A(N);
    for (int i = 0; i < N; i++) {
        A[i].resize(i + 1);
        for (int j = 0; j <= i; j++) {
            cin >> A[i][j];
        }
    }
    vector<int> ans = solve(N, A);
    for (int i = 0; i < N; i++) {
        if (i != 0) {
            cout << ' ';
        }
        cout << ans[i];
    }
    cout << endl;
    cerr << "score: " << get_score(N, A, ans) << endl;
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0