結果

問題 No.5010 Better Mo's Algorithm is Needed!! (Unweighted)
ユーザー t98slider
提出日時 2022-12-17 03:18:13
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 124 ms / 5,000 ms
コード長 1,469 bytes
コンパイル時間 3,295 ms
実行使用メモリ 7,112 KB
スコア 16,087,046,607
最終ジャッジ日時 2022-12-17 03:19:14
合計ジャッジ時間 59,011 ms
ジャッジサーバーID
(参考情報)
judge15 / judge16
純コード判定しない問題か言語
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
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);
    }
    vector<int> build(){
        Q = query.size() , D = std::max(1, int((3 * N) / (2 * 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[Div[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