結果

問題 No.3164 [Chery 7th Tune B] La vie en rose
ユーザー elphe
提出日時 2025-05-31 09:24:31
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 117 ms / 2,000 ms
コード長 1,631 bytes
コンパイル時間 986 ms
コンパイル使用メモリ 87,908 KB
実行使用メモリ 11,428 KB
最終ジャッジ日時 2025-05-31 09:24:42
合計ジャッジ時間 8,879 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <cstdint>
#include <vector>

using namespace std;

template<typename T> static inline constexpr T chmax(T& variable, const T value) noexcept
{
	if (variable < value) return (variable = value);
	else return variable;
}

static inline constexpr vector<uint64_t> prepare(const uint32_t N, const vector<uint32_t>& A) noexcept
{
	vector<uint64_t> count_connection(N + 2, 0);
	for (uint32_t i = 1; i != N + 1; ++i)
	{
		if (A[i] != 0)
			count_connection[i] = count_connection[i - 1] + A[i];
		else
			count_connection[i] = 0;
	}

	for (uint32_t i = N - 1; i != UINT32_MAX; --i)
		if (A[i] != 0)
			chmax(count_connection[i], count_connection[i + 1]);

	return count_connection;
}

static inline constexpr vector<uint64_t> solve(const uint32_t N, const uint32_t Q, const vector<uint32_t>& A, const vector<uint32_t>& B, const vector<uint32_t>& X) noexcept
{
	const vector<uint64_t>& count_connection = prepare(N, A);
	vector<uint64_t> ans(Q);
	for (uint32_t i = 0; i != Q; ++i)
	{
		if (A[X[i]] == 0)
			ans[i] = count_connection[X[i] - 1] + count_connection[X[i] + 1] + B[i];
		else
			ans[i] = count_connection[X[i]] + (B[i] - A[X[i]]);
	}

	return ans;
}

static inline void output(const uint32_t Q, const vector<uint64_t>& ans) noexcept
{
	for (uint32_t i = 0; i != Q; ++i)
		cout << ans[i] << '\n';
}

int main()
{
	cin.tie(nullptr);
	ios::sync_with_stdio(false);

	uint32_t N, Q, i;
	cin >> N;
	vector<uint32_t> A(N + 1);
	for (i = 1; i != N + 1; ++i) cin >> A[i];

	cin >> Q;
	vector<uint32_t> X(Q), B(Q);
	for (i = 0; i != Q; ++i)
		cin >> X[i] >> B[i];

	output(Q, solve(N, Q, A, B, X));
	return 0;
}
0