算法题 14 数组元素之和最小化(c++)

45 阅读1分钟

问题描述

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

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

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


测试样例

样例1:

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

样例2:

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

样例3:

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

代码

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

using namespace std;

int solution(int n, int k) {
    // PLEASE DO NOT MODIFY THE FUNCTION SIGNATURE
    // write code here
    int ans = 0;
    int no = k;
    for(int i = 1 ; i <= n ; i ++)
    {
        ans = ans + no;
        no = k * (i + 1) ;
    }
 //   cout<<ans<<endl;
    return ans;
}

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;
}

分析

要保证最大公约数为K,那么应该保证最小数为K且其他数应为K的倍数。

故以此构造的数组应为 K , 2K , 3K , 4K , 5K ....... nK;