Create 1013.将数组分成和相等的三个部分.js

Author: yinxin630

1013.将数组分成和相等的三个部分.js

@@ -0,0 +1,22 @@
+/**
+ * @param {number[]} A
+ * @return {boolean}
+ */
+var canThreePartsEqualSum = function(A) {
+    const sum = A.reduce((r, n) => r += n, 0);
+    if (sum % 3 !== 0) {
+        return false;
+    }
+
+    const avg = sum / 3;
+    let cur = 0;
+    let count = 0;
+    for (const a of A) {
+        cur += a;
+        if (cur === avg) {
+            cur = 0;
+            count++;
+        }
+    }
+    return count >= 3;
+};