一个一维维数组中只有1和-1，实现程序，求和为0的最长子串长度，并在注释中给出时间和空间复杂度

这是一个比动态规划更简单的解法。

思路就是在i从0到n，计算sum(i)，sum(i)表示从0到i的元素之和。并保存在字典dic中，value是索引i，在往后的遍历中每得到一个sum(i)就查看dic的keys是否已有此sum(i)值，如果有则用当前i位置减去保存的i，并与maxLen比较，取大的那个。遍历结束，给出结果。时间复杂度O(n)，空间复杂度O(1)。

原链接里是用python写的，我又用Java写了一下

我把hash里的key和value与思路里的变了一下位置。所以要先判断hashMap.get(sum)有没有值。有值就比较，没值就存值。

sum是指从0到i的和，i是第几个数。注意一下最开始的sum是0，所以一开始要hashMap.put(0,-1);-1是比第0个数更早的初始sum。

import java.util.HashMap;

public class yuantiku {
    public static int getLongestLength(int []a) {
        int max=0;
        int sum=0;
        HashMap<Integer,Integer> hashMap=new HashMap<>();
        hashMap.put(0,-1);
        for (int i=0;i<a.length;i++){
            sum=sum+a[i];
            if (hashMap.get(sum)==null){
                hashMap.put(sum,i);
            }else{
                int tmp=i-hashMap.get(sum);
                if (max<tmp){
                    max=tmp;
                }
            }
        }
        return max;

    }

    public static void main(String[] args) {
        int []a=new int[]{ -1, -1,-1, -1, -1, 1, 1, 1};
        System.out.println(getLongestLength(a));
    }
}

 

 

下面是简书原文

前几天猿题库的笔试题，想了挺久不知道咋做，中午问了慧伟，刚给出了答案。一直以为用动态规划，其实不然。

题目是这样的：一个一维数组中只有1和-1，实现程序，求和为0的最长子串长度，并在注释中给出时间和空间复杂度。

思路就是在i从0到n，计算sum(i)，sum(i)表示从0到i的元素之和。并保存在字典dic中，value是索引i，在往后的遍历中每得到一个sum(i)就查看dic的keys是否已有此sum(i)值，如果有则用当前i位置减去保存的i，并与maxLen比较，取大的那个。遍历结束，给出结果。时间复杂度O(n)，空间复杂度O(1)。

Python代码：

# coidng:utf-8
# @sinner
# 16/9/8

class Solution(object):
    def fun(self, l):
        dic = {0:-1}
        sum = 0
        maxLen = 0
        for x in xrange(0, len(l)):
            sum += l[x]
            if sum in dic: maxLen = max(maxLen, x - dic[sum])
            else: dic[sum] = x
        return maxLen


print Solution().fun([1, -1, -1, -1, 1, 1, 1, -1, -1])

输出为8.

 

作者：贰拾贰画生
链接：https://www.jianshu.com/p/3bf87384a92b
來源：简书
简书著作权归作者所有，任何形式的转载都请联系作者获得授权并注明出处。