LeetCode 5289. 公平分发饼干(集合划分问题/回溯)

最新推荐文章于 2024-01-07 16:54:48 发布

xylitolz

最新推荐文章于 2024-01-07 16:54:48 发布

阅读量285

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法深度优先 java

本文链接：https://blog.csdn.net/xylitolz/article/details/125247849

版权

LeetCode 专栏收录该内容

55 篇文章 0 订阅

订阅专栏

文章目录

题目
解法：回溯
进阶解法：模拟退火
集合划分问题
Reference

题目

5289. 公平分发饼干

在这里插入图片描述

解法：回溯

暴力解：枚举每包糖果属于哪个孩子， $8\times8=2^{24}$ 大概是 $10^7$

class Solution {
    int[] q;
    int k;
    int[] sum;
    int ans;
    public int distributeCookies(int[] cookies, int k) {
        q = cookies;
        this.k = k;
        sum = new int[8];
        ans = Integer.MAX_VALUE;
        dfs(0);
        return ans;
    }

    private void dfs(int u) {
        if (u == q.length) {
            int mx = 0;
            for (int i = 0; i < k; i++) {
                mx = Math.max(mx, sum[i]);
            }
            ans = Math.min(ans, mx);
            return ;
        }
        
        for (int i = 0; i < k; i++) {
            sum[i] += q[u];
            if (sum[i] < ans) {
                dfs(u + 1);
            }
            sum[i] -= q[u];
        }
    }
}