1、-1组成的数组,求和为0的最长子串长度

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

题目是这样的:一个一维数组中只有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.

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容