(M)DFS:473. Matchsticks to Square
2017-10-15 16:09
239 查看
这道题实际上是求一个数组能否分成四个和相等的子数组。思路是定义四个组(vector),遍历这个大数组,看nums[i]加在哪一个组中,加在这个组中,对下一个数字进行递归,如果返回false,则说明加在这个组中无解,就在这个组中减掉,看能不能加到下一个组中。
大神分析:
这道题让我们用数组中的数字来摆出一个正方形。跟之前有道题Partition Equal Subset Sum有点像,那道题问我们能不能将一个数组分成和相等的两个子数组,而这道题实际上是让我们将一个数组分成四个和相等的子数组。我一开始尝试着用那题的解法来做,首先来判断数组之和是否是4的倍数,然后还是找能否分成和相等的两个子数组,但是在遍历的时候加上判断如果数组中某一个数字大于一条边的长度时返回false。最后我们同时检查dp数组中一条边长度位置上的值跟两倍多一条边长度位置上的值是否为true,这种方法不幸TLE了。所以只能上论坛求助各路大神了,发现了可以用优化过的递归来解,递归的方法基本上等于brute
force,但是C++版本的直接递归没法通过OJ,而是要先给数组从大到小的顺序排序,这样大的数字先加,如果超过target了,就直接跳过了后面的再次调用递归的操作,效率会提高不少,所以会通过OJ。下面来看代码,我们建立一个长度为4的数组sums来保存每个边的长度和,我们希望每条边都等于target,数组总和的四分之一。然后我们遍历sums中的每条边,我们判断如果加上数组中的当前数字大于target,那么我们跳过,如果没有,我们就加上这个数字,然后对数组中下一个位置调用递归,如果返回为真,我们返回true,否则我们再从sums中对应位置将这个数字减去继续循环,参见代码如下:
class Solution { public: bool makesquare(vector<int>& nums) { if (nums.empty() || nums.size() < 4) return false; int sum = accumulate(nums.begin(), nums.end(), 0); if (sum % 4 != 0) return false; vector<int> sums(4, 0); sort(nums.rbegin(), nums.rend()); return helper(nums, sums, 0, sum / 4); } bool helper(vector<int>& nums, vector<int>& sums, int pos, int target) { if (pos >= nums.size()) { return sums[0] == target && sums[1] == target && sums[2] == target; } for (int i = 0; i < 4; ++i) { if (sums[i] + nums[pos] > target) continue; sums[i] += nums[pos]; if (helper(nums, sums, pos + 1, target)) return true; sums[i] -= nums[pos]; } return false; } };
相关文章推荐
- hdoj 1045 Fire Net (dfs)
- 欧拉回路输出(DFS,不用回溯!)Watchcow POJ 2230
- leetcode 684. Redundant Connection 邻接表的环的判断 + 深度优先遍历DFS
- 找质因数..打素数表..DFS解容斥问题...
- 【DFS】【最短路】【spfa】【BFS】洛谷P2296 NOIP2014提高组 day2 T2 寻找道路
- codeforces 374C Inna and Dima (dfs+记忆化)
- HDU - 1312 : Red and Black(dfs、bfs)
- 51nod 40 序列分解 暴力dfs or 折半+字典树
- HDU 1045 Fire Net(DFS)
- 走迷宫(Dfs)
- uva 548 - Tree 二叉树重构(中序和后序)+dfs
- HDU 3038 How Many Answers Are Wrong 带权并查集 dfs过程中维护变量
- 最短路的方案数 luogu1606 ,找出所有的最短路dfs
- DFS+Trie—— DrawSomething辅助 java版
- POJ 1111 Image Perimeters (裸DFS)
- Codeforces 707D Persistent Bookcase(离线dfs或在线主席树)
- 2017/8/3训练日记(DFS&&Codeforces Round #418 (Div. 2))
- CodeForces 258 B.Little Elephant and Elections(数位DP+dfs)
- pku 3009 Curling 2.0 DFS
- poj1562 & zoj1709 - Oil Deposits (DFS)