状态压缩+减枝

最新推荐文章于 2024-06-20 22:41:06 发布

wweTHEUT

最新推荐文章于 2024-06-20 22:41:06 发布

阅读量108

点赞数

分类专栏：算法学习笔记文章标签：算法记忆压缩剪枝

本文链接：https://blog.csdn.net/wweTHEUT/article/details/125092093

版权

算法学习笔记专栏收录该内容

38 篇文章 0 订阅

订阅专栏

这篇博客探讨了一种利用状态压缩和回溯算法解决火柴棒拼成正方形的问题。代码中展示了如何通过动态规划和排序来优化搜索过程，并在给定的火柴棒长度总和能被4整除的情况下，判断是否可以拼成正方形。文章深入到算法细节，包括剪枝策略和记忆化搜索，旨在帮助读者理解这类问题的解决思路。

摘要由CSDN通过智能技术生成

代码

这里自己能够想到的是状态压缩+剪纸以及回溯，至于动态规划个人直觉上有但是没有想出来。

var makesquare = function(matchsticks) {
    var lon = 0;
    let msk = 0 << matchsticks.length;
    var memo = new Map(); // 记忆，避免重复扫描
    for(let i = 0; i < matchsticks.length; i++) {
        lon += matchsticks[i];
    }
    var side = lon / 4;
    if(lon % 4 != 0) { return false; }
    matchsticks.sort((a,b)=>{
        return a - b;
    });
    if(matchsticks[0] > side) { return false; }
    const dfs = (mask,t,c,d) => {
        if(!memo.has(mask)) {
            let res = false;
            for(let i = 0; i < matchsticks.length; i++) {
                if(((mask >> i) & 1) === 0) {
                    let t_mask = mask;
                    t_mask = mask | (1 << i);
                    let to = t + matchsticks[i];
                    let tc = c;
                    if(to == side) { to = 0; tc++;}
                    if(tc == 4 && d == matchsticks.length) {return true; }
                    if(tc == 4 && d < matchsticks.length) { return false; }
                    if(to > side) { continue; }
                    if(dfs(t_mask,to,tc,d + 1)) { res = true; }
                }
            }
            memo.set(mask,res);
        }
        return memo.get(mask);
    }
    return dfs(msk,0,0,1);
};