package shuzu;

import java.util.HashMap;

/*
1013. 將陣列分成和相等的三個部分
給你一個整數陣列 A，只有可以將其劃分為三個和相等的非空部分時才返回 true，否則返回 false。

形式上，如果可以找出索引 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：

輸入：[0,2,1,-6,6,-7,9,1,2,0,1]
輸出：true
解釋：0 + 2 + 1 = -6 + 6 - 7 + 9 + 1 = 2 + 0 + 1
示例 2：

輸入：[0,2,1,-6,6,7,9,-1,2,0,1]
輸出：false
示例 3：

輸入：[3,3,6,5,-2,2,5,1,-9,4]
輸出：true
解釋：3 + 3 = 6 = 5 - 2 + 2 + 5 + 1 - 9 + 4


提示：

3 <= A.length <= 50000
-10^4 <= A[i] <= 10^4
          解題思路：
                 首先得到整個陣列的和
                 然後遍歷和/3
                 是不是可以用hashmap儲存結果然後以空間換時間呢？//字首和------未完全通過  52/55

                 思路2：
                    思路1的基礎上雙指標
                    左右齊頭並進，為sum/3後，中間的一定為sum/3
                    若兩指標碰見，則False
                    當 left + 1 < right 的約束下，可以找到陣列兩頭的和都是 sum/3,那麼中間剩下的元素和就一定也是sum/3
                      （left + 1 < right的約束就是要中間有剩下的元素，使用left < right的約束，陣列可能可以恰好被劃分成兩部分，中間沒有元素）


 */
public class p1013 {
    //未完全通過，52/55
    /*public static boolean canThreePartsEqualSum(int[] A) {
        int sum_i=0;
        HashMap<Integer,Integer>map=new HashMap<>();//key為字首和，i為index
        map.put(0,-1);
        for(int i=0;i<A.length;i++){
            sum_i+=A[i];
            map.put(sum_i,i);
        }
        System.out.println("和"+sum_i);
        if(sum_i%3!=0||!map.containsKey(2*sum_i/3)||!map.containsKey(sum_i/3))return false;
        if(map.containsKey(2*sum_i/3)&&map.containsKey(sum_i/3)){
            if(map.get(sum_i/3)>map.get(2*sum_i/3)){
                return false;
            }
        }
        return true;
    }*/

    //雙指標
    public static boolean canThreePartsEqualSum(int[] A){
        int left=0,right=A.length-1,sum=0;
        for(int i=0;i<A.length;i++)sum+=A[i];
        if(sum%3!=0)return false;
        int sumleft=A[left],sumright=A[right];
        while (left<right-1){
            if(sumleft==sum/3&&sumright==sum/3)return true;
            if(sumleft!=sum/3){
                left++;
                sumleft+=A[left];
            }
            if(sumright!=sum/3){
                right--;
                sumright+=A[right];
            }
        }
        return false;

    }

    public static void main(String[] args) {
        //1,-1,1,-1
        int a[]={};
        System.out.println(canThreePartsEqualSum(a));
    }

}

　　執行結果：