LeetCode 1208 尽可能使字符串相等
题目链接
给你两个长度相同的字符串,s 和 t。
将 s 中的第 i 个字符变到 t 中的第 i 个字符需要 |s[i] - t[i]| 的开销(开销可能为 0),也就是两个字符的 ASCII 码值的差的绝对值。
用于变更字符串的最大预算是 maxCost。在转化字符串时,总开销应当小于等于该预算,这也意味着字符串的转化可能是不完全的。
如果你可以将 s 的子字符串转化为它在 t 中对应的子字符串,则返回可以转化的最大长度。
如果 s 中没有子字符串可以转化成 t 中对应的子字符串,则返回 0。
示例 1:
1 2 3
| 输入:s = "abcd", t = "bcdf", cost = 3 输出:3 解释:s 中的 "abc" 可以变为 "bcd"。开销为 3,所以最大长度为 3。
|
示例 2:
1 2 3
| 输入:s = "abcd", t = "cdef", cost = 3 输出:1 解释:s 中的任一字符要想变成 t 中对应的字符,其开销都是 2。因此,最大长度为 1。
|
示例 3:
1 2 3
| 输入:s = "abcd", t = "acde", cost = 0 输出:1 解释:你无法作出任何改动,所以最大长度为 1。
|
前缀和+二分~
先预处理所有位置差的绝对值的前缀和,然后对每一个位置,二分不超过 cost 的最大位置,不断更新答案即可,AC代码如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
| class Solution { public: int equalSubstring(string s, string t, int maxCost) { vector<int> dif; for (int i = 0; i < s.size(); i++) { if (dif.empty()) dif.push_back(abs(s[i] - t[i])); else dif.push_back(abs(s[i] - t[i]) + dif.back()); } int ans = 0; for (int i = 0; i < s.size(); i++) { if (abs(s[i] - t[i]) > maxCost) continue; else { int pos = upper_bound(dif.begin(), dif.end(), dif[i] + maxCost - abs(s[i] - t[i])) - dif.begin(); ans = max(ans, pos - i); } } return ans; } };
|