Vasya and String CodeForces 676C 前缀和+二分

最新推荐文章于 2022-03-04 00:07:49 发布

_Sheepsheep

最新推荐文章于 2022-03-04 00:07:49 发布

阅读量121

点赞数

分类专栏： 10月训练日志-二分

行百里者半九十

本文链接：https://blog.csdn.net/qq_42106422/article/details/121000297

版权

字符串处理动态规划二分查找字符替换算法题

关键词由CSDN通过智能技术生成

10月训练日志-二分专栏收录该内容

9 篇文章 0 订阅

订阅专栏

Vasya and String CodeForces 676C

传送门：https://codeforces.com/problemset/problem/676/C

Description
High school student Vasya got a string of length $n$ as a birthday present. This string consists of letters ‘a’ and ‘b’ only. Vasya denotes beauty of the string as the maximum length of a substring (consecutive subsequence) consisting of equal letters.

Vasya can change no more than $k$ characters of the original string. What is the maximum beauty of the string he can achieve?

Input
The first line of the input contains two integers $n$ and $k (1 \leq n \leq 100000, 0 \leq k \leq n)$ — the length of the string and the maximum number of characters to change.

The second line contains the string, consisting of letters ‘a’ and ‘b’ only.

Output
Print the only integer — the maximum beauty of the string Vasya can achieve by changing no more than $k$ characters.

题意：
给出一串长度为 $n$ 的字符串，字符串中只有 $‘ a ’$ 或者 $^{'} b^{'}$ ，我们可以改变 $k$ 次字母，我们可以得到的相同的连续子串最长有多长。

思路：
假设 $x$ 是猜测的能得到的最长的连续的子串长度， $s$ 是真实的能得到的最长的连续的子串长度，那么对于所有可行的 $x$ ，都有 $x < = s$
且 $f (x) = (x < = s ? 1 : 0)$ 具有单调性，所以可以二分找答案。
用 $s u m a$ 数组和 $s u m b$ 数组记录在 $1$ 到 $i$ 这段区间里有多少个 $a$ （有多少个 $b$ ），然后二分找答案。在 $c h e c k$ 函数中，我们可以像比着一把尺子一样检测，看对于当前的答案 $x$ 是否合法。如果 $i$ 到 $i + x$ 这个区间合法，即 $i$ 到 $i + x$ 可以构成连续的一串 $a$ 或者连续的一串 $b$ ,那么一定有
连续的a成立： $s u m b [i + x - 1] - s u m b [i - 1] < = k$
意思是 $i$ 到 $i + x$ 这个区间的 $b$ 可以全部被替换成 $a$
同理，如果连续的 $b$ 成立： $s u m a [i + x - 1] - s u m a [i - 1] < = k$
意思是 $i$ 到 $i + x$ 这个区间的 $a$ 可以全部被替换成 $b$

代码：

#include<iostream>
using namespace std ;
#define ll long long
const ll N = 1e5+9 ;
void updata( ll &a , ll b ){if(b>a)a=b;}
string s ; ll n , k ;
ll suma[ N ] , sumb[ N ] ;
bool check( ll maxn ){
    for( int i = 1 ; i + maxn - 1 <= n ; i ++ ){
        if( suma[ i + maxn - 1 ] - suma[ i - 1 ] <= k || sumb[ i + maxn - 1 ] - sumb[ i - 1 ] <= k ) return 1 ;
    }
    return 0 ;
}
int main(){
    cin >> n >> k >> s ; s = "?"+s ;
    for( int i = 1 ; i <= n ; i ++ ){
        if( s[ i ] == 'a' ){ suma[ i ] = suma[ i - 1 ] + 1 ; sumb[ i ] = sumb[ i - 1 ] ; }
        else { suma[ i ] = suma[ i - 1 ] ; sumb[ i ] = sumb[ i - 1 ] + 1 ; }
    }
    ll l , r , mid , ans = -1 ;
    l = 1 , r = 2*n+10000 ;
    while( l <= r ){
        mid = l + r >> 1 ;
        ll f = check( mid ) ;
        if( f ) { l = mid + 1 ; updata( ans , mid ) ; }
        else r = mid - 1 ;
    }
    cout << ans << "\n" ;
return 0 ;
}