题目描述
给定一个由 0 和 1 组成的数组 A,将数组分成 3 个非空的部分,使得所有这些部分表示相同的二进制值。
如果可以做到,请返回任何 [i, j],其中 i+1 < j,这样一来:
- A[0], A[1], ..., A[i] 组成第一部分;
- A[i+1], A[i+2], ..., A[j-1] 作为第二部分;
- A[j], A[j+1], ..., A[A.length - 1] 是第三部分。
- 这三个部分所表示的二进制值相等。
如果无法做到,就返回 [-1, -1]。
注意,在考虑每个部分所表示的二进制时,应当将其看作一个整体。例如,[1,1,0] 表示十进制中的 6,而不会是 3。此外,前导零也是被允许的,所以 [0,1,1] 和 [1,1] 表示相同的值。
示例1
输入: [1,0,1,0,1] 输出: [0,3]
示例2
输入: [1,1,0,1,1] 输出: [-1,-1]
提示
- 3 <= A.length <= 30000
- A[i] == 0 或 A[i] == 1
题解
这题虽然名义上是个难题,其实基本没有用到什么算法,只是代码实现上略微繁琐了一点。
想象如果这个数组能分成三个子数组,每个子数组表示的数字都相同,那么首先每个子数组中 1 的数量一定要相等!
所以我们先统计 1 的数量,如果它不是 3 的倍数,那么一定不存在划分方式,直接返回无解就行了。如果数量是 0 ,就说明数组全 0 ,那么随便划分都是合理的,任意返回就行了。
接下来将 1 的数量等分为 3 份 ,然后遍历数组,找出 3 个子数组的左右边界(注意这个边界表示的是每个子数组第一个 1 和最后一个 1 的位置)。这时候还没结束,因为最后一个子数组末尾会多出来很多 0 。所以我们需要在前两个子数组后面加上等量的 0 。
最后遍历一遍三个子数组,判断是否完全相等就行了。
听起来是很简单,代码实现的时候还是有几个小细节的。比如求边界的时候,可以利用求余操作,保存到两个数组里,这样写起来美观方便。
代码
c++
class Solution { public: vector<int> threeEqualParts(vector<int>& A) { int n = A.size(); int cnt1 = accumulate(A.begin(), A.end(), 0); if (cnt1 % 3) return {-1, -1}; if (!cnt1) return {0, 2}; cnt1 /= 3; int cnt = 0, l[3], r[3]; for (int i = 0; i < n; ++i) { if (!A[i]) continue; cnt++; if ((cnt-1)%cnt1 == 0) l[(cnt-1)/cnt1] = i; if (cnt%cnt1 == 0) r[(cnt-1)/cnt1] = i; } for (int i = 0; i < 3; ++i) { r[i] += n - 1 - r[2]; } for (int i = 0; i <= r[0]-l[0]; ++i) { if (A[l[1]+i] != A[l[0]+i] || A[l[2]+i] != A[l[0]+i]) return {-1, -1}; } return {r[0], r[1]+1}; } };
python
class Solution: def threeEqualParts(self, A: List[int]) -> List[int]: n = len(A) cnt1 = sum(A) if cnt1%3 != 0: return [-1, -1] if cnt1 == 0: return [0, 2] cnt1 //= 3 cnt = 0 l, r = [0]*3, [0]*3 for i in range(n): if A[i] == 0: continue cnt += 1 if (cnt-1)%cnt1 == 0: l[(cnt-1)//cnt1] = i if cnt%cnt1 == 0: r[(cnt-1)//cnt1] = i for i in range(3): r[i] += n - 1 - r[2] if A[l[1]:r[1]+1] != A[l[0]:r[0]+1] or A[l[2]:r[2]+1] != A[l[0]:r[0]+1]: return [-1, -1] return [r[0], r[1]+1]