編寫一個函數，傳入一個int型數組，返回該數組可否分紅兩組

題目描述

編寫一個函數，傳入一個int型數組，返回該數組可否分紅兩組，使得兩組中各元素加起來的和相等，
而且，全部5的倍數必須在其中一個組中，全部3的倍數在另外一個組中（不包括5的倍數），能知足以
上條件，返回true；不知足時返回false。

輸入描述

輸入輸入的數據個數

輸出描述

返回true或者false

輸入例子

4
1
5
-5
1

輸出例子

true

算法實現

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

/**
 * Declaration: All Rights Reserved !!!
 */
public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
//        Scanner scanner = new Scanner(Main.class.getClassLoader().getResourceAsStream("data.txt"));
        while (scanner.hasNext()) {

            // 能夠被5整除的數的和
            int five = 0;
            // 能夠被3整除的數的和，不包括同時能夠被5整除的數
            int three = 0;
            List<Integer> list = new ArrayList<>();

            int num = scanner.nextInt();
            for (int i = 0; i < num; i++) {
                int v = scanner.nextInt();

                if (v % 5 == 0) {
                    five += v;
                } else if (v % 3 == 0) {
                    three += 3;
                } else {
                    list.add(v);
                }
            }

            System.out.println(divide(five, three, list));
        }

        scanner.close();
    }

    /**
     * 找出一種方法將list中的元素分紅兩個部分，分別與m，n相加，得x，y。使得x=y，若是找到這種方法返回true，不然返回false
     *
     * @param m    初始值
     * @param n    初始值
     * @param list 集合
     * @return true能夠找對應方法，false無對應方法
     */
    private static boolean divide(int m, int n, List<Integer> list) {

        int other = 0;
        for (int i : list) {
            other += i;
        }

        // 和爲偶數纔可能劃分
        if ((m + n + other) % 2 == 0) {
            // 兩個數之間的差
            int diff = Math.abs(m - n);

            // 差與餘下數的和相等，則能夠劃分
            if (diff == other) {
                return true;
            }
            // 差比餘下數的和在，則不能夠劃分
            else if (diff > other) {
                return false;
            } else {
                int sum = (m - n + other) / 2;

                return canFind(list, sum);

            }
        }

        return false;
    }

    /**
     * 在list中可否找到一組數據使用得其和是sum
     *
     * @param list 整數集合
     * @param sum  和
     * @return true：能夠找到，false：不能夠找到
     */
    private static boolean canFind(List<Integer> list, int sum) {

//        System.out.println(sum + " " + list);

        int count = (int) Math.pow(2, list.size());

        int[] mark = new int[list.size()];

        for (int i = 0; i < count; i++) {
            int val = 0;
            for (int j = 0; j < mark.length; j++) {
                if (mark[j] == 1) {
                    val += list.get(j);
                }
            }

            if (val == sum) {
                return true;
            }

            inc(mark);
        }
        return false;
    }

    /**
     * 二進制加一操做，二進制數用數組表示，從左到右是低位到高位
     *
     * @param mark 等待加一的二進制數組
     */
    private static void inc(int[] mark) {
        int carry;
        int idx = 0;
        do {
            mark[idx]++;
            carry = mark[idx] / 2;
            mark[idx] %= 2;
            idx++;
        } while (carry > 0 && idx < mark.length);
    }
}