给你两个长度相同的字符串,s
和 t
。
将 s
中的第 i
个字符变到 t
中的第 i
个字符需要 |s[i] - t[i]|
的开销(开销可能为 0),也就是两个字符的 ASCII 码值的差的绝对值。
用于变更字符串的最大预算是 maxCost
。在转化字符串时,总开销应当小于等于该预算,这也意味着字符串的转化可能是不完全的。
如果你可以将 s
的子字符串转化为它在 t
中对应的子字符串,则返回可以转化的最大长度。
如果 s
中没有子字符串可以转化成 t
中对应的子字符串,则返回 0
。
示例 1:
输入:s = "abcd", t = "bcdf", maxCost = 3 输出:3 解释:s 中的 "abc" 可以变为 "bcd"。开销为 3,所以最大长度为 3。
示例 2:
输入:s = "abcd", t = "cdef", maxCost = 3
输出:1
解释:s 中的任一字符要想变成 t 中对应的字符,其开销都是 2。因此,最大长度为 1。
示例 3:
输入:s = "abcd", t = "acde", maxCost = 0 输出:1 解释:a -> a, cost = 0,字符串未发生变化,所以最大长度为 1。
提示:
1 <= s.length, t.length <= 10^5
0 <= maxCost <= 10^6
s
和t
都只含小写英文字母。
前缀和 + 二分查找。
class Solution:
def equalSubstring(self, s: str, t: str, maxCost: int) -> int:
n = len(s)
presum = [0] * (n + 1)
for i in range(n):
presum[i + 1] = presum[i] + abs(ord(s[i]) - ord(t[i]))
left, right = 0, n
def check(l):
i = 0
while i + l - 1 < n:
j = i + l - 1
if presum[j + 1] - presum[i] <= maxCost:
return True
i += 1
return False
while left < right:
mid = (left + right + 1) >> 1
if check(mid):
left = mid
else:
right = mid - 1
return left
class Solution {
public int equalSubstring(String s, String t, int maxCost) {
int n = s.length();
int[] presum = new int[n + 1];
for (int i = 0; i < n; ++i) {
presum[i + 1] = presum[i] + Math.abs(s.charAt(i) - t.charAt(i));
}
int left = 0, right = n;
while (left < right) {
int mid = (left + right + 1) >>> 1;
if (check(mid, presum, maxCost, n)) {
left = mid;
} else {
right = mid - 1;
}
}
return left;
}
private boolean check(int l, int[] s, int maxCost, int n) {
int i = 0;
while (i + l - 1 < n) {
int j = i + l - 1;
if (s[j + 1] - s[i] <= maxCost) {
return true;
}
++i;
}
return false;
}
}
class Solution {
public:
int equalSubstring(string s, string t, int maxCost) {
int n = s.size();
vector<int> presum(n + 1);
for (int i = 0; i < n; ++i) presum[i + 1] = presum[i] + abs(s[i] - t[i]);
int left = 0, right = n;
while (left < right)
{
int mid = left + right + 1 >> 1;
if (check(mid, presum, maxCost, n)) left = mid;
else right = mid - 1;
}
return left;
}
bool check(int l, vector<int>& s, int maxCost, int n) {
int i = 0;
while (i + l - 1 < n)
{
int j = i + l - 1;
if (s[j + 1] - s[i] <= maxCost) return true;
++i;
}
return false;
}
};
func equalSubstring(s string, t string, maxCost int) int {
n := len(s)
presum := make([]int, n+1)
for i, c := range s {
presum[i+1] = presum[i] + abs(int(c)-int(t[i]))
}
left, right := 0, n
check := func(l int) bool {
i := 0
for i+l-1 < n {
j := i + l - 1
if presum[j+1]-presum[i] <= maxCost {
return true
}
i++
}
return false
}
for left < right {
mid := (left + right + 1) >> 1
if check(mid) {
left = mid
} else {
right = mid - 1
}
}
return left
}
func abs(x int) int {
if x > 0 {
return x
}
return -x
}