|
55 | 55 |
|
56 | 56 | <!-- 这里可写通用的实现逻辑 -->
|
57 | 57 |
|
| 58 | +前缀和 + 二分查找。 |
| 59 | + |
58 | 60 | <!-- tabs:start -->
|
59 | 61 |
|
60 | 62 | ### **Python3**
|
61 | 63 |
|
62 | 64 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
63 | 65 |
|
64 | 66 | ```python
|
65 |
| - |
| 67 | +class Solution: |
| 68 | + def equalSubstring(self, s: str, t: str, maxCost: int) -> int: |
| 69 | + n = len(s) |
| 70 | + presum = [0] * (n + 1) |
| 71 | + for i in range(n): |
| 72 | + presum[i + 1] = presum[i] + abs(ord(s[i]) - ord(t[i])) |
| 73 | + left, right = 0, n |
| 74 | + |
| 75 | + def check(l): |
| 76 | + i = 0 |
| 77 | + while i + l - 1 < n: |
| 78 | + j = i + l - 1 |
| 79 | + if presum[j + 1] - presum[i] <= maxCost: |
| 80 | + return True |
| 81 | + i += 1 |
| 82 | + return False |
| 83 | + |
| 84 | + while left < right: |
| 85 | + mid = (left + right + 1) >> 1 |
| 86 | + if check(mid): |
| 87 | + left = mid |
| 88 | + else: |
| 89 | + right = mid - 1 |
| 90 | + return left |
66 | 91 | ```
|
67 | 92 |
|
68 | 93 | ### **Java**
|
69 | 94 |
|
70 | 95 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
71 | 96 |
|
72 | 97 | ```java
|
| 98 | +class Solution { |
| 99 | + public int equalSubstring(String s, String t, int maxCost) { |
| 100 | + int n = s.length(); |
| 101 | + int[] presum = new int[n + 1]; |
| 102 | + for (int i = 0; i < n; ++i) { |
| 103 | + presum[i + 1] = presum[i] + Math.abs(s.charAt(i) - t.charAt(i)); |
| 104 | + } |
| 105 | + int left = 0, right = n; |
| 106 | + while (left < right) { |
| 107 | + int mid = (left + right + 1) >>> 1; |
| 108 | + if (check(mid, presum, maxCost, n)) { |
| 109 | + left = mid; |
| 110 | + } else { |
| 111 | + right = mid - 1; |
| 112 | + } |
| 113 | + } |
| 114 | + return left; |
| 115 | + } |
| 116 | + |
| 117 | + private boolean check(int l, int[] s, int maxCost, int n) { |
| 118 | + int i = 0; |
| 119 | + while (i + l - 1 < n) { |
| 120 | + int j = i + l - 1; |
| 121 | + if (s[j + 1] - s[i] <= maxCost) { |
| 122 | + return true; |
| 123 | + } |
| 124 | + ++i; |
| 125 | + } |
| 126 | + return false; |
| 127 | + } |
| 128 | +} |
| 129 | +``` |
| 130 | + |
| 131 | +### **C++** |
| 132 | + |
| 133 | +```cpp |
| 134 | +class Solution { |
| 135 | +public: |
| 136 | + int equalSubstring(string s, string t, int maxCost) { |
| 137 | + int n = s.size(); |
| 138 | + vector<int> presum(n + 1); |
| 139 | + for (int i = 0; i < n; ++i) presum[i + 1] = presum[i] + abs(s[i] - t[i]); |
| 140 | + int left = 0, right = n; |
| 141 | + while (left < right) |
| 142 | + { |
| 143 | + int mid = left + right + 1 >> 1; |
| 144 | + if (check(mid, presum, maxCost, n)) left = mid; |
| 145 | + else right = mid - 1; |
| 146 | + } |
| 147 | + return left; |
| 148 | + } |
| 149 | + |
| 150 | + bool check(int l, vector<int>& s, int maxCost, int n) { |
| 151 | + int i = 0; |
| 152 | + while (i + l - 1 < n) |
| 153 | + { |
| 154 | + int j = i + l - 1; |
| 155 | + if (s[j + 1] - s[i] <= maxCost) return true; |
| 156 | + ++i; |
| 157 | + } |
| 158 | + return false; |
| 159 | + } |
| 160 | +}; |
| 161 | +``` |
73 | 162 |
|
| 163 | +### **Go** |
| 164 | + |
| 165 | +```go |
| 166 | +func equalSubstring(s string, t string, maxCost int) int { |
| 167 | + n := len(s) |
| 168 | + presum := make([]int, n+1) |
| 169 | + for i, c := range s { |
| 170 | + presum[i+1] = presum[i] + abs(int(c)-int(t[i])) |
| 171 | + } |
| 172 | + |
| 173 | + left, right := 0, n |
| 174 | + check := func(l int) bool { |
| 175 | + i := 0 |
| 176 | + for i+l-1 < n { |
| 177 | + j := i + l - 1 |
| 178 | + if presum[j+1]-presum[i] <= maxCost { |
| 179 | + return true |
| 180 | + } |
| 181 | + i++ |
| 182 | + } |
| 183 | + return false |
| 184 | + } |
| 185 | + for left < right { |
| 186 | + mid := (left + right + 1) >> 1 |
| 187 | + if check(mid) { |
| 188 | + left = mid |
| 189 | + } else { |
| 190 | + right = mid - 1 |
| 191 | + } |
| 192 | + } |
| 193 | + return left |
| 194 | +} |
| 195 | + |
| 196 | +func abs(x int) int { |
| 197 | + if x > 0 { |
| 198 | + return x |
| 199 | + } |
| 200 | + return -x |
| 201 | +} |
74 | 202 | ```
|
75 | 203 |
|
76 | 204 | ### **...**
|
|
0 commit comments