結果

問題 No.1300 Sum of Inversions
ユーザー queeequeee
提出日時 2020-11-27 22:20:18
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,156 bytes
コンパイル時間 1,312 ms
コンパイル使用メモリ 126,592 KB
実行使用メモリ 12,672 KB
最終ジャッジ日時 2024-07-26 13:20:43
合計ジャッジ時間 9,437 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 AC 128 ms
12,544 KB
testcase_34 AC 185 ms
12,672 KB
testcase_35 WA -
testcase_36 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

//g++ t.cpp -o t && t < p.txt
//d>p.txt&&t<p.txt&&t2<p.txt

#include <iostream> // プログラムの耳と口
#include <vector> // 超有能配列秘書
#include <algorithm> // みんな大好きソートと二分探索
#include <queue> // きゅーちゃん、だいすき
#include <string> // to_string
#include <functional> // 関数を変数に入れる子
#include <set> // 値の取得・挿入・削除を高速に
#include <map> // setの妹 これまた優秀
#include <random> // 乱択さん
#include <ctime> // へびになりたい
#include <bitset> // へびになった ｼｬｰ
#include <cassert>

// #include <atcoder/all>
// using namespace atcoder; // コンパイル重い
// g++ t.cpp -o t -I .

#define DB cerr<<"D"<<endl

using namespace std; using ll=long long; using ld=long double; const int INF=1e9; const ll LINF=1e18; const double dINF = 1e18; const ld ldINF = 1e18; const double EPS = 1e-6;
using P=pair<ll,ll>;
const ll M = 1000000007;
ll mod_pow(ll x, ll a) { ll an = 1; while(a > 0) { if (a&1) an = an * x % M; x = x * x % M; a >>= 1;} return an;}
ll mod_pow(ll x, ll a, ll m) { ll an = 1; while(a > 0) { if (a&1) an = an * x % m; x = x * x % m; a >>= 1;} return an;}
void add(ll& x, ll y) {x+=y; x%=M;}; void mul(ll& x, ll y) {x*=y; x%=M;}; template<typename T, typename U> void chmax(T& x, U y) {if (x<y) x=y;}; template<typename T, typename U> void chmin(T& x, U y) {if (x>y) x=y;} bool vaild(int x, int y, int hh, int ww){return 0<=x&&x<hh&&0<=y&&y<ww;}
ll gcd(ll a, ll b) {if (b==0) return a; else return gcd(b, a%b);}
int keta(ll a) {int res=0; while(a>0) res+=a%10, a/=10; return res;}
const int up[]={1,-1,0,0}, lf[]={0,0,1,-1};

struct BIT { // 0-index
  int n; vector<ll> v;
  BIT(int m) : n(m) {v.resize(m);}
  ll sum(int i) {
    //cout<<"sum "<<i<<" "<<n<<" ";
    if (i<0 || i>=n) return 0;
    ll res=0; for(; i>=0; i=(i&(i+1))-1) res+=v[i]; res%=M; return res;
  }
  ll sum(int a, int b) {
    return (sum(b) - sum(a-1) + M )% M ;
  }
  void add(int i, ll x) {for(;i<n;i|=i+1) v[i]+=x,v[i]%=M;}
  void print() {
    for(int i=0;i<n;i++) cout<<sum(i)-sum(i-1)<<" "; cout<<endl;
  }
};

int main() {
  int n; cin>>n;
  ll a[n]; for(int i=0;i<n;i++) cin>>a[i];
  ll b[n]; copy(a,a+n,b); sort(b,b+n);
  BIT b1(n),b2(n),c1(n),c2(n);
  for(int i=0;i<n;i++) {
    int id = lower_bound(b,b+n,a[i])-b;
    b2.add(id,1); c2.add(id,a[i]);
  }
  ll an=0;
  for(int i=0;i<n;i++) {
    int id = lower_bound(b,b+n,a[i])-b;
    b1.add(id,1);
    c1.add(id,a[i]);
    b2.add(id,-1);
    c2.add(id,-a[i]);

    ll nu = b2.sum(0,id-1) * b1.sum(id+1,n-1) % M;

    //cout << "@ id " << id << " a " << a[i] << " " << nu << " " << c2.sum(0,id-1) << " " << c1.sum(id+1,n-1) << endl;
    //b1.print(); b2.print(); c1.print(); c2.print();

    an += a[i] * nu % M;
    an += (b1.sum(id+1,n-1) * c2.sum(0,id-1) % M  + b2.sum(0,id-1) * c1.sum(id+1,n-1) % M) % M;
    an %= M;
  }
  cout<<an<<endl;
}

// ・配列の大きさok？　・priority_queueはgreater？　・debug消した？
// ・落ち着いて。提出まで4分待ってね……WJ……1/10……2/10……

// Thank you for making problems and running the contest
0