A Magic Lamp 【HDU - 3183】【线段树区间最小值】

题目链接

---

  简单而言，这道题就是RMQ问题，但是我个人更喜欢用线段树来写区间最大值，因为这样子会好更新些（奈何这道题不需要更新）。

  我们要从长度为N的字符串中删除M个元素，那么岂不是只剩下（N-M）个字符串的长度，所以，我们不妨来找（N-M）的长度来的会更加简单一些，查找从（1～1+M）开始，然后找到最先出现的最小值（如果有相等结果取先出现的），然后再从这一位之后开始找到～（2+M），就这样一次往后推，直到推到最后查满整个区间即可。

---

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define INF 0x3f3f3f3f
#define pi 3.141592653589793
#define e 2.718281828459045
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int maxN = 1005;
char s[maxN];
int a[maxN], tree[maxN<<2], N, M, ans[maxN];
void buildTree(int rt, int l, int r)
{
    if(l == r)
    {
        tree[rt] = l;
        return;
    }
    int mid = (l + r)>>1;
    buildTree(rt<<1, l, mid);
    buildTree(rt<<1|1, mid+1, r);
    if(a[tree[rt<<1]] <= a[tree[rt<<1|1]]) tree[rt] = tree[rt<<1];
    else tree[rt] = tree[rt<<1|1];
}
int query(int rt, int l, int r, int ql, int qr)
{
    if(ql<=l && qr>=r) return tree[rt];
    int mid = (l + r)>>1;
    if(ql>mid) return query(rt<<1|1, mid+1, r, ql, qr);
    else if(qr<=mid) return query(rt<<1, l, mid, ql, qr);
    else
    {
        int res = query(rt<<1, l, mid, ql, qr);
        int tmp = query(rt<<1|1, mid+1, r, ql, qr);
        if(a[tmp] < a[res]) res = tmp;
        return res;
    }
}
int main()
{
    while(scanf("%s%d", s+1, &M)!=EOF)
    {
        N = (int)strlen(s+1);
        if(M >= N) { printf("0\n"); continue; }
        for(int i=1; i<=N; i++) a[i] = s[i] - '0';
        buildTree(1, 1, N);
        bool flag = false;
        int pos = 1;
        for(int i=1; i<=(N-M); i++)
        {
            pos = query(1, 1, N, pos, i+M);
            if(flag) printf("%d", a[pos]);
            else if(a[pos]) { printf("%d", a[pos]); flag = true; }
            pos++;
        }
        if(!flag) printf("0\n");
        else printf("\n");
    }
    return 0;
}