POJ 2887 Big String （块状数组）

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <list>
#define ll long long
using namespace std;
int Maxn, N ;
int sum[1005];
struct BlockList
{
    int size;
    char dat[2005];
    int at(int pos)
    {
        return dat[pos];
    }
    void insert(int pos,char c)
    {
        for(int i=++size; i>pos; --i) dat[i]=dat[i-1];
        dat[pos]=c;
    }
    void push_back(char c)
    {
        dat[++size]=c;
    }
};
BlockList block[1005];
char query(int s,int p)
{
    return block[s].at(p);
}
void insert(int s,int p,char c)
{
    p=min(p,block[s].size+1);
    block[s].insert(p,c);
}
void maintain()
{
    for(int i=1; i<=Maxn; ++i)
        sum[i]=sum[i-1]+block[i].size;
}
void MyInsert(int pos,char c)
{
    int p=lower_bound(sum+1,sum+1+Maxn,pos)-(sum);
    insert(p,pos-sum[p-1],c);
    maintain();
}
int MyQuery(int pos)
{
    int p=lower_bound(sum+1,sum+1+Maxn,pos)-(sum);
    return query(p,pos-sum[p-1]);
}
char str[1000005];
void init()
{
    int len=strlen(str)+N;
    Maxn=sqrt(len*1.0)+1;
    for(int i=0; str[i]; ++i)
        block[i/Maxn+1].push_back(str[i]);
    maintain();
}
int main()
{
    gets(str);
    int pos;
    char p[3],s[3];
    scanf("%d",&N);
    init();
    while(N--)
    {
        scanf("%s",p);
        if(p[0]==‘I‘)
        {
            scanf("%s%d",s,&pos);
            MyInsert(pos,s[0]);
        }
        else
        {
            scanf("%d",&pos);
            printf("%c\n",MyQuery(pos));
        }
    }
    return 0;
}