Loj 6282. 数列分块入门 6

题目描述

给出一个长为 nnn 的数列，以及 nnn 个操作，操作涉及单点插入，单点询问，数据随机生成。

输入格式

第一行输入一个数字 nnn。

第二行输入 nnn 个数字，第 i 个数字为 aia_iai​，以空格隔开。

接下来输入 nnn 行询问，每行输入四个数字 opt\mathrm{opt}opt、lll、rrr、ccc，以空格隔开。

若 opt=0\mathrm{opt} = 0opt=0，表示在第 lll 个数字前插入数字 rrr (ccc 忽略)。

若 opt=1\mathrm{opt} = 1opt=1，表示询问 ara_rar​ 的值（lll 和 ccc 忽略）。

输出格式

对于每次询问，输出一行一个数字表示答案。

样例

样例输入

4
1 2 2 3
0 1 3 1
1 1 4 4
0 1 2 2
1 1 2 4

样例输出

2
3

数据范围与提示

对于 100% 100\%100% 的数据，1≤n≤100000,−231≤others 1 \leq n \leq 100000, -2^{31} \leq \mathrm{others}1≤n≤100000,−231≤others、ans≤231−1 \mathrm{ans} \leq 2^{31}-1ans≤231−1。

 

ps: 暴力维护每个区间，发现用了vector的insert操作，程序变的好慢。

拓展，如果数据不是随机的，我们可以在sqrt(n)次插入操作以后，重新构造分块。 

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

#define F(i,a,b) for(int i=a;i<=b;i++) 
#define D(i,a,b) for(int i=a;i>=b;i--)
#define ms(i,a)  memset(a,i,sizeof(a)) 
#define LL       long long 
#define st(x)    ((x-1)*B+1)
#define ed(x)    min(x*B,n)
#define bl(x)    ((x-1)/B+1) 

int inline read(){
    int x=0,w=0; char c=getchar(); 
    while (c<'0' || c>'9') w+=c=='-',c=getchar(); 
    while (c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar(); 
    return w? -x: x;  
}

int const maxn=100003;  

vector<int> v[403];   

int n,B,a[maxn]; 

void update(int l,int r){
    int x=0;  
    F(i,1,bl(n)) if(v[i].size()<l) l-=v[i].size();  
    else  {
        x=i; break;  
    }
    v[x].insert(v[x].begin()+l-1,r);  
}

int query(int x){
    int k=0;  
    F(i,1,bl(n)) if(v[i].size()<x) x-=v[i].size();  
    else {
        k=i; break;  
    }
    return v[k][x-1]; 
}

int main(){
    n=read(); 
    B=(int) sqrt(n);  
    F(i,1,n) a[i]=read(); 
    F(i,1,bl(n)) F(j,st(i),ed(i)) {
        v[i].push_back(a[j]);  
    }
    F(i,1,n){
        int x=read(); 
        int l=read(); 
        int r=read(); 
        int c=read(); 
        if(x==0) update(l,r); 
        else printf("%d\n",query(r)); 
    }
    return 0; 
}