結果

問題 No.1441 MErGe
ユーザー chocorusk
提出日時 2021-03-26 22:38:42
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 601 ms / 1,000 ms
コード長 2,116 bytes
コンパイル時間 3,079 ms
コンパイル使用メモリ 181,480 KB
最終ジャッジ日時 2025-01-19 23:10:42
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#include <iostream>
#include <string>
#include <cmath>
#include <bitset>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <deque>
#include <algorithm>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <random>
#include <cassert>
#include <fstream>
#include <utility>
#include <functional>
#include <time.h>
#include <stack>
#include <array>
#include <list>
#include <atcoder/all>
#define popcount __builtin_popcount
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef pair<int, int> P;
template<typename T>
struct BIT{
    vector<T> bit;
    int size;
    BIT(int n):size(n), bit(n+1, 0){}
    T sum(int i){ //[0, i)
        T s=0;
        while(i>0){
            s+=bit[i];
            i-=(i&(-i));
        }
        return s;
    }
    T sum(int l, int r){ //[l, r)
        return sum(r)-sum(l);
    }
    void add(int i, T x){
        i++;
        while(i<=size){
            bit[i]+=x;
            i+=(i&(-i));
        }
    }
};
int main()
{
    int n, q; cin>>n>>q;
    BIT<int> bit(n+1);
    BIT<ll> bits(n+1);
    int nx[200020];
    ll a[200020];
    a[n]=0, nx[n]=n;
    for(int i=0; i<n; i++){
        cin>>a[i];
        bits.add(i, a[i]);
        nx[i]=i+1;
        bit.add(i, 1);
    }

    while(q--){
        int t, l, r; cin>>t>>l>>r;
        l--;
        int ok=0, ng=n;
        while(ng-ok>1){
            int mid=(ok+ng)/2;
            if(bit.sum(mid)<=l) ok=mid;
            else ng=mid;
        }
        if(t==1){
            for(int i=l+1; i<r; i++){
                bits.add(nx[ok], -a[nx[ok]]);
                bits.add(ok, a[nx[ok]]);
                bit.add(nx[ok], -1);
                a[ok]+=a[nx[ok]];
                a[nx[ok]]=0;
                nx[ok]=nx[nx[ok]];
            }
        }else{
            int ok1=0, ng1=n+1;
            while(ng1-ok1>1){
                int mid=(ok1+ng1)/2;
                if(bit.sum(mid)<=r) ok1=mid;
                else ng1=mid;
            }
            printf("%lld\n", bits.sum(ok1)-bits.sum(ok));
        }
    }
    return 0;
}
0