声明:只求用最简单的方法通过,不求优化,不喜勿喷
- 首先思考一个问题,如果要分成三分,那么每份中1的数量必须是一样的,因为有前导0的存在,所以0可以不一样
- 按照找个说法可以把1的数量不是整除3的去掉,其中如果1的数量是0的,直接返回[0,2],就是拆成 0,0,000000…这样的情况
- 之后来看1和0的关系,首先,每组数据的后置0肯定是固定的,最后一组数据有几个后置0,那么其他每组就有几个后置0,所以遍历的时候要记住最后一组后置0的数量,第1,2组数据中,如果出现后置0不足的情况,那么数据要排除,返回[-1.-1],如果0数量大于最后一组的后置0,那么需要拆除一部分作为下一组数据的前导0
- 想清楚了就开始遍历,从后往前:
- 首先记录到出现等一个1开始前面有几个0,用z记录,作为后置0的数量,
- 之后记录从出现等一个1开始到1的数量满足1总数/3之间出现的内容,
- 1的数量满足1总数/3,结束,开始下一组数据的遍历,先记录当前位置作为分割点。
- 记录第二组数据遍历时,出现等一个1出现了几次0,0数量小于z,返回[-1,-1];d大于z,则当前0数量-z是第三组数据的前导0,那么分割点需要减这个量
- 第二组1的数量满足1总数/3,结束遍历,记录当前位置-1位置第一个分割点(题目要求),同时比较从出现第一个1开始到遍历结束,记下来的字符串和第三组数据的字符串是不是一致,不一致就返回[-1,-1]
- 第一组数据的遍历如法炮制
- 优化todo
/**
* @param {number[]} arr
* @return {number[]}
*/
var threeEqualParts = function (arr) {
let cnt = 0;
for (var i = 0; i < arr.length; i++) {
if (arr[i] == 1) {
cnt++;
}
}
if (cnt == 0) {
return [0, 2];
}
if (cnt % 3 != 0) {
return [-1, -1];
}
let s = cnt / 3;
let now = 0;
let z = 0;
let re = [];
let j;
let bll = ""
for (var i = arr.length - 1; i >= 0; i--) {
if (now == 0 && arr[i] == 0) {
z++;
}
if (arr[i] == 1) {
now++;
}
if (now != 0) {
bll = bll + arr[i]
}
if (now == s) {
re[1] = i;
break;
}
}
j = i;
now = 0;
let mz = 0;
let bll2 = ""
for (var i = j - 1; i >= 0; i--) {
if (now == 0 && arr[i] == 0) {
mz++;
}
if (arr[i] == 1) {
if (now == 0) {
if (mz < z) {
return [-1, -1]
} else {
re[1] -= (mz - z);
}
}
now++;
}
if (now != 0) {
bll2 = bll2 + arr[i]
}
if (now == s) {
re[0] = i - 1;
break;
}
}
if (bll2 != bll) {
return [-1, -1]
}
j = i;
now = 0;
mz = 0;
bll2 = "";
for (var i = j - 1; i >= 0; i--) {
if (now == 0 && arr[i] == 0) {
mz++;
}
if (arr[i] == 1) {
if (now == 0) {
if (mz < z) {
return [-1, -1]
} else {
re[0] -= (mz - z);
}
}
now++;
}
if (now != 0) {
bll2 = bll2 + arr[i]
}
if (now == s) {
break;
}
}
if (bll2 != bll) {
return [-1, -1]
}
return re;
};