結果

問題 No.1115 二つの数列 / Two Sequences
ユーザー noritake
提出日時 2020-08-05 22:40:33
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 52 ms / 2,000 ms
コード長 1,223 bytes
コンパイル時間 1,648 ms
コンパイル使用メモリ 170,008 KB
実行使用メモリ 6,656 KB
最終ジャッジ日時 2024-09-17 16:06:45
合計ジャッジ時間 3,524 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef vector<long long> vl;
#define INF __INT32_MAX__
#define LINF __LONG_LONG_MAX__

class BIT
{
    private:
        vl node;
        ll n;
    public:
        BIT(ll n_) {
            n = n_;
            node.resize(n + 1);
        }

        void add(ll i, ll x) {
            while (i <= n) {
                node[i] += x;
                i += (i & (-i));
            }
        }

        ll sum(ll i) {
            ll sum = 0;
            while (i > 0) {
                sum += node[i];
                i -= (i & (-i));
            }
            return sum;
        }

        ll sum(ll a, ll b) {
            return sum(b) - sum(a - 1);
        }
};

int main() {
    ll N;
    cin >> N;
    vl A(N + 1), B(N + 1);
    rep(i, N) {
        cin >> A[i];
    }
    rep(i, N) {
        cin >> B[i];
    }

    vl conv(N + 1);
    rep(i, N) {
        conv[B[i]] = i + 1;
    }
    rep(i, N) {
        A[i] = conv[A[i]];
    }

    ll cnt = 0;
    BIT bit(N);
    rep(i, N) {
        bit.add(A[i], 1);
        cnt += (i + 1) - bit.sum(A[i]);
    }

    cout << cnt << endl;
}
0