Python程序找到等于数字n的连续数值的总和?

时间:2017-01-24 22:09:06

标签: python list sum-of-digits

我想在字符串中找到与给定数字相加的连续数字。

示例:

a="23410212"号码is=5 - 输出23,41,410,0212,212

此代码无效。我需要修理什么?

def find_ten_sstrsum():
    num1="2825302"
    n=0;
    total=0;
    alist=[];
    ten_str="";
    nxt=1;
    for n in range(len(num1)):
        for n1 in range(nxt,len(num1)):
            print(total)
            if(total==0):
                total=int(num1[n])+int(num1[n1])
                ten_str=num1[n]+num1[n1]
            else:
                total+=int(num1[n1])
                ten_str+=num1[n1]
            if(total==10):
                alist.append(ten_str)
                ten_str=""
                total=0
                nxt+=1
                break
            elif(total<10):
                nxt+=1
     return alist 

2 个答案:

答案 0 :(得分:5)

这种(类型)单行将起作用:

def find_ten_sstrsum(s, n):
  return list(  # list call only in Python 3 if you don't want an iterator
    filter(
      lambda y: sum(map(int, y))==n, 
      (s[i:j] for i in range(len(s)) for j in range(i+1, len(s)+1)))
  )

>>> find_ten_sstrsum('23410212', 5)
['23', '41', '410', '0212', '212']

这会使用嵌套的generator expression覆盖所有可能的slicesfilters,并使用正确的数字和。 当然,这远非最佳(特别是对于长字符串),因为一旦数字和超过n,内循环应该停止,但应该给你一个想法。

更高效,更易读的解决方案是generator function

def find_ten_sstrsum(s, n):
  for start in range(len(s)):
    for end in range(start+1, len(s)+1):
      val = sum(map(int, s[start:end]))
      if val > n:
        break
      if val == n:
        yield s[start:end]

>>> list(find_ten_sstrsum('23410212', 5))
['23', '41', '410', '0212', '212']

同时阅读summap

答案 1 :(得分:0)

您的功能有几个问题。最重要的是,您无法为其提供不同的数据。你有硬编码来处理一个特定字符串和总共10.你甚至编写了带有“十”的变量名,即使你的例子使用n = 5。您的函数应该有两个输入参数:给定的字符串和目标总和。

注意:schwobaseggl刚发布了一个可爱的Pythonic解决方案。但是,如果您需要更接近目前学习水平的功能,我会继续写这篇文章。

您有多条逻辑路径,因此很难理解您处理数据的方式。我建议采用略有不同的方法,以便您可以干净利落地处理每个部分金额:

for start in range(len(num1)):
    total = 0     # Start at 0 each time you get a new starting number.
    sum_str = ""
    for last in num1[start:]:
        print(total)
        # Don't create separate cases for the first and other additions.
        this_digit = num1[last]
        total += int(this_digit)
        ten_str += this_digit

        # If we hit the target, save the solution and
        #   start at the next digit
        if(total == target):
            alist.append(ten_str)
            break

        # If we passed the target, just
        #   start at the next digit
        elif(total > target):
            break

 return alist 

现在,这并没有解决相当你的所有问题,而且我还没有为你完成一些会计工作(变量初始化, def 行,等等。)。但是,我认为它会让您朝着正确的方向前进,并保留您的代码精神。