[编程题]vivo智能手机产能

题目链接

在vivo产线上，每位职工随着对手机加工流程认识的熟悉和经验的增加，日产量也会不断攀升。

假设第一天量产1台，接下来2天(即第二、三天)每天量产2件，接下来3天(即第四、五、六天)每天量产3件 ... ... 

以此类推，请编程计算出第n天总共可以量产的手机数量。

 

输入例子1:

11

 

输出例子1:

35

 

例子说明1:

第11天工人总共可以量产的手机数量

 

思路：

把每天的产量放到一个数组中即可。 

规律是 1， 2， 2， 3， 3， 3，4,4,4,4...

class Solution:
    def solution(self , n ):
        # write code here
        res = [-1]
        now = 1
        while len(res) < n + 1: # 没有填到第n天的产量
            for i in range(now):
                res.append(now)
            now += 1
        sum = 0
        p = 1
        while p <= n: #n天的产量
            sum += res[p]
            p +=1
        return sum