結果

問題 No.5011 Better Mo's Algorithm is Needed!! (Weighted)
ユーザー t98slider
提出日時 2022-12-17 03:05:12
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 128 ms / 5,000 ms
コード長 2,104 bytes
コンパイル時間 2,484 ms
実行使用メモリ 13,244 KB
スコア 34,956,868,287
最終ジャッジ日時 2022-12-17 03:06:13
合計ジャッジ時間 58,732 ms
ジャッジサーバーID
（参考情報） 		judge16 / judge15
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 120
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

struct Mo_Algorithm{
    ll N, D, Q;//配列の要素数,幅,クエリ数
    std::vector<std::pair<ll, ll> > query;
    Mo_Algorithm(ll n) : N(n){}
    void add_query(ll l, ll 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<ll> build(){
        Q = query.size() , D = std::max(1ll, (ll)(N / (sqrt(Q))));
        std::vector<ll> 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(), [&](ll i, ll j){
            if(Div[i] != Div[j])return Div[i] < Div[j];
            return (Div[i] & 1 ? 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;
    vector<ll> W(N), sw(N + 1);
    for(int i = 0; i < N; i++){
        cin >> W[i];
        sw[i + 1] = sw[i] + W[i];
    }
    Mo_Algorithm MA(sw.back());
    for(int i = 0; i < Q; i++){
        cin >> L >> R;
        MA.add_query(sw[--L], sw[R]);
    }
    auto ans = MA.build();
    for(int i = 0; i < ans.size(); i++){
        cout << ++ans[i] << (i + 1 == ans.size() ? '\n' : ' ');
    }
}
0