1. 程式人生 > >CCF 201703-1 分蛋糕 Python程式碼

CCF 201703-1 分蛋糕 Python程式碼

n , k = input().split()
n = int(n)
k = int(k)
a = list(map(int,input().split()))
count = 0 #人數
total = 0

for i in range(len(a)):
    total = total + a[i]
    # 如果重量不大於k 並且已經是最後一塊蛋糕了,還是算分到一個人

    if (total < k and i == n - 1):
        count = count + 1

    if (total >= k):
        count = count + 1
        total = 0
        
    # 不滿足重量條件不斷返回原來疊加
    if (total < k):
        continue

print(count)

注意 if (total < k and i == n - 1):條件必須寫在最前面,否則某些情況會少丟掉一個人