Java:VIVO笔试题目:生产手机量

原题目:一个员工第一天生产一个手机,随着时间的增长他越来越熟练,第二天和第三天可以生产2台,后面的3天每天可以生产3台,再后面4天每天可以生产4台,请写出这个员工第n天总共可以生产多少个手机
附:n = 11 时,生产35台;
思路:利用for循环 for(int i = 0; i < n ; i++) sum += k;
现在就是要操作这个k,使得随着一天天过去,k要等于:1、2、2、3、3、3、4、4、4、4、5、5、5、5、5……
利用一个栈的存储和计数功能,先把第一天的日产量放入栈中(k = 1),然后判断栈内元素的总数是否大于等于k值,如果成立,则k加1,且清空栈内元素。
这样子的话,当k是2的时候,会把第二天的日产量放入数组,然后k不会加一,再把第三天的日产量放入数组,这个时候数组内有两个元素>=k,k会加一,且数组清空。这个逻辑一直往后,日产量是多少,就会有多少天k的值不变。

class Solution {
     
    /**
     *
     * @param n int整型 第n天
     * @return int整型
     */
    public int so(int n){
     
        int pn=0;
        Stack<Integer> stack=new Stack<>();
        int k=1;
        for(int i=0;i<n;i++){
     
            pn+=k;
            stack.push(k);
            if(stack.size()>=k){
     
                k++;
                stack.clear();
            }
        }
        return pn;
    }
}

你可能感兴趣的:(java,数据结构)