数列分块入门6(单点插入)

原题:http://www.caioj.cn/problem.php?id=1249

题解:对于暴力的单点插入,需要将后面的全部移动,可以将数列分块,sqrt(n)的块暴力。当在同一个块插入过多次时需要重构。

不放每 sqrt(n)重构一次,可以保证复杂度。

#include<bits/stdc++.h>
#define reg register
#define N 220002
#define M 1005
#define pa pair<int,int>
using namespace std;
int n,m,nn,cnt;
int a[N];
vector<int> v[M];
inline int rd(){
	int x=0;int f=1;char s=getchar();
	while(s>'9' || s<'0') f=(f=='-'?-1:f),s=getchar();
	while(s>='0' && s<='9') x=(x<<1)+(x<<3)+s-'0',s=getchar();
	return x*f;
}
inline pa query(int p){
	int x=1;
	while(p>v[x].size()){p-=v[x].size();x++;} 
	return make_pair(x,p-1);
}
inline void change(int p,int c){
	pa t=query(p);
	v[t.first].insert(v[t.first].begin()+t.second,c);
}
inline void rebuild(){
	int hd=0;
	for(int i=1;i<=cnt;i++) {
		for(int j=0;j<v[i].size();j++) a[++hd]=v[i][j];
		v[i].clear();
	}
	m=sqrt(hd);
	for(int i=1;i<=hd;i++) v[(i-1)/m+1].push_back(a[i]);
	cnt=(hd-1)/m+1;
}
int main(){
//	freopen("input.in","r",stdin);
	n=rd();nn=m=sqrt(n); cnt=(n-1)/m+1;
	for(reg int i=1,x;i<=n;i++) x=rd(),v[(i-1)/m+1].push_back(x);
	for(reg int i=1,opt,l,r,c;i<=n;i++){
		opt=rd();l=rd();r=rd();c=rd();
		if(opt==0) change(l,r);
		if(opt==1) {
			pa p=query(r);
			printf("%d\n",v[p.first][p.second]);
		}
		if(i%nn==0) rebuild();
	} 
	return 0;
} 

 

 

posted @ 2019-01-10 18:18  Exception2017  阅读(209)  评论(0编辑  收藏  举报