POJ 2887 块状数组

题意：

在字符串中的pos位置插入字符串，询问pos位置的字母是谁

 

思路：

裸的块状数组，第一次写，“借鉴”了别人的，比我自己yy的短多了。。。

自己yy的又长又臭，效率还低，就给出别人的吧。。

 

PS：真心觉得块状数组是个好东西。。。

 

#include <cstdio>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <iostream>

#define N 5000

using namespace std;

struct BK
{
    int size,next;
    char s[N];
    void push(char str)
    {
        s[size++]=str;
    }
    void insert(int pos,char str)
    {
        for(int i=size++;i>pos;i--) s[i]=s[i-1];
        s[pos]=str;
    }
}bk[N];

char a[4000000];
int lena,m,bsize,cnt;

void read()
{
    gets(a);
    lena=strlen(a);
    scanf("%d",&m);
    bsize=(int)sqrt((double)(lena+m));
    cnt=0;
    bk[cnt].size=0;
    for(int i=0;i<lena;i++)
    {
        if(bk[cnt].size==bsize)
        {
            bk[cnt].next=cnt+1;
            bk[++cnt].size=0;
        }
        bk[cnt].push(a[i]);
    }
    bk[cnt].next=-1;
}

void updata(int u)
{
    if(bk[u].size<2*bsize) return;
    ++cnt;
    int i,j,k=bk[u].size;
    for(i=bsize,j=0;i<k;i++,j++) bk[cnt].s[j]=bk[u].s[i];
    bk[cnt].size=j;
    bk[u].size=bsize;
    bk[cnt].next=bk[u].next;
    bk[u].next=cnt;
}

void go()
{
    for(int i=1,pos,j;i<=m;i++)
    {
        scanf("%s",a);
        if(a[0]=='Q')
        {
            scanf("%d",&pos);
            for(j=0;pos>bk[j].size;j=bk[j].next) pos-=bk[j].size;
            printf("%c\n",bk[j].s[pos-1]);
        }
        else 
        {
            scanf("%s%d",a,&pos);
            for(j=0;pos>bk[j].size&&bk[j].next!=-1;j=bk[j].next) pos-=bk[j].size;
            bk[j].insert(min(pos-1,bk[j].size+1),a[0]);
            updata(j);
        }
    }
}

int main()
{
    read();
    go();
    return 0;
}