数组元素之和最小化 | 豆包MarsCode AI刷题

#include <iostream>
#include <string>
#include <vector>


using namespace std;

int solution(int n, int k) {
  int sum = 0;
  for (int i = 1; i <= n; i++)
    sum += k * i;
  return sum;
}

int main() {
  std::cout << (solution(3, 1) == 6) << std::endl;
  std::cout << (solution(2, 2) == 6) << std::endl;
  std::cout << (solution(4, 3) == 30) << std::endl;
}