三等分

持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第10天，点击查看活动详情

927. 三等分 - 力扣（LeetCode）

给定一个由 0 和 1 组成的数组 arr ，将数组分成  3 个非空的部分 ，使得所有这些部分表示相同的二进制值。

如果可以做到，请返回任何 [i, j]，其中 i+1 < j，这样一来：

• arr[0], arr[1], ..., arr[i] 为第一部分；
• arr[i + 1], arr[i + 2], ..., arr[j - 1] 为第二部分；
• arr[j], arr[j + 1], ..., arr[arr.length - 1] 为第三部分。
• 这三个部分所表示的二进制值相等。

如果无法做到，就返回 [-1, -1]。

注意，在考虑每个部分所表示的二进制时，应当将其看作一个整体。例如，[1,1,0] 表示十进制中的 6，而不会是 3。此外，前导零也是被允许的，所以 [0,1,1] 和 [1,1] 表示相同的值。

示例 1：

输入： arr = [1,0,1,0,1]
输出： [0,3]

示例 2：

输入： arr = [1,1,0,1,1]
输出： [-1,-1]

示例 3:

输入： arr = [1,1,0,0,1]
输出： [0,2]

提示：

• 3 <= arr.length <= 3 * 10^4
• arr[i] 是 0 或 1

思路

根据题目要求可知，如果字符串可以三等分，则等分后每个串中1的个数相等，位置相等。

1. 首先我们遍历arr，保存1的下标和个数，如果1的个数不能被3整除，则arr不能被等分。
2. 把1分为三份，遍历这三份中1的下标是否相对相等，如果不相等，不能被等分。
3. 比较每份中后缀0格式是否可以相等，如果不相等不能被等分。如果相等，按要求返回下标位置。

解题

/**
 * @param {number[]} arr
 * @return {number[]}
 */
var threeEqualParts = function (arr) {
  const ones = [];
  for (let i = 0; i < arr.length; i++) {
    if (arr[i] === 1) {
      ones.push(i);
    }
  }
  if (ones.length % 3 !== 0) return [-1, -1];
  const count = ones.length / 3;
  const j = count << 1;
  for (let i = 1; i < count; i++) {
    if (
      ones[i] - ones[i - 1] !== ones[count + i] - ones[count + i - 1] ||
      ones[i] - ones[i - 1] !== ones[j + i] - ones[j + i - 1]
    ) {
      return [-1, -1];
    }
  }
  const partLen = arr.length - 1 - ones[j];
  const idx1 = ones[0] + partLen;
  const idx2 = ones[count] + partLen;
  if (idx1 >= ones[count] || idx2 >= ones[j]) {
    return [-1, -1];
  }
  return [idx1, idx2 + 1];
};
``