有一个仅由0和1组成的01串,找到其中最长的一段子串,使得该子串中0和1的数目相等。
面试时将0改成-1,然后用前缀和,二分枚举长度然后检验,这里二分枚举错了,不满足二分的性质。
正确思路是,在前缀和的基础上,找前缀和相等的 i 和 j 并且 |i - j| 尽可能大。
这里用哈希存一下每个前缀和的最小 id ,然后O(n)跑一下取最大距离即可。
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 10;
int arr[maxn], pre_sum[maxn];
unordered_map <int, int> mp;
int main() {
int n; cin >> n;
mp[0] = 0;//特别的
for(int i = 1; i <= n; i++){
scanf("%1d", &arr[i]);
arr[i] = arr[i] == 0 ? -1 : 1;
int x = pre_sum[i] = pre_sum[i - 1] + arr[i];
//int x = pre_sum[i];
if(x == 0){
continue;//0的最小下标不用更新
}
mp[x] = mp[x] ? mp[x] : i;//存储最小下标
}
int ans = 0;
for(int i = 1; i <= n; i++){
ans = max(ans, i - mp[pre_sum[i]]);
}
cout << ans << endl;
return 0;
}