样例

import java.util.*;
public class Main{
    public static void main(String[] args){
        Scanner scan=new Scanner(System.in);
        int k=scan.nextInt();
        int n=(int)Math.sqrt(2*k)+2;
        int[] arr=new int[n+1];
        for(int i=1;i<=n;i++){
            arr[i]=i*(i+1)/2;
        }
        long res=0;int index=0;
        for(int i=1;i<n;i++){
            if(k<=arr[i]){
                index=i;
                for(int j=1;j<index;j++){
                    res+=j*j;
                }
                break;
            }
        }
        res+=(k-arr[index-1])*index;
        System.out.println(res);
    }
}

算法1

(暴力枚举) $O(n^2)$

blablabla

时间复杂度

参考文献

C++ 代码

blablabla

算法2

(暴力枚举) $O(n^2)$

blablabla

时间复杂度

参考文献

C++ 代码

blablabla