結果

問題 No.5010 Better Mo's Algorithm is Needed!! (Unweighted)
コンテスト
ユーザー t98slider
提出日時 2022-12-17 03:13:07
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 108 ms / 5,000 ms
コード長 2,264 bytes
コンパイル時間 2,811 ms
実行使用メモリ 7,112 KB
スコア 8,987,570,417
最終ジャッジ日時 2022-12-17 03:14:02
合計ジャッジ時間 55,280 ms
ジャッジサーバーID
（参考情報） 		judge10 / judge14
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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), cnt(N / D + 1);
        std::vector<bool> type(N / D + 1);
        for(int i = 0; i < Q; i++){
            Div[i] = query[i].second / D;
            cnt[Div[i]]++;
        }
        bool s = false;
        for(int i = 0; i < cnt.size(); i++){
            type[i] = s;
            if(cnt[i] >= 1)s ^= true;
        }
        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 type[i] ? query[i] > query[j] : 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