poj 1442 Black Box

ac了想了好久的题，真心高兴，虽然运行有点慢，但却是用c自己写的堆，代码很短；

思路：维护两个堆，一个小顶堆用于存储所给数列最大的一部分，一个大顶堆用于存储所给数列剩下的最小一部分。

运行过程：

7 4

3 1 -4 2 8 -1000 2

1 2 6 6

当输入1时，我们将3压入存储较大值的小顶堆中，判断存储较小值的大顶堆堆顶为NULL，则直接将3从小顶堆中取出，压入大顶堆，更新两个堆，输出大顶堆堆顶的值。

当输入2时，我们将1压入存储较大值的小顶堆中，判断存储较小值的大顶堆堆顶为不为空，并且小顶堆的堆顶1<3，则我们将1和3互换，从而保证小顶堆始终存储较大的，大顶堆始终存储较小的。因为到了区间尾，跳出循环，将小顶堆的堆顶取出插入大顶堆中，从而保证大顶堆的堆顶就是我们所要求的第n小的值。

输入6时与2相同，只是在循环里多插入小顶堆几个数值。具体见代码：

#include<stdio.h>
#define INF 2000000000 + 1000000
#define MAXN 31000

int pmax, pmin, m, n, D;
int a[MAXN],amin[MAXN<<1],amax[MAXN<<1],treemax[MAXN<<2],treemin[MAXN<<2];

void updatemax(int cur)
{
    for(int i = cur; i^1; i >>= 1)
    treemax[i>>1] = (amax[treemax[i]]>amax[treemax[i^1]]?treemax[i^1]:treemax[i]);    
}

void updatemin(int cur)
{
    for(int i = cur; i^1; i >>= 1)
    treemin[i>>1] = (amin[treemin[i]]>amin[treemin[i^1]]?treemin[i]:treemin[i^1]);
}

void build()
{
    for(D = 1; D < m + 2; D <<= 1);
    for(int i = 0; i <= D; i ++){treemax[D+i] = i; treemin[D+i] = i;}
    for(int i = 0; i <= D; i ++){amax[i] = INF; amin[i] = -INF;}
}

void init()
{
    while(~scanf("%d%d",&m,&n))
    {
        for(int i = 0; i < m; i ++)
            scanf("%d",&a[i]);
        build();
        int u;
        for(int i = 0, j = 0; i < n; i ++)
        {
            scanf("%d",&u);
            for( ; j < u; j ++)
            {
                amax[pmax++] = a[j];
                updatemax(D+pmax-1);
                if(amin[treemin[1]] != -INF && amax[treemax[1]] < amin[treemin[1]])
                {
                    int t;
                    t = amax[treemax[1]];
                    amax[treemax[1]] = amin[treemin[1]];
                    amin[treemin[1]] = t;
                    updatemax(D+treemax[1]);
                    updatemin(D+treemin[1]);
                }
            }
            amin[pmin++] = amax[treemax[1]];
            updatemin(pmin+D-1);
            amax[treemax[1]] = INF;
            updatemax(D+treemax[1]);
            printf("%d\n",amin[treemin[1]]);
        }
    }     
}

int main()
{
    pmax = 1;
    pmin = 1;
    init();
    return 0;
}