数列分块入门6
题意概括
单点插入，单点询问。
写在前面
分块的小技巧可多啦 O(∩_∩)O哈哈~
正题
用一个vector来记录每个块的元素。vector提供了强大的insert函数，插入某个元素只要写一点点就可以啦O(∩_∩)O，就是一个一个枚举块，直到找到插入位置属于哪个块，然后insert一下就OK。不过，如果插入元素集中于一个块的话就尴尬了QAQ复杂度飙升。所以我们要对块进行重新分配。我采用若一个块元素多于10倍原有元素个数时就重排~（由于数据随机分配，不进行重排其实也不会TLE，甚至直接1个vector插入也能过 QAQ 数据水？？？建议还是写下重排，因为完全可以卡掉不重排的）
代码
#include<bits/stdc++.h>
using namespace std;
#define MAXN 100005
int n, d, nn;
int a[MAXN<<1];
vector<int> v[1005];
inline void mer(){//把所有元素还原到a数组中
n = 0;
for ( int i = 1; i <= d + 1; ++i ){
if ( v[i].empty() ) break;
for ( int j = 0; j < v[i].size(); ++j ) a[++n] = v[i][j];
v[i].clear();
}
}
inline void div(){//把a数组元素分配到各个块中
d = sqrt(n);
for ( int i = 1; i <= n; ++i ) v[( i - 1 ) / d + 1].push_back(a[i]);
}
inline int Get( int wh ){
for ( int i = 1; i <= d + 1; ++i ){
if ( wh > v[i].size() ) wh -= v[i].size();
else return v[i][wh - 1];
}
}
inline void Ins( int wh, int x ){
for ( int i = 1; i <= d + 1; ++i ){
if ( wh > v[i].size() ) wh -= v[i].size();
else{
v[i].insert( v[i].begin() + wh - 1, x );//插入~
if ( v[i].size() > 10 * d ) mer(), div();//重排
return;
}
}
}
int main(){
scanf( "%d", &n );
for ( int i = 1; i <= n; ++i ) scanf( "%d", &a[i] );
div();
nn = n;
for ( int i = 1; i <= nn; ++i ){
int opt, l, r, c;
scanf( "%d%d%d%d", &opt, &l, &r, &c );
if ( opt ) printf( "%d
", Get(r) );
else Ins( l, r );
}
return 0;
}
总结
有些分块的技巧性很强，平时注意归纳(*^▽^*)
数列分块系列目录
数列分块入门6 <-