LeetCode 面试题 17.05 字母与数字【前缀和 Map】HERODING的LeetCode之路

在这里插入图片描述

解题思路:
最近遇到不少道前缀和和哈希表结合的题目,这种用空间换时间的方式非常好用,所以以这一题为例。首先要求找到子数组中数字和字母的个数相等,那么我们可以用前缀和的方法,遍历整个数组,遇到数字-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;        
    }
};

猜你喜欢

转载自blog.csdn.net/HERODING23/article/details/129472691