[HAI] open lesson maxSum

sliding window fixed size
main
Vasily Guzov 1 year ago
parent 052a872f46
commit 427731f797

@ -0,0 +1,3 @@
import { maxSum } from "./maxSum";
export default maxSum;

@ -0,0 +1,16 @@
import { describe, it, expect } from "vitest";
import { maxSum } from "./maxSum.ts";
describe("maxSum", () => {
it("[3, 2, 5, 9, 4 ,1] k=3 answer 18", () => {
const NUMS = [3, 2, 5, 9, 4, 1];
expect(maxSum(NUMS, 3)).toBe(18);
});
it("[0, 1, 2, 8, 1 ,11, 10, 0, 0 ,5] k=4 answer 18", () => {
const NUMS = [0, 1, 2, 8, 1, 11, 10, 0, 0, 5];
expect(maxSum(NUMS, 4)).toBe(30);
});
});

@ -0,0 +1,20 @@
export function maxSum(nums: number[], k: number) {
const L = nums.length;
if (k > L) return 0;
/* считаем первое окно */
let sum = 0;
for (let i = 0; i < k; i++) {
sum += nums[i];
}
/* смещем окно, переситываем результат, выбираем максимальный */
let maxSumRes = sum;
for (let i = k; i < L; i++) {
sum = sum - nums[i - k] + nums[i];
maxSumRes = Math.max(maxSumRes, sum);
}
return maxSumRes;
};
Loading…
Cancel
Save