旺崽的博客

要么天赋异禀,要么天道酬勤

0%

LeetCode 1208 尽可能使字符串相等

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;
}
};