473.火柴拼正方形

题目描述

你将得到一个整数数组 matchsticks ，其中 matchsticks[i] 是第 i 个火柴棒的长度。你要用 所有的火柴棍 拼成一个正方形。你 不能折断 任何一根火柴棒，但你可以把它们连在一起，而且每根火柴棒必须 使用一次 。

如果你能使这个正方形，则返回 true ，否则返回 false 。

示例 1:

输入: matchsticks = [1,1,2,2,2]
输出: true
解释: 能拼成一个边长为2的正方形，每边两根火柴。

示例 2:

输入: matchsticks = [3,3,3,3,4]
输出: false
解释: 不能用所有火柴拼成一个正方形。

提示:

    1 <= matchsticks.length <= 15
    1 <= matchsticks[i] <= 108

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/matchsticks-to-square
 

方法一：

动态规划，回溯，对于火柴长度先排序，然后每根火柴放入四条边看有没有成功的情况。

class Solution {
public:
    bool dfs(int n,vector<int> &matchsticks,vector<int> &edges,int d){
        if(n==matchsticks.size()) return true;
        for(int i=0;i<edges.size();i++){
            edges[i]+=matchsticks[n];
            if(edges[i]<=d&&dfs(n+1,matchsticks,edges,d)) return true;
            edges[i]-=matchsticks[n];
        }
        return false;
    }
    bool makesquare(vector<int>& matchsticks) {
        int sum=accumulate(matchsticks.begin(),matchsticks.end(),0);
        if(sum%4!=0) return false;
        sort(matchsticks.begin(),matchsticks.end(),greater<int>());
        vector<int> edges(4);
        return dfs(0,matchsticks,edges,sum/4);
    }
};