結果

問題 No.1270 Range Arrange Query
ユーザー leaf_1415
提出日時 2020-10-24 00:30:03
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 2,074 ms / 7,000 ms
コード長 5,271 bytes
コンパイル時間 1,926 ms
コンパイル使用メモリ 108,912 KB
実行使用メモリ 7,248 KB
最終ジャッジ日時 2024-07-21 14:20:16
合計ジャッジ時間 11,975 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <iostream>
#include <cstdio>
#include <cmath>
#include <ctime>
#include <cstdlib>
#include <cassert>
#include <vector>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <map>
#include <set>
#include <bitset>
#include <string>
#include <algorithm>
#include <utility>
#define rep(x, s, t) for(llint (x) = (s); (x) <= (t); (x)++)
#define chmin(x, y) (x) = min((x), (y))
#define chmax(x, y) (x) = max((x), (y))
#define all(x) (x).begin(),(x).end()
#define inf 1e18
#define mod 1000000007
using namespace std;
typedef long long llint;
typedef pair<llint, llint> P;
struct BIT{
    int size;
    vector<llint> bit;
    BIT(){size = 0;}
    BIT(int s){
        size = s;
        bit.resize(size+1);
        init();
    }
    void init(){
        for(int i = 1; i <= size; i++) bit[i] = 0;
    }
    llint query(int i){
        llint ret = 0;
        while(i > 0){
            ret += bit[i];
            i -= i&(-i);
        }
        return ret;
    }
    void add(int i, llint x){
        while(i <= size){
            bit[i] += x;
            i += i&(-i);
        }
    }
};
// range-add, range-min query
 
struct LazySegTree{
    int size;
    vector<llint> seg, delay;
    
    LazySegTree(){}
    LazySegTree(int size){
        this->size = size;
        seg.resize(1<<(size+1));
        delay.resize(1<<(size+1));
    }
    
    void init()
    {
        for(int i = 0; i < (1<<(size+1)); i++){
            seg[i] = inf;
            delay[i] = 0;
        }
    }
    
    void eval(int l, int r, int k)
    {
        if(delay[k]){
            seg[k] += delay[k];  //総和クエリのときは区間幅を乗じる必要がある
            if(l < r){
                delay[k*2] += delay[k];
                delay[k*2+1] += delay[k];
            }
            delay[k] = 0;
        }
    }
    
    void update(int i, llint val)
    {
        int l = 0, r = (1<<size)-1, k = 1;
        eval(l, r, k);
        for(int j = size-1; j >= 0; j--){
            k <<= 1;
            if(i & (1<<j)){
                k++;
                r = (l+r)/2;
            }
            else l = (l+r)/2+1;
            eval(l, r, k);
        }
        
        i += (1 << size);
        seg[i] = val;
        while(i > 1){
            i /= 2;
            seg[i] = min(seg[i*2], seg[i*2+1]);
        }
    }
    
    void add(int a, int b, int k, int l, int r, llint val)
    {
        eval(l, r, k);
        
        if(b < l || r < a) return;
        if(a <= l && r <= b){
            delay[k] += val;
            eval(l, r, k);
            return;
        }
        add(a, b, k*2, l, (l+r)/2, val);
        add(a, b, k*2+1, (l+r)/2+1, r, val);
        seg[k] = min(seg[k*2], seg[k*2+1]);
    }
    void add(int a, int b, llint val){
        if(a > b) return;
        add(a, b, 1, 0, (1<<size)-1, val);
    }
 
    llint query(int a, int b, int k, int l, int r)
    {
        eval(l, r, k);
        
        if(b < l || r < a) return inf;
        if(a <= l && r <= b) return seg[k];
        llint lval = query(a, b, k*2, l, (l+r)/2);
        llint rval = query(a, b, k*2+1, (l+r)/2+1, r);
        return min(lval, rval);
    }
    llint query(int a, int b)
    {
        return query(a, b, 1, 0, (1<<size)-1);
    }
};
llint n, Q;
LazySegTree seg(15);
BIT lbit(20005), rbit(20005);
struct Mo{
    typedef pair<P, P> Query;
    int B;
    vector<int> vec;
    vector<Query> qvec;
    vector<P> avec;
    int qid;
    llint sum = 0;
    
