我想編寫一個 python 程序，如果最后一個元素是 -1 打印 -1，則給定一個用戶輸入列表，然后如果列表中的任何元素包含不應該計數的 -1 和剩余的數組總和。例如：1) list=[1,2,3,4,5] ans should be 15 2) list=[1,2,3,4,-1] ans should be -13) list=[1,2,-1,4,5] ans should be 12 ignoring "-1"我嘗試了 2 種解決方案，但都沒有用。import sysdef totalcost(ar): if ar[-1]==-1: return -1 else: summ=0 for elem in ar: if(ar[elem]==-1): ar.remove(elem) summ=summ+elem else: summ=summ+elem return summif __name__=='__main__': ar_city=input() ar=list(map(int,input().strip().split())) result=totalcost(ar) print(result) import sysdef totalcost(ar): if ar[-1]==-1: return -1 else: summ=0 for elem in ar: if(ar[elem]<0): ar_new=ar.remove(elem) for i in ar_new: summ=summ+i else: summ=summ+elem return summif __name__=='__main__': ar_city=input() ar=list(map(int,input().strip().split())) result=totalcost(ar) print(result)
4 回答
尚方寶劍之說
TA貢獻1788條經驗 獲得超4個贊
這是一個可能的解決方案：
def totalcost(ar):
return -1 if ar[-1] == -1 else sum(x for x in ar if x != -1)
如果您不想要使用列表理解的單行解決方案，您可以執行以下操作：
def totalcost(ar):
if ar[-1] == -1:
return -1
s = 0
for x in ar:
if x != -1:
s += x
return s
千萬里不及你
TA貢獻1784條經驗 獲得超9個贊
您可以使用一些簡單的功能來改善您的生活：
sum(filter(lambda?x:?x?!=?-1,?ls))
sum- 在不使用 for 循環的情況下對 iterable 求和filter- 從 iterable 中過濾掉不需要的元素。在這種情況下，我使用ls簡單的lambda x: x != -1.
當然，這應該在您的初始條件之后使用，如下所示：
if?ls[-1]?==?-1: ????return?-1 ????return?sum(filter(lambda?x:?x?!=?-1,?ls))
白衣非少年
TA貢獻1155條經驗 獲得超0個贊
我認為您可以簡單地使用 filter 和 sum 以及 lambda 來實現。只是稍微更新了你的 totalcost 函數
import sys
def totalcost(ar):
? ? if ar[-1] == -1:
? ? ? ? return -1
? ? else:
? ? ? ? ar_filtered = filter(lambda x: x > 0, ar)
? ? ? ? return sum(ar_filtered)
if __name__ == '__main__':
? ? ar_city = [1, 2, 3, 4, 5]
? ? assert totalcost(ar_city) == 15
? ? ar_city = [1, 2, 3, 4, -1]
? ? assert totalcost(ar_city) == -1
? ? ar_city = [1, 2, -1, 4, 5]
? ? assert totalcost(ar_city) == 12
ITMISS
TA貢獻1871條經驗 獲得超8個贊
以下是你犯過的錯誤：
ar[elem]沒有ar正確迭代每個元素。您在 for-in 循環中返回
summ，因此它只會返回第一個元素的值。
這是一個根據您的代碼修改的工作示例。
def totalcost(ar):
if ar[-1] == -1:
return -1
summ = 0
for elem in ar:
if(elem != -1):
summ += elem
return summ
添加回答
舉報
0/150
提交
取消