题干如下:
小S拿到了一个长度为 的环形数组,并定义了两个下标 和 的贡献值公式为: f(i,j)=(a_i + a_jx dist(i,i)其中 dist(i,j)是下标 和 在数组中的最短距离。小S希望找到一对下标,使得它们的贡献值尽可能大。环形数组的特点是最左和最右的元素也是相邻的。你需要帮助她找到最大贡献值。 例如,给定数组[1,2,31,由于是环形数组,任意两个下标的距离都是1,因此 。
解题思路:
-
初始化最大贡献值变量:
let maxContribution = -Infinity;这行代码定义了一个变量maxContribution,并将其初始化为负无穷大。这么做的原因是,在开始遍历计算所有下标对的贡献值之前,我们不知道最大贡献值是多少,将其初始化为一个极小的值(在 JavaScript 中用-Infinity表示负无穷大),可以确保后续计算出的任何贡献值都能与其比较并在必要时更新它,使得maxContribution始终记录当前找到的最大贡献值。
-
双层
for循环遍历下标对:- 外层
for循环for (let i = 0; i < n; i++)用于控制第一个下标i的取值,它从0开始,每次递增1,直到i的值达到数组长度n - 1。这个循环会遍历环形数组中所有可能作为第一个下标的位置。 - 内层
for循环for (let j = 0; j < n; j++)用于控制第二个下标j的取值,同样从0开始,每次递增1,直到j的值达到数组长度n - 1。对于外层循环每确定的一个i值,内层循环都会完整地遍历一遍所有可能的j值,这样就实现了对环形数组中所有下标对(i, j)的遍历。
- 外层
-
计算下标对的最短距离
dist:let dist = Math.min(Math.abs(i - j), n - Math.abs(i - j));这行代码用于计算下标i和j在环形数组中的最短距离。Math.abs(i - j)计算的是按照正常顺序(非环形考虑)下标i和j的距离的绝对值,比如i = 1,j = 3,那么Math.abs(1 - 3) = 2。- 然而,在环形数组中,还存在另一种情况,就是从数组末尾绕到开头的距离更短。例如,对于长度为
5的环形数组,当i = 0,j = 3时,正常顺序距离是3,但从环形角度考虑,跨越数组边界的距离是5 - 3 = 2,这种情况下n - Math.abs(i - j)就表示跨越边界的距离(n是数组长度)。通过Math.min函数取这两种距离(正常顺序距离和跨越边界距离)中的较小值,就能得到环形数组中两个下标之间的最短距离dist。
-
计算当前下标对的贡献值
contribution:let contribution = (a[i] + a[j]) * dist;这行代码根据给定的贡献值计算公式f(i, j) = (a[i] + a[j]) * dist(i, j)来计算当前下标对(i, j)的贡献值。其中a[i]和a[j]分别是环形数组中对应下标位置的元素值,通过索引数组a来获取,然后将这两个元素值相加后再乘以前面计算得到的最短距离dist,就得到了当前下标对的贡献值contribution。
-
更新最大贡献值
maxContribution:maxContribution = Math.max(maxContribution, contribution);这行代码将当前计算得到的贡献值contribution和已经记录的最大贡献值maxContribution进行比较。Math.max函数会返回两个值中的较大值,所以如果contribution比maxContribution更大,就会将maxContribution更新为contribution的值,从而保证maxContribution始终记录着到目前为止遍历过程中找到的最大贡献值。
-
返回最大贡献值:
- 在双层
for循环结束后,maxContribution变量中存储的就是整个环形数组所有下标对中最大的贡献值了,最后通过return maxContribution;将这个最大贡献值返回。
- 在双层
function solution(n, a) {
let maxContribution = -Infinity;
for (let i = 0; i < n; i++) {
for (let j = 0; j < n; j++) {
let dist = Math.min(Math.abs(i - j), n - Math.abs(i - j));
let contribution = (a[i] + a[j]) * dist;
maxContribution = Math.max(maxContribution, contribution);
}
}
return maxContribution;
}
function main() {
console.log(solution(3, [1, 2, 3]) === 5);
console.log(solution(4, [4, 1, 2, 3]) === 12);
console.log(solution(5, [1, 5, 3, 7, 2]) === 24);
}
main();