Leetcode 698. 划分为k个相等的子集 解题思路及C++实现
生活随笔
收集整理的這篇文章主要介紹了
Leetcode 698. 划分为k个相等的子集 解题思路及C++实现
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
解題思路:
這是一個典型的深度優先搜索問題。使用一個相同大小的數組visited來記錄元素是否被訪問過(即是被已被劃分到相應的子集中)。
每找到一個劃分子集,相應對k減一,再繼續執行深度優先搜索程序。
?
class Solution { public:bool canPartitionKSubsets(vector<int>& nums, int k) {int sum = accumulate(nums.begin(), nums.end(), 0); //求總和if(sum % k != 0) return false; //如果總和不能被平均分配到k份,就返回falsevector<bool> visited(nums.size(), false); //記錄元素是否已經被劃分過return dfs(nums, visited, 0, sum/k, 0, k);}bool dfs(vector<int>& nums, vector<bool>& visited, int start, int target, int cursum, int k){if(k == 1) return true; //只剩下一個待分集合,就可以直接返回true了else if(cursum > target) return false;else if(cursum == target) return dfs(nums, visited, 0, target, 0, k-1); //找到一個分配集合,則要將搜索起始位置start重置為0,待分配子集剩下k-1個else{for(int i = start; i < nums.size(); i++){ //從start開始找if(visited[i] == false){ //看是否已被訪問過visited[i] = true; //訪問當前元素if(dfs(nums, visited, i+1, target, cursum+nums[i], k)) return true;visited[i] = false;}}return false;}} };?
?
?
總結
以上是生活随笔為你收集整理的Leetcode 698. 划分为k个相等的子集 解题思路及C++实现的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Leetcode 211. 添加与搜索单
- 下一篇: Leetcode 292. Nim 游戏