【华为OJ】【104-201301JAVA题目0-1级】

【华为OJ】【算法总篇章】

【华为OJ】【104-201301JAVA题目0-1级】

【工程下载】

题目描述

编写一个函数，传入一个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;

/** * Author: 王俊超 * Time: 2016-05-05 16:31 * CSDN: http://blog.csdn.net/derrantcm * Github: https://github.com/Wang-Jun-Chao * 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);
    }
}