2023牛客寒假算法基础集训营5 F-小沙の串串 线段树写法

题目描述：

给出一个长度为 n 的字符串和 k 次操作，每次操作可以任意选择一个字符，将其移动到字符串的最后。 输出恰好操作 k 次之后字典序最大的字符串。

输入：

第一行n，k（1<=n<=1e5,1<=k<=1e9）

第二行输入长度为n的字符串

输出：

k次操作后字典序最大的字符串

Solution：

思路：

显然，要使一个字符串s最大，第一步便是将s中最大的字符放到开头，即优先使字符串s开头最大，而在这题中，我们也可以借着这个思路去做。

首先我们找到s中最大字符的第一个在s中的位置，我们设为p。

例如字串aaccccbb中的p=3。

下面我们考虑以下3种情况：

①当p-1==k时，我们可以简单地发现只要把p-1的所有字符取出降序后放到s的最后，即使这种情况下的所求答案。

②当p-1<k时，我们会发现要使s在k次操作后字典序最大，由开头思路可知，要使字符串字典序最大，那便一定在s[1,p-1]中取出k个字符，那么接下来的操作便是继续对s[1,p-1]进行这三种情况的考虑。

③当p-1>k时，显然s[1,p-1]的部分可以全部取出，然后将k-p+1，此时我们只需对s[p+1，n]进行这三种情况的考虑。

代码具体实现：

但是由于我们要求任意区间的最大值的第一个在s中的位置，而区间范围则是1e5，故此处我们采用线段树的方式来维护任意区间的所求值。

而在具体情况时，我们需特判当n-1<=k时，此时一定能使s在k次操作后所求最大字典序最优，故我们只需考虑n-1>k时的情况，故k实际范围只为1e5,最终时间复杂度为nlogn。

同时由于我们要对初始字符串进行取出删除操作,对于取出，我们用一个新的字符串ed去储存所有取出的元素，然后只要对ed进行降序排序就是在尾部添加的字符串。对于删除，由于每个字符不会被重复访问，所以我们只需要把取出的字符变成‘A’即可，这样我们在输出保留的字符串时，只需要输出s中不等于‘A’的部分。

C++code：

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int n, k, cnt = 0;
int sh[N];
char s[N], ed[N];
struct Node {//v定义为l,r区间中的最大值,p定义为l,r区间中的最大值的第一个在字符串中的位置
    int l, r;
    int v, p;
}tr[4 * N];

void pushup(Node& u, Node& l, Node& r) {//更新父节点为l,r区间中最大值的第一个在字符串中的位置
    if (l.v == r.v)u.v = r.v, u.p = min(l.p, r.p);
    else if (l.v > r.v)u.v = l.v, u.p = l.p;
    else if (l.v < r.v)u.v = r.v, u.p = r.p;
}
void pushup(int u) {
    pushup(tr[u], tr[u << 1], tr[u << 1 | 1]);
}

void build(int u, int l, int r) {//建树
    tr[u] = { l,r };
    if (l == r) {
        tr[u].v = sh[r];
        tr[u].p = r;
    }
    else {
        int mid = l + r >> 1;
        build(u << 1, l, mid);
        build(u << 1 | 1, mid + 1, r);
        pushup(u);
    }
}

Node query(int u, int l, int r) {//求l,r中第一个最大值在线段树中的节点
    if (tr[u].l >= l && tr[u].r <= r)return tr[u];
    else {
        int mid = tr[u].l + tr[u].r >> 1;
        if (r <= mid)return query(u << 1, l, r);
        else if (l > mid)return query(u << 1 | 1, l, r);
        else {
            auto left = query(u << 1, l, r);
            auto right = query(u << 1 | 1, l, r);
            Node res;
            pushup(res, left, right);
            return res;
        }
    }
}

void trans(int l, int r) {//将字符串s中区间l,r的字符添加到字符串ed中并在s中删去，此处将s[i]变成'A'表示删去
    for (int i = l; i <= r; i++) {
        ed[++cnt] = s[i];
        s[i] = 'A';
    }
}
void solve(int l, int r) {//判断s的区间l,r中第一个最大值的位置和k的关系后再做分支
    if (r - l + 1 <= k) {//当l,r中个数都小于k时便删去l,r，存入ed中
        trans(l, r);
        return;
    }

    int now = query(1, l, r).p;
    if (now - l == k) trans(l , now - 1);
    else if (k < now - l) solve(l, now - 1);
    else if (k > now - l) {
        trans(l, now - 1);
        k -= (now - l);
        solve(now + 1, r);
    }
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> k;
    for (int i = 1; i <= n; i++) {
        cin >> s[i];
        sh[i] = s[i] - 'a' + 1;
    }
    if (n - 1 <= k) {//由于k>=n-1时一定
        sort(s + 1, s + 1 + n);//降序排序s
        reverse(s + 1, s + 1 + n);
        cout << s + 1 << endl;
    }
    else {
        build(1, 1, n);
        solve(1, n);

        sort(ed + 1, ed + 1 + cnt);//降序排序ed
        reverse(ed + 1, ed + 1 + cnt);

        for (int i = 1; i <= n; i++) {//输出s中未被删去的
            if (s[i] != 'A')cout << s[i];
        }
        for (int i = 1; i <= cnt; i++)cout << ed[i];
    }
    return 0;
}