python中使用list的删除需要注意

casenum =int(input())

for k in range(casenum):

    ans=0

    n=int(input())

    buildings=list(map(int,input().split()))


    while buildings:#如果开始的列表中还有值

        Magicalset = set()

        for i in range(n):

            t=buildings[i]
            print(buildings)
            print(n)
            print(i)

            if not (t in Magicalset):

                Magicalset.add(t)#如果这个元素没有在集合中,就加入集合,并且在初始列表中删除
                # buildings(t)#remove函数只会删除列表的第一个指定的值,如果需要删除多个,需要使用循环,这里不能用remove
                del buildings[i]
                i-=1#删除了一个元素,i回退一个
                n-=1

        ans+=1

    print(ans)

remove函数删除元素只会删除列表的第一个与指定值相同的值,要删除当前的值可使用下标索引的方法

这里在列表中删除了一个值,列表中的个数会减少,但是for循环的i和n都还是会按照之前的值来,就会报错index out of range

修改为while循环:

casenum =int(input())

for k in range(casenum):

    ans=0

    n=int(input())

    buildings=list(map(int,input().split()))


    while buildings:#如果开始的列表中还有值

        Magicalset = set()

        i=0
        while i 
  

这也说明了for循环不能取代while循环

你可能感兴趣的:(python)