結果

問題 No.2950 Max Min Product
ユーザー kotatsugamekotatsugame
提出日時 2024-10-25 21:45:00
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 386 ms / 3,000 ms
コード長 1,523 bytes
コンパイル時間 1,211 ms
コンパイル使用メモリ 91,264 KB
実行使用メモリ 18,492 KB
最終ジャッジ日時 2024-10-25 23:30:00
合計ジャッジ時間 13,928 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,820 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 243 ms
17,236 KB
testcase_04 AC 264 ms
17,632 KB
testcase_05 AC 254 ms
17,292 KB
testcase_06 AC 333 ms
18,476 KB
testcase_07 AC 333 ms
18,244 KB
testcase_08 AC 357 ms
18,404 KB
testcase_09 AC 175 ms
10,744 KB
testcase_10 AC 304 ms
17,868 KB
testcase_11 AC 261 ms
17,456 KB
testcase_12 AC 333 ms
18,400 KB
testcase_13 AC 333 ms
18,248 KB
testcase_14 AC 334 ms
18,492 KB
testcase_15 AC 347 ms
17,152 KB
testcase_16 AC 253 ms
16,048 KB
testcase_17 AC 278 ms
16,260 KB
testcase_18 AC 355 ms
17,420 KB
testcase_19 AC 355 ms
17,392 KB
testcase_20 AC 360 ms
17,376 KB
testcase_21 AC 201 ms
10,684 KB
testcase_22 AC 255 ms
16,392 KB
testcase_23 AC 174 ms
10,372 KB
testcase_24 AC 338 ms
17,356 KB
testcase_25 AC 331 ms
17,408 KB
testcase_26 AC 331 ms
17,280 KB
testcase_27 AC 260 ms
15,252 KB
testcase_28 AC 381 ms
16,876 KB
testcase_29 AC 273 ms
16,200 KB
testcase_30 AC 386 ms
15,776 KB
testcase_31 AC 375 ms
16,996 KB
testcase_32 AC 385 ms
15,780 KB
testcase_33 AC 272 ms
15,236 KB
testcase_34 AC 179 ms
9,588 KB
testcase_35 AC 263 ms
15,164 KB
testcase_36 AC 365 ms
15,820 KB
testcase_37 AC 363 ms
15,856 KB
testcase_38 AC 362 ms
15,772 KB
testcase_39 AC 2 ms
6,816 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<cassert>
#include<atcoder/modint>
#include<atcoder/lazysegtree>
using namespace std;
using mint=atcoder::modint998244353;
using dat=pair<mint,int>;
dat op(dat a,dat b)
{
	a.first+=b.first;
	a.second+=b.second;
	return a;
}
dat e(){return make_pair(mint::raw(0),0);}
dat mp(int f,dat x)
{
	if(f>=0)x.first=x.second*mint(f);
	return x;
}
int cmp(int f,int g){return f>=0?f:g;}
int id(){return -1;}
int N;
int A[2<<17];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cin>>N;
	for(int i=0;i<N;i++)cin>>A[i];
	mint ans=0,ret=0;
	atcoder::lazy_segtree<dat,op,e,int,mp,cmp,id>segmax(N),segmin(N);
	vector<pair<int,int> >mx,mn;
	mx.push_back(make_pair(N,(int)1e9));
	mn.push_back(make_pair(N,0));
	for(int i=N;i--;)
	{
		segmax.set(i,make_pair(A[i],1));
		segmin.set(i,make_pair(A[i],1));
		{//max
			while(mx.back().second<A[i])
			{
				int l=mx.back().first;
				int a=mx.back().second;
				mx.pop_back();
				ans-=segmin.prod(l,mx.back().first).first*a;
			}
			segmax.apply(i,mx.back().first,A[i]);
			ans+=segmin.prod(i+1,mx.back().first).first*A[i];
			mx.push_back(make_pair(i,A[i]));
		}
		{//min
			while(mn.back().second>A[i])
			{
				int l=mn.back().first;
				int a=mn.back().second;
				mn.pop_back();
				ans-=segmax.prod(l,mn.back().first).first*a;
			}
			segmin.apply(i,mn.back().first,A[i]);
			ans+=segmax.prod(i+1,mn.back().first).first*A[i];
			mn.push_back(make_pair(i,A[i]));
		}
		ans+=mint(A[i])*A[i];
		ret+=ans;
	}
	cout<<ret.val()<<"\n";
}
0