vivo产能问题

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

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

 public int solution (int n) {
         int ii = (int) Math.floor((-1 + Math.sqrt(1 + 8 * n)) / 2);
         int res = ii * (ii + 1) * (2 * ii + 1) / 6;
         if (ii * (ii + 1) == n){
             return res;
         }else{
             return res + (ii + 1) * (n - ii * (ii + 1) / 2);
         }
     }