vivo手机产量笔试题

题目大概描述：vivo生产智能手机，假设工人第一天产量为1台，接下来两天（第二、三天）每天生产2台，接下来三天（第四、五、六天）每天生产3台，试编程计算第n天后手机产量一共是多少？

题目分析：工人第一天生产1台手机，第2、3天都生产2台，容易看出这里有一个广义上的等差数列。

产量：1（1）、2、2（2）、3、3、3（3）、4、4、4、4（4）、5...........

天数：1、2、3、4、5、6、7、8、9...............

可以先推出第n天当天的产量i，就可以算出产量增加到 i 之前的总产量。再加上产量为i的这几天的产量，便可以算出所有的产量。

class Solution:
    def solution(self,n):
        #write code here

在函数solution下的算法（n表示第n天）：

flag, result = 0, 0
        for i in range(1, n+1):
            flag += i
            if n <= flag:
                num = i #获得当天手机产量
                a = flag - n
                b = i - a #从第b天开始，每天的产量变为i
                break
        for j in range(1, i):
            result += j**2  #每天的产量变为i之前的所有产量
        result = result + b*i #加上每天产量加1之后的天数，工人所做的量

        return result

输入11天

S = Solution()
print(S.solution(11))

结果显示：

这是本人的一次笔试经历，仅当分享。欢迎交流