結果

問題 No.5021 Addition Pyramid
コンテスト
ユーザー square1001
提出日時 2025-02-25 21:49:40
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,210 ms / 2,000 ms
コード長 3,705 bytes
コンパイル時間 1,115 ms
コンパイル使用メモリ 87,840 KB
実行使用メモリ 6,820 KB
スコア 278,602,591
最終ジャッジ日時 2025-02-25 21:50:44
合計ジャッジ時間 62,194 ms
ジャッジサーバーID
（参考情報） 		judge2 / 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_error(int a, int b) {
	int d = abs(a - b);
	return min(d, D - d);
}

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;
}

int get_opt_1(int K, vector<int> A) {
	sort(A.begin(), A.end());
	A.push_back(A[0] + D);
	pair<int, int> opt = {-INF, -1};
	for (int i = 0; i < K; i++) {
		opt = max(opt, {A[i + 1] - A[i], i});
	}
	return ((A[opt.second] + A[opt.second + 1]) / 2 + D / 2) % D;
}

int get_opt_2(int K, vector<int> A, vector<int> B) {
	int opt_error = D / 2 + 1;
	int opt_x = -1;
	for (int id = 1; id <= 1000; id++) {
		int x = myrandom::next_int(0, D);
		int error = 0;
		for (int i = 0; i < K; i++) {
			error = max(error, get_error(x, A[i]));
		}
		vector<int> C(K + 1);
		for (int i = 0; i < K + 1; i++) {
			C[i] = (B[i] - 1LL * (K - i) * x % D + D) % D;
		}
		sort(C.begin(), C.end());
		C.push_back(C[0] + D);
		int z = 0;
		for (int i = 0; i < K + 1; i++) {
			z = max(z, C[i + 1] - C[i]);
		}
		error = max(error, D / 2 - z / 2);
		if (opt_error > error) {
			opt_error = error;
			opt_x = x;
		}
	}
	return opt_x;
}

vector<int> solve(int N, const vector<vector<int> >& A) {
	const int repeats = 30;
	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++) {
			for (int j = 0; j < (i != N - 1 ? 2 : 1); j++) {
				for (int k = i - 1 + j; k >= 0; k--) {
					B[(N - i - 1 - j) + k][k] = (B[(N - i - j) + k][k] + B[(N - i - j) + k][k + 1]) % D;
				}
			}
			int center = 0;
			if (i == N - 1) {
				vector<int> e1(i + 1);
				for (int j = 0; j <= i; j++) {
					e1[j] = (A[(N - i - 1) + j][j] - B[(N - i - 1) + j][j] + D) % D;
				}
				center = get_opt_1(i + 1, e1);
			} else {
				vector<int> e1(i + 1), e2(i + 2);
				for (int j = 0; j <= i; j++) {
					e1[j] = (A[(N - i - 1) + j][j] - B[(N - i - 1) + j][j] + D) % D;
				}
				for (int j = 0; j <= i + 1; j++) {
					e2[j] = (A[(N - i - 2) + j][j] - B[(N - i - 2) + j][j] + D) % D;
				}
				center = get_opt_2(i + 1, e1, e2);
			}
			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 (best_score < score) {
			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