原題鏈接

鏈接：https://www.nowcoder.net/questionTerminal/fcd645af2750401588d9e92aea05d630

有一個(gè)僅包含’a’和’b’兩種字符的字符串s，長(zhǎng)度為n，每次操作可以把一個(gè)字符做一次轉(zhuǎn)換（把一個(gè)’a’設(shè)置為’b’，或者把一個(gè)’b’置成’a’)；但是操作的次數(shù)有上限m，問(wèn)在有限的操作數(shù)范圍內(nèi)，能夠得到最大連續(xù)的相同字符的子串的長(zhǎng)度是多少。

輸入描述:

第一行兩個(gè)整數(shù) n , m (1<=m<=n<=50000)，第二行為長(zhǎng)度為n且只包含’a’和’b’的字符串s。

輸出描述:

輸出在操作次數(shù)不超過(guò) m 的情況下，能夠得到的 最大連續(xù) 全’a’子串或全’b’子串的長(zhǎng)度。 示例：

輸入

8 1 aabaabaa

輸出

5

說(shuō)明

把第一個(gè) 'b' 或者第二個(gè) 'b' 置成 'a'，可得到長(zhǎng)度為 5 的全 'a' 子串。

————————————?? 分界線————————————————

分析：

該字符串非 a 即 b 也就是說(shuō)在區(qū)間 l~r之間把所有字符變?yōu)?a 所需的步驟數(shù)是 該區(qū)間內(nèi) b 的數(shù)量。反之亦然. 用數(shù)組 count[i] 表示 字符串中位置區(qū)間 0~i 包含的 a 的個(gè)數(shù) 則 區(qū)間 l~r 的 a 的個(gè)數(shù)為 count[r]-count[l-1] b 的個(gè)數(shù)用 a 的個(gè)數(shù)算出 即 區(qū)間 l~r 的 b 的個(gè)數(shù)為 r+1-count[r]-(l+1-1-count[l-1])=r+1-count[r]-l+count[l-1]在區(qū)間 l~r 的 a 和 b 的個(gè)數(shù)已知的情況下 若 區(qū)間長(zhǎng)度step內(nèi)的 a 的個(gè)數(shù) <= m 則 可以通過(guò) m 個(gè)步驟 產(chǎn)生 長(zhǎng)度為step的字符串 b 若 區(qū)間長(zhǎng)度step內(nèi)的 b 的個(gè)數(shù) <= m 則 可以通過(guò) m 個(gè)步驟 產(chǎn)生 長(zhǎng)度為step的字符串 a歸納為 ：若 區(qū)間長(zhǎng)度step內(nèi)的 b 或 a 的個(gè)數(shù) <= m 則 可以通過(guò) m 個(gè)步驟 產(chǎn)生 長(zhǎng)度為step的字符串這樣 就可以直接計(jì)算出一個(gè)字符串長(zhǎng)度（區(qū)間長(zhǎng)度step）是否可行，因此不需要進(jìn)行遞推，可以直接進(jìn)行二分搜索，得到最大長(zhǎng)度。 檢查一個(gè)長(zhǎng)度step是否可行的時(shí)間復(fù)雜度為O(n),二分搜索的時(shí)間復(fù)雜度為O(log n)。 因此，該方法總的時(shí)間復(fù)雜度為 O(n*log n)

代碼：

#include<iostream> #include<string> #include<cstring> using namespace std; int count[50005]; int n,m; int ans=0;//檢查當(dāng)前區(qū)間長(zhǎng)度(step)是否能在 m 個(gè)步驟內(nèi)實(shí)現(xiàn)全 a 或 全 b bool func(int step){for(int i=0;i+step<n;i++){if(m>=step+1-(count[i+step]-count[i-1]))return true; //檢查 a 公式 m>=區(qū)間內(nèi) b 的個(gè)數(shù) if(m>=count[i+step]-count[i-1])return true; //檢查 b 公式： m>=區(qū)間內(nèi) a 的個(gè)數(shù) }return false; }int main(){cin>>n>>m;string str;cin>>str;//輸入并計(jì)算出 count 數(shù)組 int sum=0;for(int i=0;i<str.size();i++){if(str[i]=='a')count[i]=++sum;else count[i]=sum;}//二分搜索最大區(qū)間值 int l=0,r=n-1,mid; while(l<r){mid=l+(r-l)/2;if(func(mid)){l=mid+1;}else r=mid;}ans=max(ans,l);cout<<ans<<endl;return 0;}

總結(jié)

以上是生活随笔為你收集整理的题解-今日头条2018校招后端方向（第四批）的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。