結果

問題 No.1300 Sum of Inversions
ユーザー ttkkggwwttkkggww
提出日時 2020-11-28 05:00:08
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,074 ms / 2,000 ms
コード長 1,441 bytes
コンパイル時間 2,739 ms
コンパイル使用メモリ 215,700 KB
実行使用メモリ 43,800 KB
最終ジャッジ日時 2023-10-09 23:29:43
合計ジャッジ時間 27,074 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,348 KB
testcase_01 AC 1 ms
4,352 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 812 ms
34,456 KB
testcase_04 AC 787 ms
33,388 KB
testcase_05 AC 605 ms
28,488 KB
testcase_06 AC 892 ms
38,284 KB
testcase_07 AC 811 ms
36,512 KB
testcase_08 AC 943 ms
40,056 KB
testcase_09 AC 915 ms
39,784 KB
testcase_10 AC 438 ms
24,180 KB
testcase_11 AC 456 ms
24,548 KB
testcase_12 AC 754 ms
33,604 KB
testcase_13 AC 695 ms
33,036 KB
testcase_14 AC 1,074 ms
42,880 KB
testcase_15 AC 927 ms
39,724 KB
testcase_16 AC 758 ms
35,028 KB
testcase_17 AC 420 ms
23,312 KB
testcase_18 AC 499 ms
26,280 KB
testcase_19 AC 618 ms
30,156 KB
testcase_20 AC 666 ms
30,924 KB
testcase_21 AC 636 ms
30,792 KB
testcase_22 AC 561 ms
28,392 KB
testcase_23 AC 894 ms
38,056 KB
testcase_24 AC 615 ms
29,100 KB
testcase_25 AC 518 ms
25,728 KB
testcase_26 AC 476 ms
25,496 KB
testcase_27 AC 537 ms
27,724 KB
testcase_28 AC 998 ms
40,792 KB
testcase_29 AC 688 ms
30,600 KB
testcase_30 AC 952 ms
39,760 KB
testcase_31 AC 611 ms
28,644 KB
testcase_32 AC 623 ms
29,380 KB
testcase_33 AC 199 ms
15,444 KB
testcase_34 AC 253 ms
15,576 KB
testcase_35 AC 519 ms
43,800 KB
testcase_36 AC 542 ms
43,620 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