数列分块入门6

题目链接
当单块过大时,进行重新建块

#include<stdio.h>
#include<set>
#include<vector>
#include<algorithm>
#include<queue>
#include<iostream>
#include<string.h>
#include<string>
#include<cmath>
#include<cstdlib>
using namespace std;
typedef long long int ll;
const int N=100000+10;
int v[N],st[2*N];
vector<ll>s[N];
int n,m,top,blo;
pair<int,int>query(int b)
{
    int x=1;
    while(b>s[x].size())
        b-=s[x].size(),x++;
    return make_pair(x,b-1);
}
void rebuild()
{
    top=0;
    for(int i=1;i<=m;i++)
    {
        for(vector<ll>::iterator j=s[i].begin();j!=s[i].end();j++)
            st[++top]=(*j);
        s[i].clear();
    }
    int blo2=sqrt(top);
    for(int i=1;i<=top;i++)
        s[(i-1)/blo2+1].push_back(st[i]);
    m=(top-1)/blo2+1;
}
void inst(int a,int b)
{
    pair<int,int>t=query(a);
    s[t.first].insert(s[t.first].begin()+t.second,b);
    if(s[t.first].size()>20*blo)
        rebuild();
}
int main()
{
    scanf("%d",&n);
    blo=sqrt(n);
    for(int i=1;i<=n;i++)scanf("%d",v+i);
    for(int i=1;i<=n;i++)
    s[(i-1)/blo+1].push_back(v[i]);
    m=(n-1)/blo+1;
    for(int i=1;i<=n;i++)
    {
        int l,r,f,c;
        scanf("%d%d%d%d",&f,&l,&r,&c);
        if(f==0)inst(l,r);
        if(f==1)
        {
            pair<int,int>t=query(r);
            printf("%d\n",s[t.first][t.second]);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Pandauncle/article/details/80725716