結果

問題 No.1282 Display Elements
ユーザー saxofone111
提出日時 2021-03-06 16:14:08
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 161 ms / 2,000 ms
コード長 1,943 bytes
コンパイル時間 3,034 ms
コンパイル使用メモリ 200,504 KB
最終ジャッジ日時 2025-01-19 12:25:20
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"

#define MOD 1000000007
#define rep(i, n) for(ll i=0; i < (n); i++)
#define rrep(i, n) for(ll i=(n)-1; i >=0; i--)
#define ALL(v) v.begin(),v.end()
#define rALL(v) v.rbegin(),v.rend()
#define FOR(i, j, k) for(ll i=j;i<k;i++)
#define debug_print(var) cerr << #var << "=" << var <<endl;
#define DUMP(i, v)for(ll i=0;i<v.size();i++)cerr<<v[i]<<" "
#define fi first
#define se second

using namespace std;
typedef long long int ll;
typedef vector<ll> llvec;
typedef vector<double> dvec;
typedef pair<ll, ll> P;
typedef long double ld;
struct edge{ll x, c;};

struct segment_tree{
  ll N;
  llvec v;
  ll init=0;//initial value
  ll f(ll a, ll b){ //function
    return a+b;
  }
  segment_tree(ll n){
    N=1;
    while(N<n){
      N*=2;
    }
    v = llvec(2*N-1, init);
  }
  
  void set(ll i, ll val){
    i += N-1;
    v[i] = val;
    while(i>0){
      i = (i-1)/2;
      v[i] = f(v[i*2+1], v[i*2+2]);
    }
  }
  void add(ll i, ll val){
    i += N-1;
    v[i] += val;
    while(i>0){
      i = (i-1)/2;
      v[i] = f(v[i*2+1], v[i*2+2]);
    }
  }

  ll get(ll L, ll R){// L <= i < R
    L += N-1;
    R += N-1;
    ll vl = init;
    ll vr = init;
    while(L<R){
      if(L%2==0){
        vl = f(vl, v[L]);
        L++;
      }
      if(R%2==0){
        vr = f(vr, v[R-1]);
        R--;
      }
      R=(R-1)/2;
      L=(L-1)/2;
    }
    return f(vl, vr);
  }

  ll operator[](ll i){
    return v[i+N-1];
  }
  
};

/**************************************
** A main function starts from here  **
***************************************/
int main(){
  ll N;
  cin >> N;
  llvec a(N), b(N);
  rep(i, N)cin >> a[i];
  rep(i, N)cin >> b[i];
  sort(ALL(a));
  llvec c=b;
  sort(ALL(b));
  segment_tree sg(N);
  ll ans = 0;
  rep(i, N){
    ll k = lower_bound(ALL(b), c[i]) - b.begin();
    ll j = lower_bound(ALL(b), a[i]) - b.begin();
    sg.add(k, 1);
    ans += sg.get(0, j);
  }
  cout << ans << endl;
  return 0;
}
0