LeetCode 面试题 17.05 字母与数字前缀和 MapHERODING的LeetCode之路

Posted 2023-03-13 HERODING23

解题思路：
最近遇到不少道前缀和和哈希表结合的题目，这种用空间换时间的方式非常好用，所以以这一题为例。首先要求找到子数组中数字和字母的个数相等，那么我们可以用前缀和的方法，遍历整个数组，遇到数字-1，遇到字母+1，同时用Map将当前前缀和和其位置索引存储下来，这样在找子数组时，如果Map中已经存在了某一前缀和，那么该前缀和位置和当前前缀和位置所包含的区间里，字母和数字个数相同。找到这样最大的区间即可，代码如下：

class Solution 
public:
    vector<string> findLongestSubarray(vector<string>& array) 
        int n = array.size();
        int sum = 0;
        unordered_map<int, int> mp;
        mp[0] = 0;
        int right = 0, maxLen = 0;
        for(int i = 0; i < n; i ++) 
           sum += (isalpha(array[i][0])) ? 1 : -1;
            if(mp.count(sum)) 
                int len = i - mp[sum] + 1;
                if(len > maxLen) 
                    maxLen = len;
                    right = i;
                
             else 
                mp[sum] = i + 1;
            
        
        vector<string> res;
        if(right == 0) 
            return res;
        
        int left = right - maxLen + 1;
        res = vector<string>(array.begin() + left, array.begin() + right + 1);
        return res;        
    
;