結果

問題 No.1209 XOR Into You
ユーザー milanis48663220
提出日時 2020-09-01 00:29:33
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 284 ms / 2,000 ms
コード長 1,396 bytes
コンパイル時間 1,048 ms
コンパイル使用メモリ 98,752 KB
実行使用メモリ 76,928 KB
最終ジャッジ日時 2024-11-17 03:00:28
合計ジャッジ時間 9,309 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <algorithm>
#include <iomanip>
#include <vector>
#include <queue>
#include <set>
#include <map>

using namespace std;
typedef long long ll;

int N;
int A[100001], B[100001];

template <typename T>
struct bit{
    int n;
    vector<T> data;

    bit(int n_){
        n = 1;
        while(n < n_) n *= 2;
        data = vector<T>(n+1);
        for(int i = 0; i <= n; i++) data[i] = 0;
    }
    
    T sum(int i){
        T ret = 0;
        while(i > 0){
            ret += data[i];
            i -= i&-i;
        }
        return ret;
    }

    void add(int i, T x){
        while(i <= n){
            data[i] += x;
            i += i&-i;
        }
    }
};

map<int, queue<int>> mp;

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout << setprecision(10) << fixed;
    cin >> N;
    for(int i = 0; i < N; i++) {
        cin >> A[i];
        if(i >= 1){
            mp[A[i]^A[i-1]].push(i);
        }
    }
    for(int i = 0; i < N; i++) cin >> B[i];
    if(A[0] != B[0]){
        cout << -1 << endl;
        return 0;
    }
    ll ans = 0;
    bit<ll> bt(N);
    for(int i = 1; i < N; i++){
        int b = B[i]^B[i-1];
        if(mp[b].empty()){
            cout << -1 << endl;
            return 0;
        }
        int idx = mp[b].front(); mp[b].pop();
        ans += (i-1-bt.sum(idx));
        bt.add(idx, 1);
    }
    cout << ans << endl;
}
0