PAT A1057 Stack (30 分) 分块

    题目大意：模拟栈的操作，给出三种指令，"Push x"，将正数 x（<10^5) 压入栈中；"Pop"，栈顶元素出栈；”PeekMedian"，输出栈中所有元素的中位数。指令不超过10^5条。

    首先，N个数的中位数应当是从小到大的第(N+1) / 2个数（从1开始计）。

    寻找N个不超过10^5的正数的中位数，比较快速的方法是在压栈和出栈过程中，记录从1 ~ 10^5所有数出现的次数，然后依次遍历累加，直到当前的和 >= (N+1) / 2，说明当前遍历到的数就是中位数。但因为本题中指令数也很多，如果每次都要遍历 10^5 量级，也会超时。

     因此，为了进一步降低时间复杂度，需要使用分块操作，例如，用 block[0] 记录 0~316（sqrt(10^5) 约为 316)出现的总次数， block[1]记录 317 ~ 633 出现的总次数，依次类推，block[316]记录的就是 100172 ~ 100488 出现的总次数，因此  数组 int block[317] 足以覆盖 10^5的所有数。那么，首先遍历block数组， 累加计数，直到和 >= (N + 1) / 2，说明中位数应当在block[K]内。然后将累加的和退回遍历完block[K-1]的状态，并依次遍历累加 block[K]中所有元素的出现次数，直到 >= (N+1)/2，说明这个数就是N个数的中位数。

    注意在push和pop操作中需要维护 stack、blockCount、和 numCount这三个变量。

AC代码：

#include <cstdio>
#include <stack>

using namespace std;

int blockCount[317] = {0};
int numCount[100010] = {0};
stack<int> st;

void pop()
{
    if(st.size() > 0)
    {
        int top = st.top();
        blockCount[top / 317]--;
        numCount[top]--;
        printf("%d\n", top);
        st.pop();
    }
    else printf("Invalid\n");
}

void peekMedian()
{
    if(st.size() > 0)
    {
        int mid = (st.size() + 1) / 2;
        int cnt = 0, blockIndex = 0;
        while(cnt < mid)
            cnt += blockCount[blockIndex++];
        cnt -= blockCount[--blockIndex];
        for (int i = blockIndex * 317; i < (blockIndex + 1) * 317; ++i)
        {
            cnt += numCount[i];
            if(cnt >= mid)
            {
                printf("%d\n", i);
                break;
            }
        }
    }
    else printf("Invalid\n");
}

int main()
{

    int N;
    scanf("%d", &N);
    for (int i = 0; i < N; ++i)
    {
        char cmd[20];
        scanf("%s", cmd);
        if(cmd[1] == 'o') pop();
        else if(cmd[1] == 'e') peekMedian();
        else if(cmd[1] == 'u')
        {
            int num;
            scanf("%d", &num);
            st.push(num);
            blockCount[num / 317]++;
            numCount[num]++;
        }
    }
    return 0;
}