大厂真题：【模拟】小红书2023秋招提前批-小红的数组构造

题目描述与示例

题目描述

小红希望你构造一个数组，满足以下条件：

1. 数组共有n个元素，且所有元素两两不相等。

2. 所有元素的最大公约数等于k。

3. 所有元素之和尽可能小。

请你输出数组元素之和的最小值。

输入描述

两个正整数n和k

1 ≤ n, k ≤ 10^5
1

输出描述

一个正整数，代表数组元素之和的最小值。

示例一

输入

3 1
1

输出

6
1

说明

构造数组[1, 2, 3]为满足题意的数组

示例二

输入

2 6
1

输出

18
1

说明

构造数列[6, 12]为满足题意的数列。

时空限制

时间限制： 3S

内存限制： 512MB

解题思路

本题较为简单，构造[k, 2k, 3k, ..., nk]这样的数组即为符合要求的数组（这个结论很容易用反证法证明），使用等差数列求和公式可知该数组的和为

k + 2k + 3k + ... + nk = (1 + 2 + 3 + ... + n) * k = (1 + n) * n // 2 * k
1

故对于输入的两个参数n和k，只需要输出(1 + n) * n * k // 2即为答案。

注意，使用C++和Java的同学要用long而不是int，防止计算结果过大而爆内存。

关于等差数列求和公式和最大公约数相关内容，可详见文档常用数学概念、公式、方法汇总

代码

Python

# 题目：【数学】小红书2023秋招提前批-小红的数组构造
# 作者：闭着眼睛学数理化
# 算法：数学
# 代码有看不懂的地方请直接在群上提问

n, k = map(int, input().split())
print((1 + n) * n * k // 2)
1
2
3
4
5
6
7

Java

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        long n = scanner.nextLong();
        long k = scanner.nextLong();
        System.out.println((1 + n) * n * k / 2);
        scanner.close();
    }
}
1
2
3
4
5
6
7
8
9
10
11

C++

#include 

using namespace std;

int main() {
    long n, k;
    cin >> n >> k;
    cout << (1 + n) * n * k / 2 << endl;
    return 0;
}
1
2
3
4
5
6
7
8
9
10

时空复杂度

时间复杂度：O(1)。

空间复杂度：O(1)。

---

华为OD算法/大厂面试高频题算法练习冲刺训练

• 华为OD算法/大厂面试高频题算法冲刺训练目前开始常态化报名！目前已服务100+同学成功上岸！

• 课程讲师为全网50w+粉丝编程博主@吴师兄学算法 以及小红书头部编程博主@闭着眼睛学数理化

• 每期人数维持在20人内，保证能够最大限度地满足到每一个同学的需求，达到和1v1同样的学习效果！

• 60+天陪伴式学习，40+直播课时，300+动画图解视频，300+LeetCode经典题，200+华为OD真题/大厂真题，还有简历修改、模拟面试、专属HR对接将为你解锁

• 可上全网独家的欧弟OJ系统练习华子OD、大厂真题

• 可查看链接 大厂真题汇总 & OD真题汇总(持续更新)

• 绿色聊天软件戳 od1336了解更多