結果

問題 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;
}
0