結果

問題 No.2950 Max Min Product
ユーザー Carpenters-Cat
提出日時 2024-10-25 23:14:38
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 743 bytes
コンパイル時間 4,442 ms
コンパイル使用メモリ 259,348 KB
最終ジャッジ日時 2025-02-24 23:53:48
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 4 WA * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
int N;
using Vi = std::vector<int> ;
using VVi = vector<Vi>;
using Vm = vector<mint>;
int main () {
	cin >> N;
	vector<int> A(N);
	for (auto& a : A) cin >> a;
	vector<int> ID(N);
	iota(ID.begin(), ID.end(), 0);
	sort(ID.begin(), ID.end(), [&](int a, int b){return A[a] > A[b];});
	set<int> st;
	st.insert(-1);
	st.insert(N);
	vector<mint> sum(N + 1, 0);
	for (int i = 0; i < N; i ++) {
		sum[i + 1] = sum[i] + A[i];
	}
	mint ans = 0;
	for (auto& i : ID) {
		mint pl = A[i];
		int l = *prev(st.lower_bound(i)) + 1, r = *st.lower_bound(i);
		ans += pl * (sum[r] - sum[l]);
		st.insert(i);
	}
	cout << ans.val() << endl;
}
0