hdu3183A Magic Lamp【RMQ】

今天开始为期2天的RMQ，争取能做上6-7题，本来也不是多大的知识点。

其实第一场网络赛的之前就知道这个算法，dp的logn求静态区间最大最小值的巧妙算法，只不过这个题改成了要你算下标了。还是没有理解好ST算法的本质，导致TLE了两次后才幡然悔悟借鉴苟神博客才A掉。自己最开始还自以为是的觉得每次查询出来当前的最值然后改成0就可以了，那还和线段树有什么两样？？居然没有想到，第几次找的最值和数组的下标也有很大关系。比方说已知的数组有6个数，我想保留4个，那第一个数一定是要在前三个里面找的！！要不然后面没得选了。再有一个bug就是：保留的数组非0 的前面的0需要省略，但是非0的后面不能省略。

/************
hdu3183
2015.1.19
31MS 1900K 1733B C++
************/
#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define MAXN 55555
#define MAXM 11111
int mi[MAXN][17], mx[MAXN][17], w[MAXN];
int n, q;
void rmqinit()
{
    for(int i = 1; i <= n; i++) mi[i][0] = mx[i][0] = i;
    int m = (int)(log(n * 1.0) / log(2.0));
    for(int i = 1; i <= m; i++)
        for(int j = 1; j <= n; j++)
        {
            mx[j][i] = mx[j][i - 1];
            mi[j][i] = mi[j][i - 1];
            if(j + (1 << (i - 1)) <= n)
            {
                if(w[mx[j][i]] < w[mx[j + (1 << (i - 1))][i - 1]]) mx[j][i] = mx[j + (1 << (i - 1))][i - 1];
                if(w[mi[j][i]] > w[mi[j + (1 << (i - 1))][i - 1]]) mi[j][i] = mi[j + (1 << (i - 1))][i - 1];
            }
        }
}
int rmqmin(int l,int r)
{
    int m = (int)(log((r - l + 1) * 1.0) / log(2.0));
    if(w[mi[l][m]] > w[mi[r - (1 << m) + 1][m]]) return mi[r - (1 << m) + 1][m];
    else return mi[l][m];
}
int ans[1005];
int main()
{
    //freopen("cin.txt","r",stdin);
    char str[1005];
    int num;
    while(~scanf("%s",str))
    {
        int len=strlen(str);
        n=len;
        for(int i=0;i<len;i++) w[i+1]=str[i]-'0';
        scanf("%d",&num);
        int m=len-num;
        rmqinit();
        int i,j;
        i=j=1;
        while(m--)
        {
            i=rmqmin(i,len-m);
            //printf("pos=%d   ",pos);
            ans[j++]=w[i++];
        }
        for(i=1;i<j;i++)
            if(ans[i]!=0) break;
        if(i==j){printf("0\n");continue;}
        for(;i<j;i++)
            printf("%d",ans[i]);
        printf("\n");
    }
    return 0;
}