class Solution {
public:
vector<int> threeEqualParts(vector<int>& arr) {
}
};
927. 三等分
给定一个由 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 * 104arr[i] 是 0 或 1原站题解
cpp 解法, 执行用时: 32 ms, 内存消耗: 37.8 MB, 提交时间: 2023-06-27 17:33:02
class Solution {
public:
vector<int> threeEqualParts(vector<int>& arr) {
int n = arr.size();
int cnt = accumulate(arr.begin(), arr.end(), 0);
if (cnt % 3) return {-1, -1};
if (!cnt) return {0, n - 1};
cnt /= 3;
auto find = [&](int x) {
int s = 0;
for (int i = 0; i < n; ++i) {
s += arr[i];
if (s == x) return i;
}
return 0;
};
int i = find(1), j = find(cnt + 1), k = find(cnt * 2 + 1);
for (; k < n && arr[i] == arr[j] && arr[j] == arr[k]; ++i, ++j, ++k) {}
return k == n ? vector<int>{i - 1, j} : vector<int>{-1, -1};
}
};
golang 解法, 执行用时: 36 ms, 内存消耗: 6.6 MB, 提交时间: 2023-06-27 17:32:39
func threeEqualParts(arr []int) []int {
find := func(x int) int {
s := 0
for i, v := range arr {
s += v
if s == x {
return i
}
}
return 0
}
n := len(arr)
cnt := 0
for _, v := range arr {
cnt += v
}
if cnt%3 != 0 {
return []int{-1, -1}
}
if cnt == 0 {
return []int{0, n - 1}
}
cnt /= 3
i, j, k := find(1), find(cnt+1), find(cnt*2+1)
for ; k < n && arr[i] == arr[j] && arr[j] == arr[k]; i, j, k = i+1, j+1, k+1 {
}
if k == n {
return []int{i - 1, j}
}
return []int{-1, -1}
}
java 解法, 执行用时: 1 ms, 内存消耗: 46.4 MB, 提交时间: 2023-06-27 17:32:18
class Solution {
private int[] arr;
public int[] threeEqualParts(int[] arr) {
this.arr = arr;
int cnt = 0;
int n = arr.length;
for (int v : arr) {
cnt += v;
}
if (cnt % 3 != 0) {
return new int[]{-1, -1};
}
if (cnt == 0) {
return new int[]{0, n - 1};
}
cnt /= 3;
int i = find(1), j = find(cnt + 1), k = find(cnt * 2 + 1);
for (; k < n && arr[i] == arr[j] && arr[j] == arr[k]; ++i, ++j, ++k) {
}
return k == n ? new int[]{i - 1, j} : new int[]{-1, -1};
}
private int find(int x) {
int s = 0;
for (int i = 0; i < arr.length; ++i) {
s += arr[i];
if (s == x) {
return i;
}
}
return 0;
}
}
python3 解法, 执行用时: 84 ms, 内存消耗: 17.3 MB, 提交时间: 2023-06-27 17:32:01
class Solution:
def threeEqualParts(self, arr: List[int]) -> List[int]:
# 每一区间内第一个1的位置
def find(x):
s = 0
for i, v in enumerate(arr):
s += v
if s == x:
return i
n = len(arr)
# 1 的个数必须是3的倍数
cnt, mod = divmod(sum(arr), 3)
if mod:
return [-1, -1]
if cnt == 0:
return [0, n - 1]
i, j, k = find(1), find(cnt + 1), find(cnt * 2 + 1)
while k < n and arr[i] == arr[j] == arr[k]:
i, j, k = i + 1, j + 1, k + 1
return [i - 1, j] if k == n else [-1, -1]