結果

問題 No.1115 二つの数列 / Two Sequences
コンテスト
ユーザー kawara_y_kyopro
提出日時 2020-07-17 21:42:08
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 34 ms / 2,000 ms
コード長 2,378 bytes
コンパイル時間 1,114 ms
コンパイル使用メモリ 106,492 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-29 22:45:12
合計ジャッジ時間 3,019 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <iostream>
#include <array>
#include <algorithm>
#include <vector>
#include <bitset>
#include <set>
#include <unordered_set>
#include <cmath>
#include <complex>
#include <deque>
#include <iterator>
#include <numeric>
#include <map>
#include <unordered_map>
#include <queue>
#include <stack>
#include <string>
#include <tuple>
#include <utility>
#include <limits>
#include <iomanip>
#include <functional>
#include <cassert>
using namespace std;

using ll=long long;
template<class T> using V = vector<T>;
template<class T, class U> using P = pair<T, U>;
using vll = V<ll>;
using vvll = V<vll>;
#define rep(i, k, n) for (ll i=k; i<(ll)n; ++i)
#define REP(i, n) rep(i, 0, n)
#define ALL(v) v.begin(),v.end()
template < class T > inline bool chmax(T& a, T b) {if (a < b) { a=b; return true; } return false; }
template < class T > inline bool chmin(T& a, T b) {if (a > b) { a=b; return true; } return false; }
#define DEBUG_VLL(vec) REP(sz, vec.size()) std::cerr<<vec[sz]<<(sz==vec.size()-1?'\n':' ');

const long long MOD = 1000000007;
const long long HIGHINF = (long long)1e18;
const int INF = (int)1e9;

V<int> l, r;

ll merge(V<int>& a, int n, int left, int mid, int right) {
    int i, j, k;
    ll cnt = 0;
    int n1 = mid - left, n2 = right - mid;
    for (i = 0; i < n1; i++) l[i] = a[left + i];
    for (i = 0; i < n2; i++) r[i] = a[mid + i];
    l[n1] = r[n2] = INF;
    i = j = 0;
    for (k = left; k < right; k++) {
        if (l[i] <= r[j]) {
            a[k] = l[i++];
        } else {
            a[k] = r[j++];
            cnt += n1 - i;
        }
    }
    return cnt;
}

ll mergesort(V<int>& a, int n, int left, int right) {
    int mid;
    ll v1, v2, v3;
    if (left + 1 < right) {
        mid = (left + right) >> 1;
        v1 = mergesort(a, n, left, mid);
        v2 = mergesort(a, n, mid, right);
        v3 = merge(a, n, left, mid, right);
        return v1 + v2 + v3;
    } else return 0;
}

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    int n; cin >> n;
    V<int> a(n), b(n);
    for (int i = 0; i < n; i++) cin >> a[i];
    for (int i = 0; i < n; i++) {
        int tb; cin >> tb;
        b[(--tb)] = i;
    }
    V<int> aidx(n);
    for (int i = 0; i < n; i++) aidx[i] = b[a[i] - 1];
    l.resize((n + 1) / 2 + 2), r.resize((n + 1) / 2 + 2);
    ll ans = mergesort(aidx, n, 0, n);
    cout << ans << '\n';
    return 0;
}
0