結果

問題 No.1919 Many Monster Battles
コンテスト
ユーザー i_am_noob
提出日時 2023-02-11 13:56:44
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,176 ms / 2,000 ms
コード長 1,984 bytes
コンパイル時間 2,472 ms
コンパイル使用メモリ 212,764 KB
最終ジャッジ日時 2025-02-10 14:22:24
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
using pii=pair<int,int>;

#define all(a) a.begin(),a.end()
#define pb push_back
#define sz(a) ((int)a.size())

const int maxn=200005,mod=1000000007;
int add(int x, int y){x+=y; if(x>=mod) x-=mod; return x;}
int sub(int x, int y){x-=y; if(x<0) x+=mod; return x;}

struct BIT{
    int val[maxn];
    vector<pii> vec;
    void modify(int p, int x, bool flag=1){if(flag) vec.pb({p,x}); for(++p; p<maxn; p+=p&-p) val[p]=add(val[p],x);}
    int query(int r){
        int res=0;
        for(++r; r; r-=r&-r) res=add(res,val[r]);
        return res;
    }
    void clear(){
        for(auto [p,x]: vec) modify(p,mod-x,0);
        vec.clear();
    }
}bit1,bit2;

int n,res;
vector<array<int,3>> vec;

vector<int> vec1;

void solve1(int l, int r){
    if(l==r) return;
    int mid=l+r>>1;
    solve1(l,mid),solve1(mid+1,r);
    vec1.resize(r-l+1);
    for(int i=l; i<=r; ++i) vec1[i-l]=i;
    sort(all(vec1),[&](int i, int j){return vec[i][1]!=vec[j][1]?vec[i][1]<vec[j][1]:vec[i][2]>vec[j][2];});
    bit1.clear(),bit2.clear();
    for(int i: vec1){
        if(i<=mid) bit1.modify(vec[i][2],1),bit2.modify(vec[i][2],vec[i][0]);
        else res=add(res,sub(1ll*bit1.query(vec[i][2]-1)*vec[i][0]%mod,bit2.query(vec[i][2]-1)));
    }
}

void solve2(){
    sort(all(vec));
    vector<int> tmp;
    for(int i=0; i<n; ++i) tmp.pb(vec[i][2]);
    sort(all(tmp)),tmp.resize(unique(all(tmp))-tmp.begin());
    for(int i=0; i<n; ++i) vec[i][2]=lower_bound(all(tmp),vec[i][2])-tmp.begin();
    solve1(0,n-1);
}

int a[maxn],b[maxn];

signed main(){
    ios_base::sync_with_stdio(0),cin.tie(0);
    cin >> n;
    for(int i=0; i<n; ++i) cin >> a[i];
    for(int i=0; i<n; ++i) cin >> b[i];
    vec.resize(n);
    for(int i=0; i<n; ++i) vec[i]={a[i],a[i]+b[i],a[i]-b[i]};
    solve2();
    cout << add(res,res) << ' ';
    res=0;
    for(int i=0; i<n; ++i) vec[i]={b[i],a[i]+b[i],b[i]-a[i]};
    solve2();
    cout << add(res,res) << "\n";
}
0