POJ 2887：Big String（分块）

http://poj.org/problem?id=2887

题意：给出一个字符串，还有n个询问，第一种询问是给出一个位置p和字符c，要在位置p的前面插入c（如果p超过字符串长度，自动插在最后），第二种询问是给出一个位置p，查找第p个位置的字符（p保证合法）。

思路：暴力分块。一开始建成块之后，每次插入就在每个块的字符串插入字符，因为询问最多只有2000个，所以就算极端情况也不会超时。

注意：sz和cnt其中一个要向上取整。用string类会超时。

#include <cstring>
#include <iostream>
#include <cstdio>
#include <string>
#include <cmath>
using namespace std;
char s[1010][5010];
char str[1000010];
int n, block[1010], sz, cnt;

void Build() {
    sz = sqrt(n);
    cnt = (n + sz - 1) / sz; // 记得向上取整, 因为这WA了一次
    for(int i = 0; i < n; i++) s[i / sz + 1][block[i / sz + 1]++] = str[i];
}

void Insert(char c, int p) {
    int i;
    for(i = 1; i <= cnt; i++) {
        if(p - block[i] <= 0) break;
        p -= block[i];
    }
    if(i == cnt + 1) { s[cnt][block[cnt]++] = c; return ; }
    block[i]++;
    int len = strlen(s[i]);
    for(int j = len - 1; j >= p - 1; j--) s[i][j+1] = s[i][j];
    s[i][p-1] = c;
}

void Query(int p) {
    int i;
    for(i = 1; i <= cnt; i++) {
        if(p - block[i] <= 0) break;
        p -= block[i];
    }
    printf("%c
", s[i][p-1]);
}

int main() {
    scanf("%s", str); n = strlen(str);
    int q; scanf("%d", &q);
    Build();
    while(q--) {
        char op[3], c[3]; int num;
        scanf("%s", op);
        if(op[0] == 'I') {
            scanf("%s%d", c, &num);
            Insert(c[0], num);
        } else {
            scanf("%d", &num);
            Query(num);
        }
    }
    return 0;
}