結果

問題 No.1649 Manhattan Square
ユーザー 蜜蜂蜜蜂
提出日時 2021-07-28 22:24:00
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 801 ms / 3,000 ms
コード長 3,262 bytes
コンパイル時間 4,512 ms
コンパイル使用メモリ 253,076 KB
実行使用メモリ 58,476 KB
最終ジャッジ日時 2024-10-03 16:00:05
合計ジャッジ時間 32,696 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 3 ms
5,248 KB
testcase_02 AC 7 ms
5,248 KB
testcase_03 AC 7 ms
5,248 KB
testcase_04 AC 7 ms
5,248 KB
testcase_05 AC 6 ms
5,248 KB
testcase_06 AC 6 ms
5,248 KB
testcase_07 AC 757 ms
58,176 KB
testcase_08 AC 748 ms
58,092 KB
testcase_09 AC 765 ms
58,100 KB
testcase_10 AC 764 ms
58,136 KB
testcase_11 AC 753 ms
58,200 KB
testcase_12 AC 597 ms
41,384 KB
testcase_13 AC 648 ms
42,084 KB
testcase_14 AC 625 ms
43,004 KB
testcase_15 AC 627 ms
44,524 KB
testcase_16 AC 563 ms
37,240 KB
testcase_17 AC 596 ms
41,464 KB
testcase_18 AC 451 ms
32,124 KB
testcase_19 AC 583 ms
38,272 KB
testcase_20 AC 587 ms
38,820 KB
testcase_21 AC 682 ms
46,320 KB
testcase_22 AC 769 ms
58,308 KB
testcase_23 AC 775 ms
58,364 KB
testcase_24 AC 767 ms
58,476 KB
testcase_25 AC 771 ms
58,328 KB
testcase_26 AC 765 ms
58,284 KB
testcase_27 AC 772 ms
58,236 KB
testcase_28 AC 764 ms
58,352 KB
testcase_29 AC 745 ms
58,268 KB
testcase_30 AC 758 ms
58,276 KB
testcase_31 AC 756 ms
58,280 KB
testcase_32 AC 763 ms
58,272 KB
testcase_33 AC 765 ms
58,280 KB
testcase_34 AC 765 ms
58,336 KB
testcase_35 AC 766 ms
58,136 KB
testcase_36 AC 773 ms
58,196 KB
testcase_37 AC 801 ms
58,236 KB
testcase_38 AC 799 ms
58,312 KB
testcase_39 AC 772 ms
58,136 KB
testcase_40 AC 740 ms
58,280 KB
testcase_41 AC 763 ms
58,264 KB
testcase_42 AC 413 ms
41,236 KB
testcase_43 AC 2 ms
5,248 KB
testcase_44 AC 2 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

//g++ 3.cpp -std=c++14 -O2 -I .
#include <bits/stdc++.h>
using namespace std;

#include <atcoder/all>
using namespace atcoder;

using ll = long long;
using ld = long double;

using vi = vector<int>;
using vvi = vector<vi>;
using vll = vector<ll>;
using vvll = vector<vll>;
using vld = vector<ld>;
using vvld = vector<vld>;

#define fi first
#define se second
#define pb push_back
#define pq_big(T) priority_queue<T,vector<T>,less<T>>
#define pq_small(T) priority_queue<T,vector<T>,greater<T>>
#define all(a) a.begin(),a.end()
#define rep(i,start,end) for(ll i=start;i<(ll)(end);i++)
#define per(i,start,end) for(ll i=start;i>=(ll)(end);i--)

using mint = modint998244353;

struct S{
  //y=i範囲の点の数,xの合計,yの合計,xyの合計
  int quant;
  mint sum_x;
  mint sum_y;
  mint sum_pro;
};

S op(S a,S b){
  return {a.quant+b.quant,a.sum_x+b.sum_x,a.sum_y+b.sum_y,a.sum_pro+b.sum_pro};
}

S e(){
  return {0,0,0,0};
}

//i = 0 to N-1 j = i+1 to N (a[i]-a[j])^2
mint sum_differ(vll &a){
  int sz=a.size();

  mint res=0,sum=0;
  rep(i,0,sz){
    sum+=a[i];
    mint x=a[i]*a[i];
    x*=2*sz;
    res+=x;
  }
  sum*=sum;
  res-=2*sum;
  res/=2;
  return res;
}

int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);

  int n;
  cin>>n;

  vll x(n),y(n);
  vll val_x,val_y;
  rep(i,0,n){
    cin>>x[i]>>y[i];
    val_x.pb(x[i]);
    val_y.pb(y[i]);
  }

  sort(all(val_x));
  sort(all(val_y));
  unique(all(val_x));
  unique(all(val_y));

  map<ll,int> inv_x;
  map<ll,int> inv_y;

  rep(i,0,val_x.size()){
    inv_x[val_x[i]]=i;
  }
  rep(i,0,val_y.size()){
    inv_y[val_y[i]]=i;
  }

  //座圧後の点
  vector<pair<int,int>> point(n);
  rep(i,0,n){
    point[i].fi=inv_x[x[i]];
    point[i].se=inv_y[y[i]];
  }
  sort(all(point));


  //x_coordinate[i] := 座圧後にx座標がiとなる点の番号
  vvi x_coordinate(n+100);
  rep(i,0,n){
    x_coordinate[point[i].fi].pb(i);
  }

  mint ans=0;

  //左から平面走査
  //seg[i] := 座圧後にy座標がiとなる点についてのいろいろ
  segtree<S,op,e> seg(n+100);
  rep(i,0,n+100){
    if(x_coordinate[i].size()==0)continue;

    //寄与の計算と更新
    for(int num:x_coordinate[i]){
      int x=point[num].fi,y=point[num].se;
      mint ori_x=val_x[x],ori_y=val_y[y];
      S range=seg.prod(0,y);

      ans+=range.quant*ori_x*ori_y;
      ans-=ori_x*range.sum_y;
      ans-=ori_y*range.sum_x;
      ans+=range.sum_pro;

      S now=seg.get(y);
      now.quant++;
      now.sum_x+=ori_x;
      now.sum_y+=ori_y;
      now.sum_pro+=ori_x*ori_y;
      seg.set(y,now);
    }
  }

  //初期化
  rep(i,0,n+100){
    seg.set(i,e());
  }

  //右から平面走査
  per(i,n+99,0){
    if(x_coordinate[i].size()==0)continue;

    //寄与の計算と更新
    for(int num:x_coordinate[i]){
      int x=point[num].fi,y=point[num].se;
      mint ori_x=val_x[x],ori_y=val_y[y];
      S range=seg.prod(0,y);

      ans-=range.quant*ori_x*ori_y;
      ans+=ori_x*range.sum_y;
      ans+=ori_y*range.sum_x;
      ans-=range.sum_pro;

      S now=seg.get(y);
      now.quant++;
      now.sum_x+=ori_x;
      now.sum_y+=ori_y;
      now.sum_pro+=ori_x*ori_y;
      seg.set(y,now);
    }
  }

  ans*=2;
  ans+=sum_differ(x);
  ans+=sum_differ(y);

  cout<<ans.val()<<"\n";
}
0