Submission #495782

#	Time	Username	Problem	Language	Result	Execution time	Memory
495782		zaneyu	Pairs (IOI07_pairs)	C++14	100 / 100	254 ms	9656 KiB

pairs

/*input
3 8 10 20 10 10 10 10 10 20 10 20 10 10 20 20 20 10 10 20 10 20 20 20 10 20 20 20
*/
#include<bits/stdc++.h>
using namespace std;
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
typedef tree<long long,null_type,less_equal<long long>,rb_tree_tag,tree_order_statistics_node_update> indexed_set;
#pragma GCC optimize("Ofast")
#pragma GCC target("avx2")
//order_of_key #of elements less than x
// find_by_order kth element
using ll=long long;
using ld=long double;
using pii=pair<int,int>;
#define f first
#define s second
#define pb push_back
#define REP(i,n) for(int i=0;i<n;i++)
#define REP1(i,n) for(ll i=1;i<=n;i++)
#define FILL(n,x) memset(n,x,sizeof(n))
#define ALL(_a) _a.begin(),_a.end()
#define sz(x) (int)x.size()
#define SORT_UNIQUE(c) (sort(c.begin(),c.end()),c.resize(distance(c.begin(),unique(c.begin(),c.end()))))
const ll maxn=3e5+5;
const ll maxlg=__lg(maxn)+2;
const ll INF64=4e18;
const int INF=0x3f3f3f3f;
const ll MOD=1e9+7;
const ld PI=acos(-1);
const ld eps=1e-9;
#define lowb(x) x&(-x)
#define MNTO(x,y) x=min(x,(__typeof__(x))y)
#define MXTO(x,y) x=max(x,(__typeof__(x))y)
template<typename T1,typename T2>
ostream& operator<<(ostream& out,pair<T1,T2> P){
    out<<P.f+1<<' '<<P.s+1;
    return out;
}
template<typename T>
ostream& operator<<(ostream& out,vector<T> V){
    REP(i,sz(V)) out<<V[i]<<((i!=sz(V)-1)?"\n":"");
    return out;
}
ll mult(ll a,ll b){
    return a*b%MOD;
}
ll mult(ll a,ll b,ll mod){
    ll res=0;
    while(b){
        if(b&1) res=(res+a)%mod;
        a=(a+a)%mod;
        b>>=1;
    }
    return res;
}
ll mypow(ll a,ll b,ll mod){
    if(b<=0) return 1;
    a%=mod;
    ll res=1LL;
    while(b){
        if(b&1) res=(res*a)%mod;
        a=(a*a)%mod;
        b>>=1;
    }
    return res;
}

ll mypow(ll a,ll b){
    a%=MOD;
    if(a==0) return 0;
    if(b<=0) return 1;
    ll res=1LL;
    while(b){
        if(b&1) res=(res*a)%MOD;
        a=(a*a)%MOD;
        b>>=1;
    }
    return res;
}
int a[maxn];
pii arr[maxn];
int d,m;
struct bit{
    int bit[maxn];
    void upd(int x,int p){
        x+=m;
        while(x<maxn){
            bit[x]+=p;
            x+=lowb(x);
        }
    }
    int query(int x){
        x+=m;
        int ans=0;
        while(x){
            ans+=bit[x];
            x-=lowb(x);
        }
        return ans;
    }
}ls,mr;
ll ans=0;
void rec(int l,int r){
    if(l==r) return;
    int mid=(l+r)/2;
    vector<pii> v;
    //arr[j].s>arr[i].s
    for(int i=mid+1;i<=r;i++){
        v.pb({arr[i].s,i});
        mr.upd(arr[i].f+arr[i].s,1);
    }
    vector<pii> v2;
    for(int i=mid;i>=l;i--) v2.pb({arr[i].s,i});
    sort(ALL(v)),sort(ALL(v2));
    reverse(ALL(v));
    for(auto x:v2){
        int i=x.s;
        while(sz(v) and v.back().f<=x.f){
            int j=v.back().s;
            mr.upd(arr[j].f+arr[j].s,-1);
            ls.upd(arr[j].f-arr[j].s,1);
            v.pop_back();
        }
        ans+=mr.query(d+arr[i].f+arr[i].s)+ls.query(d+arr[i].f-arr[i].s);
    }
    while(sz(v)){
        int j=v.back().s;
        mr.upd(arr[j].f+arr[j].s,-1);
        ls.upd(arr[j].f-arr[j].s,1);
        v.pop_back();
    }
    for(int i=mid+1;i<=r;i++) ls.upd(arr[i].f-arr[i].s,-1);
    rec(l,mid),rec(mid+1,r);
}
int pref[80][160][160];
pair<int,pii> b[maxn];
int get(int i,int a,int b,int c,int d){
    MXTO(a,0),MXTO(b,0),MNTO(c,2*m),MNTO(d,2*m);
    return pref[i][c][d]-pref[i][a][d]-pref[i][c][b]+pref[i][a][b];
}
int main(){
    ios::sync_with_stdio(false),cin.tie(0);
    int B,n;
    cin>>B>>n>>d>>m;
    if(B==1){
        REP(i,n) cin>>a[i];
        sort(a,a+n);
        ll ans=0; 
        REP(i,n) ans+=upper_bound(a,a+n,a[i]+d)-lower_bound(a,a+n,a[i]-d);
        cout<<(ans-n)/2;
        return 0;
    }
    if(B==2){
        REP(i,n) cin>>arr[i].f>>arr[i].s;
        sort(arr,arr+n);
        rec(0,n-1);
        cout<<ans;
        return 0;
    }
    REP(i,n){
        int x,y,z;
        cin>>x>>y>>z;
        pref[x][y-z+m][y+z]++;
        b[i]={x,{y-z+m,y+z}};
    }
    REP1(i,m){
        REP1(j,2*m){
            REP1(k,2*m){
                pref[i][j][k]+=pref[i][j-1][k]+pref[i][j][k-1]-pref[i][j-1][k-1];
            }
        }
    }
    REP(i,n){
        REP1(j,m){
            int D=d-abs(b[i].f-j);
            if(D<0) continue;
            ans+=get(j,b[i].s.f-D-1,b[i].s.s-D-1,b[i].s.f+D,b[i].s.s+D);
        }
    }
    cout<<(ans-n)/2;
}

• Subtask 1: 6 / 6
• Subtask 2: 6 / 6
• Subtask 3: 6 / 6
• Subtask 4: 6 / 6
• Subtask 5: 6 / 6
• Subtask 6: 7 / 7
• Subtask 7: 7 / 7
• Subtask 8: 8 / 8
• Subtask 9: 8 / 8
• Subtask 10: 10 / 10
• Subtask 11: 10 / 10
• Subtask 12: 10 / 10
• Subtask 13: 10 / 10