結果

問題 No.1300 Sum of Inversions
ユーザー nullnull
提出日時 2020-11-28 00:56:56
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 124 ms / 2,000 ms
コード長 6,277 bytes
コンパイル時間 1,592 ms
コンパイル使用メモリ 152,284 KB
実行使用メモリ 10,512 KB
最終ジャッジ日時 2024-07-26 21:32:31
合計ジャッジ時間 6,013 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 95 ms
8,816 KB
testcase_04 AC 91 ms
8,632 KB
testcase_05 AC 75 ms
7,760 KB
testcase_06 AC 108 ms
9,464 KB
testcase_07 AC 104 ms
9,192 KB
testcase_08 AC 114 ms
9,904 KB
testcase_09 AC 115 ms
9,884 KB
testcase_10 AC 62 ms
6,972 KB
testcase_11 AC 60 ms
7,116 KB
testcase_12 AC 98 ms
8,784 KB
testcase_13 AC 92 ms
8,452 KB
testcase_14 AC 124 ms
10,324 KB
testcase_15 AC 120 ms
9,608 KB
testcase_16 AC 98 ms
9,056 KB
testcase_17 AC 63 ms
6,896 KB
testcase_18 AC 73 ms
7,432 KB
testcase_19 AC 84 ms
8,072 KB
testcase_20 AC 87 ms
8,128 KB
testcase_21 AC 83 ms
8,248 KB
testcase_22 AC 77 ms
7,752 KB
testcase_23 AC 109 ms
9,456 KB
testcase_24 AC 79 ms
7,832 KB
testcase_25 AC 65 ms
7,232 KB
testcase_26 AC 66 ms
7,324 KB
testcase_27 AC 73 ms
7,564 KB
testcase_28 AC 124 ms
9,856 KB
testcase_29 AC 81 ms
8,100 KB
testcase_30 AC 114 ms
9,744 KB
testcase_31 AC 75 ms
7,904 KB
testcase_32 AC 77 ms
7,988 KB
testcase_33 AC 69 ms
10,388 KB
testcase_34 AC 80 ms
10,384 KB
testcase_35 AC 77 ms
10,340 KB
testcase_36 AC 83 ms
10,512 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/*
このコード、と～おれ!
Be accepted!
∧＿∧　
（｡･ω･｡)つ━☆・*。
⊂　　 ノ 　　　・゜+.
　しーＪ　　　°。+ *´¨)
 　　　　　　　　　.· ´¸.·*´¨) ¸.·*¨)
          　　　　　　　　　　(¸.·´ (¸.·'* ☆
*/

#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstring>
#include <vector>
#include <numeric>
#include <iostream>
#include <random>
#include <map>
#include <unordered_map>
#include <queue>
#include <regex>
#include <functional>
#include <complex>
#include <list>
#include <cassert>
#include <iomanip>
#include <set>
#include <stack>
#include <bitset>
#include <array>

////多倍長整数, cpp_intで宣言
//#include <boost/multiprecision/cpp_int.hpp>
//#include <boost/multiprecision/cpp_dec_float.hpp>
//using namespace boost::multiprecision;
//
//#pragma GCC target ("avx2")
//#pragma GCC target("arch=skylake-avx512")
//#pragma GCC optimize ("O3")
//#pragma GCC target ("sse4")
//#pragma GCC optimize ("unroll-loops")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define repeat(i, n, m) for(int i = n; i < (m); ++i)
#define rep(i, n) for(int i = 0; i < (n); ++i)
#define printynl(a) printf(a ? "yes\n" : "no\n")
#define printyn(a) printf(a ? "Yes\n" : "No\n")
#define printYN(a) printf(a ? "YES\n" : "NO\n")
#define printim(a) printf(a ? "possible\n" : "imposible\n")
#define printdb(a) printf("%.50lf\n", a) //少数出力
#define printLdb(a) printf("%.50Lf\n", a) //少数出力
#define printdbd(a) printf("%.16lf\n", a) //少数出力(桁少なめ)
#define prints(s) printf("%s\n", s.c_str()) //string出力
#define all(x) (x).begin(), (x).end()
#define deg_to_rad(deg) (((deg)/360.0L)*2.0L*PI)
#define rad_to_deg(rad) (((rad)/2.0L/PI)*360.0L)
#define Please return
#define AC 0
#define manhattan_dist(a, b, c, d) (abs(a - c) + abs(b - d)) /*(a, b) から (c, d) のマンハッタン距離 */
#define inf numeric_limits<double>::infinity();
#define linf numeric_limits<long double>::infinity()

using ll = long long;
using ull = unsigned long long;

constexpr int INF = 1073741823;
constexpr int MINF = -1073741823;
constexpr ll LINF = ll(4661686018427387903);
constexpr ll MOD = 1e9 + 7;
constexpr ll mod = 998244353;
constexpr long double eps = 1e-6;
const long double PI = acosl(-1.0L);

using namespace std;

