题目如下:
解题思路:本题包了一层壳,去掉外表后题目是给定一个正整数组成的数组,求出最长的一段子数组的长度,要求子数组的和不大于cost。解题方法也不难,记per_cost[i]为abs(s[i] - t[i])的值,cost[i]为sum(per_cost[0:i])的值。对于任意一个下标i,很容易通过二分查找的方法找出cost中另外一个下标j,使得cost[i:j] <= cost。
代码如下:
class Solution(object): def equalSubstring(self, s, t, maxCost): """ :type s: str :type t: str :type maxCost: int :rtype: int """ cost = [] amount = 0 per_cost = [] for cs,ct in zip(s,t): amount += abs(ord(cs) - ord(ct)) cost.append(amount) per_cost.append(abs(ord(cs) - ord(ct))) #cost.sort() #print cost #print per_cost import bisect res = -float('inf') for i in range(len(cost)): inx = bisect.bisect_right(cost,cost[i] + maxCost - per_cost[i]) res = max(res,inx - i) return res