数据机构专题 - 解题报告 - D

这题的诀窍在于我们需要找到区间内最大值的同时保证剩下的元素能被选全，保证m位密码能被填满，区间最大值可以用线段树维护，这就需要我们维护每次选择的区间，每一次选择的区间左端点一定要在上一次选择出来的元素右方开始，右端点与选择的次数有关，第一次在第n-m+1位，每一次右端点++即可保证剩余元素能被选择最终填满m位。
我构造了一个映射ys[x][k]，来表示数字x的第k次出现的位置，用数组cnt[ ]辅助记录和查询。查询时我通过while(ys[ans][cnt[ans]] < head) cnt[ans]++; 来找到在区间内的第一个出现的最大值的位置。那么下一次的区间左端点也找到了。以此类推，m次O(logn)的查询即可结束，而且不需要更新修改的线段树，总体看下来不难。
具体见代码：

#include<bits/stdc++.h>
#define maxn 200005
#define maxm 30000005
#define FOR(a, b, c) for(int a=b; a<=c; a++)
#define hrdg 1000000007
#define inf 2147483647
#define llinf 9223372036854775807
#define ll long long
#define pi acos(-1.0)
#define ls p<<1
#define rs p<<1|1
using namespace std;

int n, m;
int tr[maxn<<1], a[maxn], ys[20][maxn], cnt[20];
char ch;

void pushup(int p)			//维护区间最大值的无修改线段树
{
    tr[p] = max(tr[ls], tr[rs]);
}
void build(int p, int l, int r)
{
    if(l==r)
    {
        tr[p] = a[l];
        return;
    }
    int mid = (l+r)>>1;
    build(ls, l, mid);
    build(rs, mid+1, r);
    pushup(p);
}
int query(int nl, int nr, int l, int r, int p)
{
    int ret = 0;
    if(nl <= l && nr>= r) return tr[p];
    int mid = (l+r)>>1;
    if(nl <= mid)
        ret = max(ret, query(nl, nr, l, mid, ls));
    if(nr >= mid+1)
        ret = max(ret, query(nl, nr, mid+1, r, rs));
    return ret;
}

int main()
{
    while(cin>>n>>m)
    {
        memset(tr, 0, sizeof(tr));
        memset(cnt, 0, sizeof(cnt));
        memset(ys, 0, sizeof(ys));
        memset(a, 0, sizeof(a));
        FOR(i, 1, n)
        {
            cin>>ch;
            if(ch>='0' && ch<='9') a[i] = ch-'0';
            if(ch>='a' && ch<='f') a[i] = ch-'a'+10;
            cnt[a[i]]++;
            ys[a[i]][cnt[a[i]]] = i;			//记录每次数字a[i]出现时的位置
        }
        //FOR(i, 1, n) printf("%d ", a[i]); cout<<endl;
        build(1, 1, n);				//建树
        int t = 0;
        memset(cnt, 0, sizeof(cnt));		//辅助cnt[]数组清空
        while(m--)
        {
            int head = t + 1, tail = n - m;
            int ans = query(head, tail, 1, n, 1);
            while(ys[ans][cnt[ans]] < head) cnt[ans]++;		//找到区间内该最值第一次出现的位置
            t = ys[ans][cnt[ans]];
            if(ans>=10) printf("%c", ans-10+'a');
            else printf("%d", ans);
        }
        printf("\n");
    }
    return 0;
}