题解 | #数组分组#

又是一道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);
            }
        }
    }
}


全部评论

相关推荐

野猪不是猪🐗:我assume that你must技术aspect是solid的,temperament也挺good的,however面试不太serious,generally会feel style上不够sharp
点赞 评论 收藏
分享
牛客383479252号:9,2学生暑期实习失利开始投小厂,给这群人整自信了
点赞 评论 收藏
分享
(黑话警告⚠️:hc=岗位数量,&nbsp;mt=导师,&nbsp;ld=直属领导,&nbsp;cr=代码审查)25年1月,我加入了字节某前端团队,并期望能在这里待到秋招并尝试转正。然而,就在上周,ld&nbsp;找我1v1,告诉我,我的能力和团队预期不太匹配,并和我劝退。晴天霹雳吗?肯定是有的。那一刻,脑子里嗡嗡作响,各种情绪翻涌。但冷静下来想想,这几个月,自己在能掌控的范围内,确实有不少地方做得不尽如人意。所以,我想把这段不算成功的经历复盘一下,希望能给同样在努力转正的你提个醒,避开我踩过的坑。一、ld&nbsp;的要求要注意刚进组时,ld就和我聊过转正的事。我当时发问:“咱们这儿有hc&nbsp;吗?”&nbsp;ld没直接回答,只是说:“看能力,能力到了...
牛客上的彭于晏:过来人告诉你,入职后要做的第一件事儿不是说主动找活儿做,你要先学会融入团队,摸清ld的性格,投其所好。然后才是展示你的能力,能力上可以说技术或者业务,以业务能力为主,技术能力为辅。优先保证自己对业务需求的开发保证质量效率,然后再谈技术的问题,不要你觉得啥啥啥不行就想着整体优化了(发现校招生最喜欢干这事儿),我工作快5年了发现搞这种的最后都没啥好的结果,产出没有还引入新的bug,校招或者实习的水平看到的问题别人看不到嘛?为什么别人不去搞?浪费时间还没收益的事儿不要去做,技术上的能力体现在对于一个新需求,在不符合现在业务发展的架构设计上,你能拿出好的技术方案同时能考虑到后续业务发展逐渐将技术架构引入合理的架构,这是一个漫长的过程而不是一次性的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务