C. Flexible String

Problem - C - Codeforces

题意：给定长度均为n的字符串a和b，最多可以替换k个字母，求替换后最多有多少个区间[l,r]满足a[l,r]等于b[l,r]

思路：注意到数据范围最多有10个不同的字符，我们可以二进制枚举替换哪些字符。贪心的策略是能替换就替换，相同的不需要替换。一个长度为n的子串对答案的贡献是n*(n+1)/2

#include <bits/stdc++.h>
#define lowbit(x) x&(-x)
#define ios cin.sync_with_stdio(false)
#define PII pair<int,int>
#define int long long
typedef long long ll;
const int N=1e6+10;
const int inf=0x3f3f3f3f;

using namespace std;
int n,k;
string a,b;
int check(int x)
{
    int ret=0;
    for(int i=0;i<=10;i++)
    {
        if(x>>i&1) ret++;
    }
    return ret;
}
void solve()
{
    cin>>n>>k;
    cin>>a>>b;
    map<char,int>mp;
    int cnt=0;
    for(int i=0;i<n;i++) 
    {
        if(mp.count(a[i])==0)
        {
            mp[a[i]]=++cnt;
        }
    }
    int ans=0;
    for(int i=0;i<1<<(cnt);i++)
    {
        if(check(i)>k) continue;
        int ret=0;
        int len=0;
        for(int j=0;j<n;j++)
        {
            if(a[j]==b[j]||(mp[a[j]]&&(i>>(mp[a[j]]-1))&1)) len++;
            else
            {
                ret+=len*(len+1)/2;
                len=0;
            }
        }
        ret+=len*(len+1)/2;
        ans=max(ans,ret);
    }
    cout<<ans<<'\n';
}
signed main()
{
    //ios;
    int _t=1;
    cin>>_t;
    while(_t--) solve();
    system("pause");
    return 0;
}