题目是这样的:一个一维数组中只有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)。
代码如下:
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
来源:oschina
链接:https://my.oschina.net/u/4267186/blog/3620560