    Mo(){
        init();
    }
    void init(){
        vec.clear();
        qvec.clear();
        avec.clear();
        qid = 0;
        
        lbit.init(), rbit.init();
    }
    void setArray(vector<int> &src){
        vec = src;
        B = sqrt((int)src.size());
    }
    int addQuery(int l, int r)
    {
        qvec.push_back(Query(P(l/B, r), P(l, qid)));
        return qid++;
    }
    void calc()
    {
        if(qvec.size() == 0) return;
        sort(qvec.begin(), qvec.end());
        
        int l = qvec[0].second.first, r = qvec[0].first.second;
        for(int i = 0; i < l; i++) erase(vec[i], 0);
        for(int i = (int)vec.size()-1; i > r; i--) erase(vec[i], 1);
        //for(int i = l; i <= r; i++) add(vec[i]);
        avec.push_back(P(qvec[0].second.second, get(l, r)));
        
        for(int i = 1; i < qvec.size(); i++){
            int nl = qvec[i].second.first, nr = qvec[i].first.second;
            while(l < nl) erase(vec[l], 0), l++;
            while(l > nl) l--, add(vec[l], 0);
            while(r < nr) r++, add(vec[r], 1);
            while(r > nr) erase(vec[r], 1), r--;
            int qid = qvec[i].second.second;
            avec.push_back(P(qid, get(l, r)));
        }
        sort(avec.begin(), avec.end());
    }
    int getAnswer(int qid)
    {
        return avec[qid].second;
    }
    
    void add(int x, int l)
    {
        if(l == 0){
            seg.add(1, x-1, -1);
            lbit.add(x, -1);
            sum -= rbit.query(x-1);
        }
        else{
            seg.add(x+1, n, -1);
            rbit.add(x, -1);
            sum -= lbit.query(n) - lbit.query(x);
        }
    }
    void erase(int x, int l)
    {
        if(l == 0){
            seg.add(1, x-1, 1);
            sum += rbit.query(x-1);
            lbit.add(x, 1);
        }
        else{
            seg.add(x+1, n, 1);
            sum += lbit.query(n) - lbit.query(x);
            rbit.add(x, 1);
        }
    }
    int get(llint l, llint r)
    {
        return seg.query(1, n) * (r-l+1) + sum;
    }
};
llint a[20005];
llint l[20005], r[20005], qid[20005];
llint linv[20005], rinv[20005];
BIT bit(20005);
Mo mo;
int main(void)
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    
    cin >> n >> Q;
    for(int i = 1; i <= n; i++) cin >> a[i];
    
    for(int i = 1; i <= n; i++){
        linv[i] = linv[i-1] + bit.query(n) - bit.query(a[i]);
        bit.add(a[i], 1);
    }
    bit.init();
    for(int i = n; i >= 1; i--){
        rinv[i] = rinv[i+1] + bit.query(a[i]-1);
        bit.add(a[i], 1);
    }
    
    seg.init();
    for(int i = 1; i <= n; i++) seg.update(i, 0);
    
    vector<int> vec;
    for(int i = 1; i <= n; i++) vec.push_back(a[i]);
    mo.setArray(vec);
    
    for(int q = 1; q <= Q; q++){
        cin >> l[q] >> r[q];
        qid[q] = mo.addQuery(l[q]-1, r[q]-1);
    }
    mo.calc();
    
    for(int q = 1; q <= Q; q++){
        llint ans = mo.getAnswer(qid[q]);
        ans += linv[l[q]-1] + rinv[r[q]+1];
        cout << ans << endl;
    }
    
    return 0;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0