文章目錄

• • 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阿明），一起加油、一起學習進步！

總結

以上是生活随笔為你收集整理的天池 在线编程 三等分（模拟）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。