結果

問題 No.2873 Kendall's Tau
ユーザー テナガザルテナガザル
提出日時 2024-09-06 23:16:43
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 755 ms / 4,500 ms
コード長 2,802 bytes
コンパイル時間 1,703 ms
コンパイル使用メモリ 137,408 KB
実行使用メモリ 14,856 KB
最終ジャッジ日時 2024-09-06 23:17:32
合計ジャッジ時間 12,912 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 1 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 622 ms
10,452 KB
testcase_08 AC 755 ms
14,856 KB
testcase_09 AC 629 ms
10,112 KB
testcase_10 AC 726 ms
14,608 KB
testcase_11 AC 655 ms
11,136 KB
testcase_12 AC 676 ms
13,912 KB
testcase_13 AC 174 ms
6,940 KB
testcase_14 AC 546 ms
12,472 KB
testcase_15 AC 96 ms
6,940 KB
testcase_16 AC 98 ms
6,940 KB
testcase_17 AC 451 ms
10,368 KB
testcase_18 AC 336 ms
8,192 KB
testcase_19 AC 441 ms
10,412 KB
testcase_20 AC 98 ms
6,940 KB
testcase_21 AC 357 ms
9,264 KB
testcase_22 AC 147 ms
6,944 KB
testcase_23 AC 362 ms
9,552 KB
testcase_24 AC 43 ms
6,944 KB
testcase_25 AC 88 ms
6,940 KB
testcase_26 AC 451 ms
10,648 KB
testcase_27 AC 270 ms
7,808 KB
testcase_28 AC 554 ms
11,520 KB
testcase_29 AC 613 ms
12,672 KB
testcase_30 AC 69 ms
6,944 KB
testcase_31 AC 135 ms
6,944 KB
testcase_32 AC 353 ms
8,576 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <map>
#include <cmath>
#include <algorithm>

using namespace std;

template <typename T> class segTree
{
private:
  const T e;
  int num;
  std::vector<T> dat;
  T (*const eval)(T &, T &) {};
public:
  segTree(std::vector<T> &v, T E, T (*func)(T &, T &)) : e(E), eval(func)
  {
    int siz = static_cast<int>(v.size());
    for (num = 1; num < siz; num <<= 1);
    dat = std::vector<T> (2 * num - 1, e);
    for (int i = 0; i < siz; ++i) dat[i + num - 1] = v[i];
    for (int i = num - 2; i >= 0; --i) dat[i] = eval(dat[i * 2 + 1], dat[i * 2 + 2]);
  }
  segTree(int n, T E, T (*func)(T &, T &)) : e(E), eval(func)
  {
    for (num = 1; num < n; num <<= 1);
    dat = std::vector<T> (2 * num - 1, e);
  }
  void update_a(int i, T val)
  {
    for (i += num - 1, dat[i] = val; i != 0;)
    {
      i = (i - 1) / 2;
      dat[i] = eval(dat[i * 2 + 1], dat[i * 2 + 2]);
    }
  }
  void update_r(int i, T val)
  {
    for (i += num - 1, dat[i] = eval(dat[i], val); i != 0;)
    {
      i = (i - 1) / 2;
      dat[i] = eval(dat[i * 2 + 1], dat[i * 2 + 2]);
    }
  }
  T getval(int left, int right)
  {
    left = max(0, left), right = min(num, right);
    T ansl = e, ansr = e;
    for (left += num - 1, right += num - 1; left < right; left >>= 1, right >>= 1)
    {
      if (!(left & 1)) ansl = eval(ansl, dat[left]);
      if (--right & 1) ansr = eval(dat[right], ansr);
    }
    return eval(ansl, ansr);
  }
  T getval(int id) {return dat[num - 1 + id];}
};

int main()
{
  int n;
  cin >> n;
  vector<int> x(n), y(n);
  for (int i = 0; i < n; ++i) cin >> x[i] >> y[i];
  auto cal = [&](vector<int> &a) -> long long
  {
    long long ret = 0;
    map<int, int> mp;
    for (int i = 0; i < n; ++i) ++mp[a[i]];
    for (int i = 0; i < n; ++i) ret += n - mp[a[i]];
    return ret / 2;
  };
  long long r = cal(x), s = cal(y);
  vector<pair<int, int>> xy(n);
  auto cal2 = [&]() -> long long
  {
    vector<int> all;
    for (auto [a, b] : xy) all.push_back(b);
    sort(all.begin(), all.end());
    all.erase(unique(all.begin(), all.end()), all.end());
    int siz = all.size();
    segTree<int> sg(siz, 0, [](int &a, int &b) -> int {return a + b;});
    long long ret = 0;
    for (auto [a, b] : xy)
    {
      int id = lower_bound(all.begin(), all.end(), b) - all.begin();
      ret += sg.getval(0, id);
      sg.update_r(id, 1);
    }
    return ret;
  };
  for (int i = 0; i < n; ++i) xy[i] = {x[i], -y[i]};
  sort(xy.begin(), xy.end());
  for (int i = 0; i < n; ++i) xy[i].second *= -1;
  long long p = cal2();
  for (int i = 0; i < n; ++i) xy[i] = {-x[i], -y[i]};
  sort(xy.begin(), xy.end());
  for (int i = 0; i < n; ++i) xy[i].second *= -1;
  long long q = cal2();
  double ans = (p - q) / sqrt((double)r * (double)s);
  printf("%.10f\n", ans);
}
0