【贪心/Trie】【CF1083B】 The Fair Nut and Strings

Description

有 \(k\) 个长度为 \(n\) 的只含 \(a\) 或 \(b\) 字符串，并不知道它们具体是多少，只知道它们的字典序不小于字符串 \(A\)，同时不大于字符串 \(B\)。定义一个字符串是合法的当且仅当它是这 \(k\) 个字符串之一的前缀（如果它是多个串的前缀那么只计算一次）。求合法的字符串最大可能是多少

Input

第一行是两个整数 \(n\) 和 \(k\)

下面两行，第一行是长度为 \(n\) 的字符串 \(A\)，第二行是长度为 \(n\) 的字符串 \(B\)

Output

输出一个数代表答案。

Hint

\(1~\leq~n~\leq~5~\times~10^5~,~1~\leq~k~\leq~10^9\)

Solution

我们考虑假如对这 \(k\) 个字符串建出一棵踹Trie树，那么显然一个节点对应一个合法的字符串，答案即为树上节点个数。于是我们的问题即被转化为了最大化Trie树上的节点个数。考虑在一个节点，在合法的条件下孩子数为 \(2\) 的答案显然不劣于孩子数为 \(1\) 的答案。于是我们依照此按照层数进行贪心，尽可能的多分节点即可。考虑因为最后一层的节点最多有 \(k\) 个，所以当算到一层的节点数不小于 \(k\) 时，后面就无需枚举，直接分配给每层 \(k\) 个节点计算答案即可。

Code

#include <cstdio>
#include <algorithm>
#ifdef ONLINE_JUDGE
#define freopen(a, b, c)
#endif
#define rg register
#define ci const int
#define cl const long long

typedef long long int ll;

namespace IPT {
    const int L = 1000000;
    char buf[L], *front=buf, *end=buf;
    char GetChar() {
        if (front == end) {
            end = buf + fread(front = buf, 1, L, stdin);
            if (front == end) return -1;
        }
        return *(front++);
    }
}

template <typename T>
inline void qr(T &x) {
    rg char ch = IPT::GetChar(), lst = ' ';
    while ((ch > '9') || (ch < '0')) lst = ch, ch=IPT::GetChar();
    while ((ch >= '0') && (ch <= '9')) x = (x << 1) + (x << 3) + (ch ^ 48), ch = IPT::GetChar();
    if (lst == '-') x = -x;
}

template <typename T>
inline void ReadDb(T &x) {
    rg char ch = IPT::GetChar(), lst = ' ';
    while ((ch > '9') || (ch < '0')) lst = ch, ch = IPT::GetChar();
    while ((ch >= '0') && (ch <= '9')) x = x * 10 + (ch ^ 48), ch = IPT::GetChar();
    if (ch == '.') {
        ch = IPT::GetChar();
        double base = 1;
        while ((ch >= '0') && (ch <= '9')) x += (ch ^ 48) * ((base *= 0.1)), ch = IPT::GetChar();
    }
    if (lst == '-') x = -x;
}

namespace OPT {
    char buf[120];
}

template <typename T>
inline void qw(T x, const char aft, const bool pt) {
    if (x < 0) {x = -x, putchar('-');}
    rg int top=0;
    do {OPT::buf[++top] = x % 10 + '0';} while (x /= 10);
    while (top) putchar(OPT::buf[top--]);
    if (pt) putchar(aft);
}

const int maxn = 500010;

ll n, k, ans;
char MU[maxn], CU[maxn];

int main() {
    freopen("1.in", "r", stdin);
    qr(n); qr(k);
    do MU[1] = IPT::GetChar(); while ((MU[1] > 'z') || (MU[1] < 'a'));
    for (rg int i = 2; i <= n; ++i) MU[i] = IPT::GetChar();
    do CU[1] = IPT::GetChar(); while ((CU[1] > 'z') || (CU[1] < 'a'));
    for (rg int i = 2; i <= n; ++i) CU[i] = IPT::GetChar();
    ll pre = 1;
    for (rg int i = 1; i <= n; ++i) {
        pre <<= 1;
        if (MU[i] == 'b') --pre;
        if (CU[i] == 'a') --pre;
        if (pre >= k) {
            ans += 1ll * k * (n - i + 1);
            break;
        }
        ans += pre;
    }
    qw(ans, '\n', true);
    return 0;
}

秒客网