void scans(string & str) {
	char c;
	str = "";
	scanf("%c", &c);
	if (c == '\n')scanf("%c", &c);
	while (c != '\n' && c != -1 && c != ' ') {
		str += c;
		scanf("%c", &c);
	}
}

void scanc(char& str) {
	char c;
	scanf("%c", &c);
	if (c == -1)return;
	while (c == '\n') {
		scanf("%c", &c);
	}
	str = c;
}

double acot(double x) {
	return PI / 2 - atan(x);
}

ll LSB(ll n) { return (n & (-n)); }

template<typename T>
inline T chmin(T & a, const T & b) {
	if (a > b)a = b;
	return a;
}

template<typename T>
inline T chmax(T & a, const T & b) {
	if (a < b)a = b;
	return a;
}

//atcoder library
//#include <atcoder/all>
//using namespace atcoder;

//random_device seed_gen;
//mt19937 engine(seed_gen());
//uniform_distribution dist(-1.0, 1.0);

/*-----------------------------------------ここからコード-----------------------------------------*/

/*
* @title modint
* @docs kyopro/docs/modint.md
*/

template<int mod>
struct modint {
	int val;

	modint() : val(0) {};
	modint(ll x) : val(x >= 0 ? x % mod : (mod + x % mod) % mod) {};

	modint& operator=(const modint& x) {
		val = x.val;
		return *this;
	}

	modint& operator+=(const modint& x) {
		val += x.val;
		if (val >= mod)val -= mod;
		return *this;
	}

	modint& operator-=(const modint& x) {
		val += mod - x.val;
		if (val >= mod)val -= mod;
		return *this;
	}

	modint& operator*=(const modint& x) {
		val = (int)((ll)val * (ll)x.val % mod);
		return *this;
	}

	modint& operator/=(const modint& x) {
		int a = x.val, b = mod, u = 1, v = 0, t;
		while (b > 0) {
			t = a / b;
			swap(a -= t * b, b);
			swap(u -= t * v, v);
		}
		*this *= modint(u);
		return *this;
	}

	modint operator++() {
		val = (val + 1 == mod ? 0 : val + 1);
		return *this;
	}

	modint operator--() {
		val = (val == 0 ? mod - 1 : val - 1);
		return *this;
	}

	modint operator+(const modint& x) const {
		return (modint(*this) += x);
	}

	modint operator-(const modint& x) const {
		return (modint(*this) -= x);
	}

	modint operator*(const modint& x) const {
		return (modint(*this) *= x);
	}

	modint operator/(const modint& x) const {
		return (modint(*this) /= x);
	}

	bool operator==(const modint& x)const {
		return (val == x.val);
	}

	bool operator!=(const modint& x)const {
		return (val != x.val);
	}

	bool operator<(const modint& x)const {
		return (val < x.val);
	}

	bool operator>(const modint& x)const {
		return (val > x.val);
	}

	modint pow(ll n) {
		modint ret(1), a(val);
		while (n > 0) {
			if (n % 2) ret *= a;
			a *= a;
			n /= 2;
		}
		return ret;
	}

	static int getmod() { return mod; };
};

using ModInt = modint<MOD>;
using Modint = modint<mod>;

/*
* @title binary-indexed-tree
* @docs kyopro/docs/BIT.md
*/

template<typename T>
//0-indexed/内部的に 1-indexed
struct BIT {
	vector<T> tree;
	//初期化
	BIT(int n) : tree(n) {
		tree.assign(n + 1, 0);
	}

	int LSB(int n) { return (n & (-n)); }

	//[0, n) の sum を返す/0-indexed
	T sum(int n) {
		T ret = 0;
		//実は 1-indexed だが半開区間なので見た目がそのまま
		for (; n >= 0; n -= LSB(n)) {
			ret += tree[n];
			if (!n)break;
		}
		return ret;
	}


	//[i, j) の sum を返す/0-indexed
	T sum(int i, int j) {
		return sum(j) - sum(i);
	}

	//n 番目に x を足す
	void add(int n, T x) {
		int siz = tree.size();
		for (++n; n < siz; n += LSB(n))tree[n] += x;
	}
};



int main() {

	int n;
	scanf("%d", &n);
	vector<ll> a(n), b;
	rep(i, n)scanf("%lld", &a[i]);
	b = a;
	sort(all(b));
	b.erase(unique(all(b)), b.end());
	rep(i, n) a[i] = distance(b.begin(), lower_bound(all(b), a[i]));
	Modint ans = 0;
	BIT<Modint> bit1(n), bit2(n), bit3(n), bit4(n);
	rep(i, n) {
		ans += bit2.sum(a[i] + 1, n) + bit4.sum(a[i] + 1, n) * b[a[i]];
		bit2.add(a[i], bit1.sum(a[i] + 1, n) + bit3.sum(a[i] + 1, n) * b[a[i]]);
		bit4.add(a[i], bit3.sum(a[i] + 1, n));
		bit1.add(a[i], b[a[i]]);
		bit3.add(a[i], 1);
	}
	printf("%d\n", ans.val);

	Please AC;
}
0