Leetcode 525. 连续数组（前缀和性质/map）

给定一个二进制数组 nums , 找到含有相同数量的 0 和 1 的最长连续子数组，并返回该子数组的长度。

示例 1:

输入: nums = [0,1]
输出: 2
说明: [0, 1] 是具有相同数量0和1的最长连续子数组。
示例 2:

输入: nums = [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。

提示：

1 <= nums.length <= 105
nums[i] 不是 0 就是 1

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/contiguous-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

看到连续子数组 + 相同数量一眼前缀和水题。注意到：

对于一个区间[l, r]为候选区间当且仅当one[r] - one[l - 1] == zero[r] - zero[l - 1]，移项即one[r] - zero[r] == one[l - 1] - zero[l - 1];

因此求出diff[i] = one[i] - zero[i]，找到数组中相距最远的两个diff[i]和diff[j]即可。可以直接暴力map，由于范围不大也可以用桶，需要注意的是用桶的话需要对负数偏移一下。

class Solution {
public:
    int n, a[100005], one[100005], zero[100005], d[100005];
    map<int, int> mp;
    int findMaxLength(vector<int>& nums) {
        n = nums.size();
        one[0] = zero[0] = 0;
        for(int i = 1; i <= n; i++) {
            a[i] = nums[i - 1];
            one[i] = one[i - 1];
            zero[i] = zero[i - 1];
            if(a[i]) one[i]++;
            else zero[i]++;
        }
        //对于l, r当且仅当one[r] - one[l - 1] == zero[r] - zero[l - 1]
        //即one[r] - zero[r] == one[l - 1] - zero[l - 1];
        for(int i = 1; i <= n; i++) {
            d[i] = one[i] - zero[i];
        }
        int ans = 0;
        for(int i = 0; i <= n; i++) {
            if(mp.find(d[i]) == mp.end()) {
                mp[d[i]] = i;
            } else {
                ans = max(ans, i - mp[d[i]]);
            }
        }
        return ans;
    }
};