在编译此贪婪算法的python问题时遇到问题

时间:2020-04-05 20:17:10

标签: python list algorithm greedy

我正在尝试这个问题,这是一个贪婪的算法问题。庆祝晚会的问题。 当我运行它时,如您在下面看到的,它说列表索引必须是整数。 您能帮我吗,我是算法编码的新手。 我也愿意寻求更好和有效的解决方案。

问题:

a=[1,5.4,2.1,3.4,3.1,2.0,1.8,8.9,10,23,4,5,5,2,1.6,1.9]
a.sort()
q=0
z={}
for i in range(len(a)):
    if (a[q]+1.0)>=a[i]:
        if q not in z.keys():
            z[q]=[]
        z[q].append(a[i])
    else:
        q=a[i]
        if q not in z.keys():
            z[q]=[]
        z[q].append(a[i])

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-32-60bee6e37157> in <module>
      4 z={}
      5 for i in range(len(a)):
----> 6     if (a[q]+1.0)>=a[i]:
      7         if q not in z.keys():
      8             z[q]=[]

TypeError: list indices must be integers or slices, not float

2 个答案:

答案 0 :(得分:0)

else块的开头,您说q=a[i]。由于a中有浮点数,因此在循环中的某个时刻,q被设置为浮点数。即使该float类似于2.0,当您尝试将其用作列表索引时,python仍然会引发错误。要解决此问题,您需要从列表a中删除所有浮点数。

答案 1 :(得分:0)

问题是在将q分配给a中的值之后使用q

a=[1,5.4,2.1,3.4,3.1,2.0,1.8,8.9,10,23,4,5,5,2,1.6,1.9]
a.sort()
q=0
z={}
for i in range(len(a)):
    if (a[q]+1.0)>=a[i]: # this is the problem that you have an error
        if q not in z.keys():
            z[q]=[]
        z[q].append(a[i])
    else:
        q=a[i] #here you are assigned the value to q, which can be a float
        if q not in z.keys():
            z[q]=[]
        z[q].append(a[i])

当您检查if (a[q]+1.0)>=a[i]时,它将使用列表a并使用值q检查索引。由于该值可以是浮点数,因此可能会出错,因为index必须为int。

您可以更改循环以跟踪索引:

a=[1,5.4,2.1,3.4,3.1,2.0,1.8,8.9,10,23,4,5,5,2,1.6,1.9]
a.sort()
q=0
qidx=0
z={}
for i in range(len(a)):
    if (a[qidx]+1.0)>=a[i]:
        if q not in z.keys():
            z[q]=[]
        z[q].append(a[i])
    else:
        q=a[i]
        qidx = i
        if q not in z.keys():
            z[q]=[]
        z[q].append(a[i])

将输出

{0: [1, 1.6, 1.8, 1.9, 2.0, 2], 2.1: [2.1, 3.1], 3.4: [3.4, 4], 5: [5, 5, 5.4], 8.9: [8.9], 10: [10], 23: [23]}