leetcode 17.05

面试题 17.05. 字母与数字

给定一个放有字母和数字的数组，找到最长的子数组，且包含的字母和数字的个数相同。

返回该子数组，若存在多个最长子数组，返回左端点下标值最小的子数组。若不存在这样的数组，返回一个空数组。

示例 1:

输入: [“A”,“1”,“B”,“C”,“D”,“2”,“3”,“4”,“E”,“5”,“F”,“G”,“6”,“7”,“H”,“I”,“J”,“K”,“L”,“M”]

输出: [“A”,“1”,“B”,“C”,“D”,“2”,“3”,“4”,“E”,“5”,“F”,“G”,“6”,“7”]

示例 2:

输入: [“A”,“A”]

输出: []

提示：

array.length <= 100000

思路分析

1. 因为数组中只存在字母和数字，所以我们可以把字母看成-1，数字看成1，求前缀和及在此前缀和之前出现过相同前缀和的位置差即可知道每个位置是否能够成为符合题意的子数组。
2. 如何求：直接使用变量sum，在是字母的地方++，数字则–，那么sum一定是一个值，我们只需要用哈希表保存sum值第一次出现的位置，当第以后出现sum值的时候，直接取出这个位置，然后就得到了当前sum值在当前位置的最长子数组，即i - p[sum] + 1, 当len 比这个长度更小的时候更新左端点和len，最终如果左端点未被更新过则不存在这样的子数组，被更新过则返回[l, l + len)的字符数组。

c++代码

class Solution {
public:
    vector<string> findLongestSubarray(vector<string>& array) {
        vector<string> ans;

        int len = 0, l = INT_MAX, length;
        int n = array.size();
 

        unordered_map<int,int> p;
        int sum = 0;
        for(int i = 0 ; i < n;i++){
        	// 只保存第一次出现的位置，保证子数组最长
            if(!p.count(sum))
                p[sum] = i;
            if(isDigital(array[i])) sum++;
            else sum--;
            if(p.count(sum) > 0){
            	// 存在sum并且长度需要更新才更新左端点
                if(len < i - p[sum] + 1){
                    len = i - p[sum] + 1;
                    l = p[sum];
                }
            }
        }
        if(l == INT_MAX) return ans;
        for(int i = l;i < l + len;i++) ans.push_back(array[i]);
       return ans;

    }
    // 判断是否是数字
    bool isDigital(string x){
        return x[0] - '0' >= 0 && x[0] - '0' < 10;
    }
};

这道题的分析就到这里，如有不正请指出问题，谢谢。