LeetCode 525 连续数组

题目链接：LeetCode 525 连续数组

题目大意：
给定一个二进制数组\(nums\)，找到含有相同数量的\(0\)和\(1\)的最长连续子数组，并返回该子数组的长度。

题解：
将\(0\)换作\(-1\)，则问题变为求最长的连续子数组，其元素和为\(0\)。
首先计算数组的前缀和，设\(preSum[j] - preSum[i] = 0\)，则表示存在长为\(j - i\)的连续子数组，其元素和为\(0\)。
可以用哈希表存储第一次出现的前缀和，之后再次出现相等的前缀和即可直接计算子数组长度。

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        int sum = 0, ans = 0;
        unordered_map<int, int> hashTable;
        hashTable[0] = -1;
        for (int i = 0; i < nums.size(); ++i) {
            sum += (nums[i] ? 1 : -1);
            if (hashTable.count(sum)) {
                ans = max(ans, i - hashTable[sum]);
            } else {
                hashTable[sum] = i;
            }
        }
        return ans;
    }
};