結果

問題 No.5010 Better Mo's Algorithm is Needed!! (Unweighted)
ユーザー t98slidert98slider
提出日時 2022-12-17 03:00:57
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 104 ms / 5,000 ms
コード長 2,029 bytes
コンパイル時間 2,922 ms
実行使用メモリ 7,100 KB
スコア 10,093,845,997
最終ジャッジ日時 2022-12-17 03:01:56
合計ジャッジ時間 54,589 ms
ジャッジサーバーID
(参考情報)
judge14 / judge10
純コード判定しない問題か言語
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 120
権限があれば一括ダウンロードができます

ソースコード

diff #

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

struct Mo_Algorithm{
    int N, D, Q;//配列の要素数,幅,クエリ数
    std::vector<std::pair<int, int> > query;
    Mo_Algorithm(int n) : N(n){}
    void add_query(int l,int r){
        query.emplace_back(l, r);
    }
    template< typename ADD, typename DEL, typename ASK>
    void build(const ADD& add,const DEL& del, const ASK& ask){
        Q = query.size() , D = std::max(1, int(N / sqrt(Q)));
        std::vector<int> order(Q), Div(Q);
        for(int i = 0; i < Q; i++)Div[i] = query[i].second / D;
        std::iota(order.begin(), order.end(), 0);
        std::sort(order.begin(), order.end(), [&](int i,int j){
            if(Div[i] != Div[j])return Div[i] < Div[j];
            return query[i] < query[j];
        });
        int cl = 0, cr = 0, l,r;
        for(auto &&qi : order){
            std::tie(l, r) = query[qi];
            while(cl > l)cl--, add(cl);
            while(cr < r)add(cr), cr++;
            while(cl < l)del(cl), cl++;
            while(cr > r)cr--, del(cr);
            ask(qi);
        }
    }
    vector<int> build(){
        Q = query.size() , D = std::max(1, int(N / sqrt(Q)));
        std::vector<int> order(Q), Div(Q);
        for(int i = 0; i < Q; i++)Div[i] = query[i].second / D;
        std::iota(order.begin(), order.end(), 0);
        std::sort(order.begin(), order.end(), [&](int i,int j){
            if(Div[i] != Div[j])return Div[i] < Div[j];
            if(Div[i] & 1)query[i] > query[j];
            return query[i] < query[j];
        });
        return order;
    }
};

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int N, Q, WT, ST, L, R;
    cin >> N >> Q >> WT >> ST;
    Mo_Algorithm MA(N);
    vector<int> W(N);
    for(auto &&v:W)cin >> v;
    for(int i = 0; i < Q; i++){
        cin >> L >> R;
        MA.add_query(--L, --R);
    }
    auto ans = MA.build();
    for(int i = 0; i < ans.size(); i++){
        cout << ++ans[i] << (i + 1 == ans.size() ? '\n' : ' ');
    }
}
0