题目大意:在给定的长度为$n(nleqslant5 imes10^5)$的字符串$A$和字符串$B$中找到最多$k$个字符串,使得这$k$个字符串不同的前缀字符串的数量最多(只包含字符$a$和$b$)。
题解:考虑给这$k$个字符串建一个$trie$树,答案就是它所含的节点数,考虑贪心,在每一层尽可能多的分叉。注意一层最多只能有$k$个点
卡点:无
C++ Code:
#include <cstdio> #define maxn 500010 int n, k; long long ans; char A[maxn], B[maxn]; int main() { scanf("%d%d", &n, &k); scanf("%s%s", A, B); long long now = 1; for (int i = 0; i < n; ++i) { now <<= 1; now -= (A[i] == 'b') + (B[i] == 'a'); if (now >= k) { printf("%lld ", ans + static_cast<long long> (n - i) * k); return 0; } ans += now; } printf("%lld ", ans); return 0; }