CF 1083 B. The Fair Nut and Strings

B. The Fair Nut and Strings 

题目链接

题意：

　　在给定的字符串a和字符串b中找到最多k个字符串，使得不同的前缀字符串的数量最多。

分析：

　　 建出trie树，给定的两个字符串就是trie树上的两条长度为n路径，那么就是在第n层的所有节点中，找到不大于k个点，（第n层的每个点向上到根的路径的路径上的字符组成一个长度为n字符串）。

　　两个第n层的节点一共会构成2n-lca个不同的前缀。所有可以根据这个贪心的选。每次尽量选已经走过的路径尽量少的路径。　

　　从n往后枚举，计算长度为答案为i的可以选几个。

　　注意判断一下最后是否可以选择k个， 和已经选了的路径。

代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<iostream>
#include<cctype>
#include<set>
#include<map>
#include<queue>
#include<vector>
#define fi(s) freopen(s,"r",stdin)
#define fo(s) freopen(s,"w",stdout)
using namespace std;
typedef long long LL;

inline int read() {
    int x=0,f=1;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-1;
    for(;isdigit(ch);ch=getchar())x=x*10+ch-'0';return x*f;
}

const int N = 1000005;
char a[N], b[N];

int main () {
    int n = read(); LL k; cin >> k;
    scanf("%s%s", a + 1, b + 1);
     
    LL now = 1, ans = 0, tot = 0;
    
    for (int i = 1; i <= n; ++i) {
        now = now * 2;
        if (a[i] == 'b') now --;
        if (b[i] == 'a') now --;
        if (now >= k) { tot = k; break; }
    }
    if (tot == 0) tot = now;
    
    now = 1;LL last = 0;
    for (int i = 1; i <= n; ++i) {
        now = 1ll * now * 2;
        if (a[i] == 'b') now --;
        if (b[i] == 'a') now --;
        LL t = min(k, now); t = min(t, tot); t = min(t, now - last); t = max(t, 0ll);
        ans += 1ll * t * (n - i + 1);
        k -= t; tot -= t; last += t;
        if ((!k) || (!tot)) break;
    }
    cout << ans;
    return 0;
}