ตรวจสอบอาร์เรย์ที่แบ่งเป็น 3 ส่วนมีผลรวมเท่ากันใน JavaScript

ปัญหา

เราต้องเขียนฟังก์ชัน JavaScript ที่รับอาร์เรย์ของจำนวนเต็มเป็นอาร์กิวเมนต์เดียว ฟังก์ชันควรคืนค่า true ถ้าและก็ต่อเมื่อเราสามารถแบ่งอาร์เรย์ออกเป็น 3 ส่วนที่ไม่ว่างเปล่าและมีผลรวมเท่ากัน มิฉะนั้นคืนค่า false

ตัวอย่างเช่น หากอินพุตคือ

const arr = [3, 3, 6, 5, -2, 2, 5, 1, -9, 4];

ผลลัพธ์ควรเป็น

const output = true;

คำอธิบายผลลัพธ์

เพราะ

3 + 3 = 6 = 5 - 2 + 2 + 5 + 1 - 9 + 4

ตัวอย่างโค้ด

const arr = [3, 3, 6, 5, -2, 2, 5, 1, -9, 4]; 
const thirdSum = (arr = []) => {
  const sum = arr.reduce((acc, val) => acc + val, 0);
  if(!Number.isInteger(sum / 3)){
    return false;
  }
  let count = 0;
  let curr = 0;
  const target = sum / 3;
  for(const num of arr){
    curr += num;
    if(curr === target){
      curr = 0;
      count += 1;
    }
  }
  return count === 3 && curr === 0;
};
console.log(thirdSum(arr));

ผลลัพธ์

ผลลัพธ์ในคอนโซลจะเป็น

true