結果

問題 No.1332 Range Nearest Query
ユーザー penguinman
提出日時 2020-10-30 03:07:32
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 541 ms / 2,500 ms
コード長 2,337 bytes
コンパイル時間 2,272 ms
コンパイル使用メモリ 183,364 KB
実行使用メモリ 31,656 KB
最終ジャッジ日時 2024-07-22 10:35:17
合計ジャッジ時間 20,467 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using std::cin;
using std::cout;
using std::vector;
using ll=long long;
#define endl "\n";
const int inf=INT_MAX;
//セグ木
struct Segtree{
    int N;
    vector<int> node;
    Segtree(int n){
        N=1;
        while(N<n) N*=2;
        node.resize(N*2-1,-inf);
    }
    void update(int ind,int val){
        ind+=N-1;
        node[ind]=val;
        while(ind){
            ind=(ind-1)/2;
            node[ind]=std::max(node[ind*2+1],node[ind*2+2]);
        }
    }
    int lower_bound(int left,int val,int now=0,int l=0,int r=-1){
        if(r<0) r=N;
        if(node[now]<val||r<=left) return r;
        if(now>=N-1) return l;
        int left_ind=lower_bound(left,val,now*2+1,l,(l+r)/2);
        if(left_ind==(l+r)/2) return lower_bound(left,val,now*2+2,(l+r)/2,r);
        return left_ind;
    }
};
int main(){
    //入力、宣言
    int N; cin>>N;
    vector<int> X(N);
    for(int i=0;i<N;i++) cin>>X[i];
    int Q; cin>>Q;
    vector<int> l(Q),r(Q),x(Q),ans(Q,inf);
    for(int i=0;i<Q;i++){
        cin>>l[i]>>r[i]>>x[i];
        l[i]--;
        r[i]--;
    }
    //本題。1 回目では x≤X_i を満たすものについて、2 回目では x≥X_i を満たすものについて求めている。
    for(int _=0;_<2;_++){
        //座圧
        vector<int> comp(N),rev(N);
        {
            vector<std::pair<int,int>> memo(N);
            for(int i=0;i<N;i++) memo[i]=std::make_pair(X[i],i);
            sort(memo.begin(),memo.end());
            for(int i=0;i<N;i++){
                comp[memo[i].second]=i;
                rev[i]=memo[i].first;
            }
        }
        vector<vector<int>> left(N),ind(N);
        for(int i=0;i<Q;i++){
            left[r[i]].push_back(l[i]);
            ind[r[i]].push_back(i);
        }
        Segtree seg(N);
        for(int i=0;i<N;i++){
            seg.update(comp[i],i);
            for(int j=0;j<left[i].size();j++){
                int l=lower_bound(rev.begin(),rev.end(),x[ind[i][j]])-rev.begin();
                int index=seg.lower_bound(l,left[i][j]);
                if(index<N) ans[ind[i][j]]=std::min(ans[ind[i][j]],rev[index]-x[ind[i][j]]);
            }
        }
        //上下を反転
        for(int i=0;i<N;i++) X[i]=-X[i];
        for(int i=0;i<Q;i++) x[i]=-x[i];
    }
    for(int i=0;i<Q;i++) cout<<ans[i]<<endl;
}
0