找含有的0和1数量相同的子串的长度……设一个前缀和sum的变量，遇到1加1，遇到0减1，sum=0时即为0和1数量相等的位置，一开始觉得只要没次sum=0重新开始计数，保存最大值就可以了，但是忽略了一种情况，当sum[i]==sum[j]时，i~j中间的0和1的数量也是相同的，长度为i-j；在处理前缀和时，可以开一个数组记录前缀和为x时的位置i+1，如果在sum！=0时，前缀和vis[sum]不为0，即已出现过sum，那么此时也可更新子串的长度max(ans,i+1-v[sum])
#include<iostream> #include<algorithm> #include<cmath> #include<queue> #include<vector> #include<functional> #include<string>
#include<string.h> using namespace std; typedef long long ll; string s; int v[1000005*2]; //不要忘了*2 int main() { cin >> s; int ans = 0,sum=1e6; //sum一开始要设置为1e6+，数组下标不为0 memset(v,0,sizeof(v));
for (int i = 0; i < s.length(); i++) { if (s[i] == '0') sum--; else sum++; if (sum == 1e6) ans = i + 1; else if (!v[sum]) v[sum] = i + 1; else ans = max(ans, i + 1 - v[sum]); } cout << ans << endl; return 0; }