题解 | #数组分组#

又是一道dfs的经典题。
注意几个特殊条件
    1)sum/2不能整除;
    2)输入为5 -5,即sum = 0, target = 0;

import java.util.*;
public class Main {
    public static void main(String[] args) {
        // 将非5 非3 倍数的数字分为两组,使其差值为5和3组的差值
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int sum3 = 0, sum5 = 0, sum = 0;
        List<Integer> list = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            int num = sc.nextInt();
            sum += num;
            if (num % 5 == 0) {
                sum5 += num;
                continue;
            } else if (num % 3 == 0) {
                sum3 += num;
                continue;
            } else {
                list.add(num);
            }
        }

        if (sum % 2 != 0) {
            System.out.println("false");
            return;
        }
        int target = sum / 2 - sum5;


        if (list.size() == 0 && target == 0) {
            System.out.println("true");
            return;
        }

        if (list.size() == 1) {
            if (list.get(0) != target) {
                System.out.println("false");
            } else {
                System.out.println("true");
            }
            return;
        }

        int[] nums = new int[list.size()];
        for (int i = 0; i < nums.length; i++) {
            nums[i] = list.get(i);
        }
        // 组合,m个数据边选择x个数据(不可重复取),使其和为target
        // 再次安排dfs
        for (int tag = 1; tag <= nums.length; tag++) {
            List<List<Integer>> arrange = new ArrayList<>();
            List<Integer> tmp = new ArrayList<>();
            boolean[] visited = new boolean[nums.length];
            dfs(arrange, tmp, nums, visited, tag, 0);
            for (List<Integer> tmpList : arrange) {
                int sumList = 0;
                for (Integer integer : tmpList) {
                    sumList += integer;
                    if (sumList == target) {
                        System.out.println("true");
                        return;
                    }
                }
            }
        }
        System.out.println("false");
    }

    /**
     * 因为需要选择x个数求和,因此只需要组合即可,不需要排列
     */
    public static void dfs(List<List<Integer>> arrange, List<Integer> tmp, int[] nums,
                           boolean[] visited, int tag, int cur) {
        if (tmp.size() == tag) {
            arrange.add(new ArrayList<>(tmp));
            return;
        }
        for (int i = cur; i < nums.length; i++) {
            if (!visited[i]) {
                tmp.add(nums[i]);
                visited[i] = true;
                dfs(arrange, tmp, nums, visited, tag,i + 1);
                // 回溯
                visited[i] = false;
                tmp.remove(tmp.size() - 1);
            }
        }
    }
}


全部评论

相关推荐

那么好了好了:他本来公司就是做这个的,不就是正常的游戏客户端和服务器开发,软硬件联动,有啥恶心不恶心的,提前告诉你就是怕你接受不了,接受不了就没必要再往后走流程浪费时间,虽然这公司是一坨。
点赞 评论 收藏
分享
05-11 11:48
河南大学 Java
程序员牛肉:我是26届的双非。目前有两段实习经历,大三上去的美团,现在来字节了,做的是国际电商的营销业务。希望我的经历对你有用。 1.好好做你的CSDN,最好是直接转微信公众号。因为这本质上是一个很好的展示自己技术热情的证据。我当时也是烂大街项目(网盘+鱼皮的一个项目)+零实习去面试美团,但是当时我的CSDN阅读量超百万,微信公众号阅读量40万。面试的时候面试官就告诉我说觉得我对技术挺有激情的。可以看看我主页的美团面试面经。 因此花点时间好好做这个知识分享,最好是单拉出来搞一个板块。各大公司都极其看中知识落地的能力。 可以看看我的简历对于博客的描述。这个帖子里面有:https://www.nowcoder.com/discuss/745348200596324352?sourceSSR=users 2.实习经历有一些东西删除了,目前看来你的产出其实很少。有些内容其实很扯淡,最好不要保留。有一些点你可能觉得很牛逼,但是面试官眼里是减分的。 你还能负责数据库表的设计?这个公司得垃圾成啥样子,才能让一个实习生介入数据库表的设计,不要写这种东西。 一个公司的财务审批系统应该是很稳定的吧?为什么你去了才有RBAC权限设计?那这个公司之前是怎么处理权限分离的?这些东西看着都有点扯淡了。 还有就是使用Redis实现轻量级的消息队列?那为什么这一块不使用专业的MQ呢?为什么要使用redis,这些一定要清楚, 就目前看来,其实你的这个实习技术还不错。不要太焦虑。就是有一些内容有点虚了。可以考虑从PR中再投一点产出
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务