LeetCode 473.Matchsticks to Square【Java】

题目描述

火柴拼正方形

AC代码

运行时间：升序剪枝>乱序剪枝>降序剪枝

/*
剪枝：
1.从大到小枚举所有边
2.每条边内部的木棒长度规定成从大到小
3.如果当前木棒拼接失败，则跳过接下来所有长度相同的木棒
4.如果当前木棒拼接失败，而且是当前边的第一个，则直接剪掉当前分支
5.如果当前木棒拼接失败，而且是当前边的最后一个，那么直接剪掉当前分支
*/
class Solution {

    boolean[] vis;
    public boolean makesquare(int[] nums) {
        int sum=0;
        for(int x:nums) sum+=x;
        if(sum==0||sum%4!=0) return false;
        vis=new boolean[nums.length];
        //因为是从大到小枚举所有边，所以需要从大到小排个序
        Arrays.sort(nums);
        reverse(nums);
        return dfs(nums,0,0,sum/4);
    }

    boolean dfs(int[] nums,int u,int cur,int length){
        if(cur==length){u++;cur=0;}
        if(u==4) return true;
        for(int i=0;i<nums.length;i++){
            if(!vis[i]&&cur+nums[i]<=length){
                vis[i]=true;
                if(dfs(nums,u,cur+nums[i],length)) return true;
                vis[i]=false;
                if(cur==0) return false;
                if(cur+nums[i]==length) return false;
                while(i+1<nums.length&&nums[i+1]==nums[i]) i++;
            }
        }
        return false;
    }

    void reverse(int []nums){
        int temp = 0;
        int len=nums.length;
        for (int i = 0; i < len>>1; i++) {
            temp = nums[i];
            nums[i] = nums[len - i - 1];
            nums[len - i - 1] = temp;
        }
    }

}