結果

問題 No.1300 Sum of Inversions
ユーザー ttkkggwwttkkggww
提出日時 2020-11-28 05:00:08
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 965 ms / 2,000 ms
コード長 1,441 bytes
コンパイル時間 3,338 ms
コンパイル使用メモリ 218,412 KB
実行使用メモリ 43,904 KB
最終ジャッジ日時 2024-09-12 22:01:36
合計ジャッジ時間 25,284 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 698 ms
34,572 KB
testcase_04 AC 642 ms
33,692 KB
testcase_05 AC 522 ms
28,672 KB
testcase_06 AC 830 ms
38,292 KB
testcase_07 AC 770 ms
36,840 KB
testcase_08 AC 863 ms
40,392 KB
testcase_09 AC 850 ms
40,064 KB
testcase_10 AC 404 ms
24,320 KB
testcase_11 AC 420 ms
24,500 KB
testcase_12 AC 691 ms
33,888 KB
testcase_13 AC 642 ms
33,156 KB
testcase_14 AC 965 ms
43,280 KB
testcase_15 AC 857 ms
39,936 KB
testcase_16 AC 703 ms
35,328 KB
testcase_17 AC 390 ms
23,576 KB
testcase_18 AC 464 ms
26,496 KB
testcase_19 AC 562 ms
30,492 KB
testcase_20 AC 612 ms
31,104 KB
testcase_21 AC 602 ms
30,976 KB
testcase_22 AC 500 ms
28,544 KB
testcase_23 AC 808 ms
38,320 KB
testcase_24 AC 543 ms
29,312 KB
testcase_25 AC 448 ms
25,728 KB
testcase_26 AC 433 ms
25,344 KB
testcase_27 AC 506 ms
27,800 KB
testcase_28 AC 895 ms
41,088 KB
testcase_29 AC 601 ms
30,764 KB
testcase_30 AC 856 ms
39,976 KB
testcase_31 AC 541 ms
28,708 KB
testcase_32 AC 553 ms
29,568 KB
testcase_33 AC 201 ms
15,756 KB
testcase_34 AC 258 ms
15,756 KB
testcase_35 AC 514 ms
43,904 KB
testcase_36 AC 540 ms
43,904 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
//#include<atcoder/all>
using namespace std;
using ll = long long;

ll MOD = 998244353;

class BIT{
	public:
	vector<ll>tree;
	int n;
	BIT(int n){
		this->n = n;
		tree.assign(n+1,0);
	}
	void add(int a,ll w){
		a++;
		for(int x=a;x<=n;x += x&-x)(tree[x] += w)%=MOD;
	}
	ll sum(int a){
		if(a==0)return 0;
		ll ret = 0;
		for(int x = a; x>0; x -= x&-x)(ret+=tree[x])%=MOD;
		return ret;
	}
};

int main()
{
	int n;
	cin >> n;
	vector<ll> a(n);
	map<int,int> atoidx,idxtoa;
	vector<ll> l(n),x(n),r(n),y(n);
	for(auto &i:a)cin >> i;
	{
		set<int> zaatu;
		for(auto &i:a)zaatu.insert(i);
		int idx = 0;
		for(auto &i:zaatu){
			atoidx[i]=idx;
			idxtoa[idx]=i;
			idx++;
		}
		BIT bitlr(n),bitxy(n);
		for(int i = 0;i<n;i++){
			ll idx = n-1-atoidx[a[i]];
			//cout<<idx<<endl;
			bitlr.add(idx,1);
			bitxy.add(idx,a[i]);
			l[i]=bitlr.sum(idx);
			x[i]=bitxy.sum(idx);
		}
		//for(auto &i:l)cout<<i<<' ';cout<<endl;
		//for(auto &i:x)cout<<i<<' ';cout<<endl;
		bitlr = BIT(n);bitxy=BIT(n);
		for(int i = n-1;i>=0;i--){
			ll idx = atoidx[a[i]];
			bitlr.add(idx,1);
			bitxy.add(idx,a[i]);
			r[i]=bitlr.sum(idx);
			y[i]=bitxy.sum(idx);
		}
		//for(auto &i:r)cout<<i<<' ';cout<<endl;
		//for(auto &i:y)cout<<i<<' ';cout<<endl;
	}
	ll ans = 0;

	for(int i = 0;i<n;i++){
		if(l[i]==0||r[i]==0)continue;
		(ans += x[i]*r[i])%=MOD;
		(ans += y[i]*l[i])%=MOD;
		(ans += a[i]*(l[i]*r[i]%MOD))%=MOD;
	}

	cout<<ans<<endl;
}
0