文章目录
- 1. 题目
- 2. 解题
1. 题目
给定一个由 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] 表示相同的值。
3 <= A.size <= 30000
A[i] = 0 or A[i] = 1示例
样例 1:
输入:[1,0,1,0,1]
输出:[0,3]样例 2:
输出:[1,1,0,1,1]
输出:[-1,-1]
https://tianchi.aliyun.com/oj/245867719779445300/277517718521058001
https://leetcode-cn.com/problems/three-equal-parts/
2. 解题
- 就是检查每个部分尾部的0的个数一样,确定开始结束位置,见注释
class Solution {
public:/*** @param A: an array* @return: divide the array into 3 non-empty parts*/vector<int> threeEqualParts(vector<int> &A) {// write your code hereint sum = accumulate(A.begin(), A.end(), 0);if(sum%3 != 0)//1的个数不能整除3return {-1,-1};int i, j, part = sum/3, one = 0, tail0 = 0, n = A.size();if(part == 0)//没有1return {0,2};for(i = n-1; i >= 0; --i){if(A[i] == 1)one++;else{if(one==0)tail0++;//第三部分的尾部0个数}if(one == part)break;}int mid0 = 0, start = i;//start 为第三部分的可能开始点for(i--; i>=0 && A[i] != 1; i--){mid0++;//第二第三部分中间的0的个数}if(mid0 < tail0)//第二部分尾0的个数不能跟第三部分尾0一样多return {-1,-1};start -= mid0-tail0;//分一部分前置0给第三部分int count = 0;for(i = start-1, j = n-1; i >= 0; --i,--j){ // 比较 第二第三部分是否一样if(A[i]==1) count++;if(A[i] != A[j])return {-1,-1};if(count == part)//1的个数够了,跳出break;}mid0 = 0;int start1 = i;//第二分部可能的开始位置for(i--; i>=0 && A[i] != 1; i--){mid0++;//第一第二部分中间的0的个数}if(mid0 < tail0)//同上return {-1,-1};count = 0;start1 -= mid0-tail0;//第二部分开始的位置for(i = start1-1, j = n-1; i >= 0; --i,--j){if(A[i] == 1) count++;if(A[i] != A[j])return {-1,-1};if(count == part)break;}return {start1-1, start};//第一部分结束,第三部分开始}
};
50ms C++
我的CSDN博客地址 https://michael.blog.csdn.net/
长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!