結果

問題 No.1209 XOR Into You
ユーザー ir5
提出日時 2024-06-17 22:23:28
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 381 ms / 2,000 ms
コード長 2,797 bytes
コンパイル時間 2,271 ms
コンパイル使用メモリ 155,896 KB
実行使用メモリ 17,408 KB
最終ジャッジ日時 2024-06-17 22:23:44
合計ジャッジ時間 15,557 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <cassert>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <iostream>
#include <numeric>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <functional>
#include <iomanip>
using namespace std;
using ll = long long;

class range {private: struct I{int x;int operator*(){return x;}bool operator!=(I& lhs){return x<lhs.x;}void operator++(){++x;}};I i,n;
public:range(int n_):i({0}),n({n_}){}range(int i_,int n_):i({i_}),n({n_}){}I& begin(){return i;}I& end(){return n;}};


template<class T, class U> ostream& operator<<(ostream& os, const pair<T, U>& p){ return os << "{" << p.first << ", " << p.second << "}"; }
template<typename T> ostream& operator<<(ostream& os, const vector<T>& obj) { os << "{"; for (const auto& e : obj) os << e << ", "; return os << "}"; }
template<typename T> ostream& operator<<(ostream& os, const set<T>& obj) { os << "set{"; for (const auto& e : obj) os << e << ", "; return os << "}"; }
template<typename T, typename U> ostream& operator<<(ostream& os, const map<T, U>& obj) { os << "map{"; for (const auto& e : obj) os << e << ", "; return os << "}"; }

#ifdef ONLINE_JUDGE
#define dump(expr) ;
#else
#define dump(expr) { cerr << "\033[33m#L" << __LINE__ << ": " << expr << "\033[39m" << endl; }
#endif

template<typename T>
struct BinaryIndexedTree {
  int size;
  vector<T> ft;

  BinaryIndexedTree(int size_) : size(size_), ft(size, (T)0) {}

  void add(int i, T value) {
    // val[i] += value
    for (; i < size; i |= i + 1)
      ft[i] += value;
  }

  T sum(int i) {
    // val[0] + ... + val[i]
    T s = 0;
    for (; i >= 0; i = (i & (i + 1)) - 1)
      s += ft[i];
    return s;
  }
};

ll compute_inversion_number(vector<ll> vs) {
  // vs must be a permutation of (0, 1, ..., n - 1)
  int n = static_cast<int>(vs.size());
  BinaryIndexedTree<ll> bt(n);
  ll res = 0LL;
  for (int i = n - 1; i >= 0; --i) {
    res += bt.sum(vs[i]);
    bt.add(vs[i], 1);
  }
  return res;
}

ll solve() {
  int n; cin >> n;
  vector<ll> vs(n), vt(n);
  for (int i : range(n)) cin >> vs[i];
  for (int i : range(n)) cin >> vt[i];

  if (vs[0] != vt[0]) return -1;
  if (vs[n - 1] != vt[n - 1]) return -1;

  for (int i : range(n - 1)) vs[i] ^= vs[i + 1], vt[i] ^= vt[i + 1];
  vs.resize(n - 1);
  vt.resize(n - 1);
  n--;

  map<pair<int, int>, int> mp;
  map<int, int> count;
  for (int i : range(n)) {
    int t = vt[i];
    int c = count[t];
    mp[{t, c}] = i;
    count[t]++;
  }

  count.clear();
  for (int i : range(n)) {
    int s = vs[i];
    int c = count[s];
    if (!mp.count({s, c})) return -1;
    vs[i] = mp[{s, c}];
    count[s]++;
  }

  dump(vs);
  return compute_inversion_number(vs);
}

int main() {
  cout << fixed << setprecision(12);
  cout << solve() << endl;
}
0