結果

問題 No.743 Segments on a Polygon
コンテスト
ユーザー beet
提出日時 2018-10-05 22:33:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,552 bytes
コンパイル時間 1,921 ms
コンパイル使用メモリ 183,784 KB
実行使用メモリ 102,684 KB
最終ジャッジ日時 2024-10-12 13:19:01
合計ジャッジ時間 24,629 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 1 TLE * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using Int = signed;
using T = Int;
struct BinaryIndexedTree
{
  vector< T > data;

  BinaryIndexedTree(Int sz)
  {
    data.assign(++sz, 0);
  }

  T sum(Int k)
  {
    T ret = 0;
    for(++k; k > 0; k -= k & -k) ret += data[k];
    return (ret);
  }

  void add(Int k, T x)
  {
    for(++k; k < (Int)data.size(); k += k & -k) data[k] += x;
  }
};


struct SegmentTreeBIT
{
  Int sz;
  vector< BinaryIndexedTree > seg;
  vector< vector< Int > > beet;

  SegmentTreeBIT(Int n)
  {
    sz = 1;
    while(sz < n) sz <<= 1;
    beet.assign(2 * sz, vector<Int>());
  }

  void update(Int x, Int y, Int z)
  {
    x += sz;    
    while(x) {
      seg[x].add(lower_bound(begin(beet[x]), end(beet[x]), y) - begin(beet[x]), z);
      Int k=lower_bound(begin(beet[x]), end(beet[x]), y) - begin(beet[x]);
      assert(beet[x][k]==y);
      x>>=1;
    }
  }

  void build()
  {
    seg.reserve(sz * 2);
    for(Int k = 0; k < sz * 2; k++) {
      sort(begin(beet[k]), end(beet[k]));
      beet[k].erase(unique(begin(beet[k]), end(beet[k])), end(beet[k]));
      seg.emplace_back(BinaryIndexedTree(beet[k].size()));
    }
  }

  Int query(Int k,Int x,Int y){
    Int p = lower_bound(begin(beet[k]), end(beet[k]), y) - begin(beet[k]);
    Int q = lower_bound(begin(beet[k]), end(beet[k]), x) - begin(beet[k]);
    assert(beet[k][p]==y);
    assert(beet[k][q]==x);
    return (seg[k].sum(p) - seg[k].sum(q));
  }

  Int query(Int a, Int b, Int x, Int y)
  {
    Int res=0;
    for(Int l=a+sz,r=b+sz;l<r;l>>=1,r>>=1){
      if(l&1) res+=query(l++,x,y);
      if(r&1) res+=query(--r,x,y);
    }
    return res;
  }

  void prequery(Int a, Int b, Int x, Int y)
  {
    for(Int l=a+sz,r=b+sz;l<r;l>>=1,r>>=1){
      if(l&1) beet[l].emplace_back(x),beet[l].emplace_back(y),l++;
      if(r&1) r--,beet[r].emplace_back(x),beet[r].emplace_back(y);
    }
  }

  void preupdate(Int x, Int y)
  {
    x += sz;
    while(x) {
      beet[x].push_back(y);
      x>>=1;
    }
  }
};
//INSERT ABOVE HERE
signed main(){  
  Int n,m;
  scanf("%d %d",&n,&m);
  vector<Int> a(n),b(n);
  for(Int i=0;i<n;i++) scanf("%d %d",&a[i],&b[i]);

  SegmentTreeBIT seg(m);
  
  for(Int i=0;i<n;i++){
    if(a[i]>b[i]) swap(a[i],b[i]);
    seg.prequery(0,a[i],a[i],b[i]);
    seg.prequery(a[i],b[i],b[i],m);
    seg.preupdate(a[i],b[i]);
  }

  seg.build();

  long long ans=0;
  for(Int i=0;i<n;i++){
    ans+=seg.query(0,a[i],a[i],b[i]);
    ans+=seg.query(a[i],b[i],b[i],m);
    seg.update(a[i],b[i],1);
  }

  printf("%lld\n",ans);
  
  return 0;
}
0