给定一个由 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的个数。
如果个数为0,直接返回[0,2];
如果个数不为0,判断个数能否被3整除,如果不能整除的话返回[-1, -1]。
2.当个数能被3整除时,算出个数/3,即为每部分1的个数。
能确定的是,第三部分的值,即第三部分第一个1后的数字是确定的。
例如:
0 1 1 0 0 0 1 1 0 0 0 0 0 1 1 0 0
按着1的个数分成3份,根据第三部分确定下来值为1100。
0 - 1 1 0 0 0 - 1 1 0 0 0 0 0 - 1 1 0 0
3.将横杠这个向右移
0 1 - 1 0 0 0 1 - 1 0 0 0 0 0 1 - 1 0 0
0 1 1 - 0 0 0 1 1 - 0 0 0 0 0 1 1 - 0 0
0 1 1 0 - 0 0 1 1 0 - 0 0 0 0 1 1 0 - 0
0 1 1 0 0 - 0 1 1 0 0 - 0 0 0 1 1 0 0
就得出结果了。
我写的,哈哈,逗乐了:
#Python3
class Solution:
def threeEqualParts(self, arr: List[int]) -> List[int]:
onenum = arr.count(1)
if onenum == 0:
return [0, 2]
elif onenum % 3 != 0:
return [-1, -1]
someone = onenum // 3
first = second = third = 0
i = len(arr)
while i :
i -= 1
if arr[i] == 1:
third += 1
if third == someone :
third = i
break
print("third", third)
while i :
i -= 1
if arr[i] == 1:
second += 1
if second == someone :
second = i
break
print("second", second)
while i :
i -= 1
if arr[i] == 1:
first += 1
if first == someone :
first = i
break
while third < len(arr) and arr[first] == arr[second] == arr[third]:
first, second, third = first + 1, second + 1, third + 1
if third == len(arr) :
return [first - 1, second]
else :
return [-1, -1]
return [i, j]
看的这个哥们的题解,理解贼快:
作者:lcbin
链接:https://leetcode.cn/problems/three-equal-parts/solution/by-lcbin-7ir1/
Python3:
class Solution:
def threeEqualParts(self, arr: List[int]) -> List[int]:
def find(x):
s = 0
for i, v in enumerate(arr):
s += v
if s == x:
return i
n = len(arr)
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]
C++:
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};
}
};