字节青训营 数组元素之和最小化

问题描述

小C希望构造一个包含n个元素的数组，且满足以下条件：

1. 数组中的所有元素两两不同。
2. 数组所有元素的最大公约数为 k。
3. 数组元素之和尽可能小。

任务是输出该数组元素之和的最小值。

---

测试样例

样例1：

输入：n = 3 ,k = 1
输出：6

样例2：

输入：n = 2 ,k = 2
输出：6

样例3：

输入：n = 4 ,k = 3
输出：30

import java.util.ArrayList;
import java.util.List;

public class Main {
    public static int solution(int n, int k) {
        List<Integer> list=new ArrayList<>();
        for(int i=1;i<=n;i++){
            list.add(k*i);
        }
        return list.stream().mapToInt(Integer::intValue).sum();
    }

    public static void main(String[] args) {
        System.out.println(solution(3, 1) == 6);
        System.out.println(solution(2, 2) == 6);
        System.out.println(solution(4, 3) == 30);
    }
}