結果

問題 No.1115 二つの数列 / Two Sequences
ユーザー umezo
提出日時 2020-07-19 02:29:23
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 111 ms / 2,000 ms
コード長 1,065 bytes
コンパイル時間 2,905 ms
コンパイル使用メモリ 192,392 KB
最終ジャッジ日時 2025-01-12 00:41:15
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define ALL(v) v.begin(), v.end()
typedef long long ll;

#include <bits/stdc++.h>
using namespace std;

const ll MAX=500000;
const ll INFTY=1000000010;

ll L[MAX/2+2],R[MAX/2+2];

ll merge(int A[], int left, int mid ,int right){
  int n1=mid-left;
  int n2=right-mid;
  ll cnt=0;
  rep(i,n1) L[i]=A[left+i];
  rep(i,n2) R[i]=A[mid+i];
  L[n1]=INFTY;
  R[n2]=INFTY;
  int i=0,j=0;
  for(int 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(int A[],int left,int right){
  int mid;
  ll v1,v2,v3;
  if(left+1<right){
    mid=(left + right)/2;
    v1=mergeSort(A,left,mid);
    v2=mergeSort(A,mid,right);
    v3=merge(A,left,mid,right);
    return v1+v2+v3;
  }else return 0;
}

int main() {
  int n;
  cin>>n;
  
  int A[MAX],B[MAX],C[MAX];
  rep(i,n) cin>>A[i];
  rep(i,n) cin>>B[i];
  rep(i,n) C[B[i]]=i;
  rep(i,n){
    int tmp=C[A[i]];
    A[i]=tmp+1;
  }
  
  ll ans=mergeSort(A,0,n);
  cout<<ans<<endl;
  
  return 0;
}
0