题解 | #分发糖果#

思路： 我的题解思路，先从小到大排序，然后累加得到sum，累加到target临界点的时候，进行判断，分为三种情况。

1：累加和正好等于target，此时直接退出
2: 累加和比target大1，此时 -2 -2 + 3（需保证有足够的2（至少2个）来减）
3：累加和比target大2 ， 此时直接-2 （需保证至少有一个2）

以上都要记录下使用的2，3，的个数
如果可以满足sum==target，说明当前第一个人可以分配出target

然后使用剩下的2，3，重复上述操作，得到第二个人能否分配出target，如果可以，那么说明有两个人都能分配出target，剩下一个人必然满足。否则无法进行均分

 
// 注意类名必须为 Main, 不要有任何 package xxx 信息
import java.util.Arrays;
import java.util.Arrays;
import java.util.Scanner;
 
public class Main {
    static int left_2 = 0;
    static int left_3 = 0;
 
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int groups = scanner.nextInt();
        while (groups-- >= 1) {
            int N = scanner.nextInt();
            int[] int_arr = new int[N];
            for (int i = 0; i < int_arr.length; i++) {
                int_arr[i] = scanner.nextInt();
            }
            int sum = Arrays.stream(int_arr).sum();
            if (sum % 3 != 0) {
                System.out.println("NO");
            }
            else {
                int target = sum / 3;
                boolean A_result = canndy(int_arr,target);
                int[] int_arr_left = new int[left_2 + left_3];
                for (int i = 0; i < left_2; i++) {
                    int_arr_left[i] = 2;
                }
                for (int i = left_2; i < left_3; i++) {
                    int_arr_left[i] = 3;
                }
                boolean B_result = canndy(int_arr_left,target);
                if(A_result && B_result){
                    System.out.println("YES");
                }else {
                    System.out.println("NO");
                }
            }
 
        }
    }
 
    public static boolean  canndy(int[] int_arr, int target) {
        Arrays.sort(int_arr);
         int number_2 = 0;
         int number_3 = 0;
 
         int A_number_2 = 0;
         int A_number_3 = 0;
        for (int i = 0; i < int_arr.length; i++) {
            if (int_arr[i] == 2) {
                number_2++;
            } else {
                number_3++;
            }
        }
 
        int sum = 0;
        for (int i = 0; i < int_arr.length; i++) {
            if (int_arr[i] == 2) {
                A_number_2++;
            }
            if (int_arr[i] == 3) {
                A_number_3++;
            }
            sum = sum + int_arr[i];
            if (sum == target) {
                left_2 = number_2 - A_number_2;
                left_3 = number_3 - A_number_3;
                return true;
            }
            if (sum > target) {
                if (sum - target == 1) {
                    if (A_number_2 >= 2 && number_3 - A_number_3 >= 1) {
                        sum = sum - 2 - 2 + 3;
                        A_number_2 = A_number_2 - 2;
                        A_number_3 = A_number_3 + 1;
                        left_2 = number_2 - A_number_2;
                        left_3 = number_3 - A_number_3;
                        return true;
                    }
                    else {
                        left_2 = number_2 - A_number_2;
                        left_3 = number_3 - A_number_3;
                        return false;
                    }
                }
                if (sum - target == 2 ) {
                    if(A_number_2>=1){
                        A_number_2 = A_number_2 - 1;
                        sum = sum - 2;
                        left_2 = number_2 - A_number_2;
                        left_3 = number_3 - A_number_3;
                        return true;
                    }
                    else {
                        left_2 = number_2 - A_number_2;
                        left_3 = number_3 - A_number_3;
                        return false;
                    }
                }
            }
        }
        return true;
    }
}