JAVA程序设计：三等分（LeetCode：927）

给定一个由 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的个数必须是3的整数倍 将1等分成3份 再根据最后一份末尾0的个数推导 前两份的最后一个坐标，最后判断3份去掉前导0 是不是一样。

class Solution {
    public int[] threeEqualParts(int[] A) {
    	
    	int n=A.length;
    	int[] error=new int[] {-1,-1};
    	
    	int sum=0;
    	for(int i=0;i<n;i++)
    		sum+=A[i];
    	
    	if(sum%3!=0) return error;
    	
    	int t=sum/3;
    	
    	if(t==0)
    		return new int[] {0,n-1};
    	
    	int l=-1,r=-1,ll=-1,rr=-1,lll=-1,rrr=-1;
    	int num=0;
    	for(int i=0;i<n;i++) {
    		if(A[i]==0) continue;
    		num+=1;
    		if(num==1) l=i;
    		if(num==t) r=i;
    		if(num==t+1) ll=i;
    		if(num==2*t) rr=i;
    		if(num==2*t+1) lll=i;
    		if(num==3*t) rrr=i;
    	}
    	
    	int[] arr1=Arrays.copyOfRange(A, l, r+1);
    	int[] arr2=Arrays.copyOfRange(A, ll, rr+1);
    	int[] arr3=Arrays.copyOfRange(A, lll, rrr+1);
    	
    	if(!Arrays.equals(arr1, arr2)) return error;
    	if(!Arrays.equals(arr1, arr3)) return error;

    	int x=ll-r-1;
    	int y=lll-rr-1;
    	int z=n-rrr-1;
    	
    	if(x<z || y<z) return error;
    	
    	return new int[] {r+z,rr+z+1};
    }
}