生活随笔
收集整理的這篇文章主要介紹了
LeetCode 698. 划分为k个相等的子集（回溯）
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
文章目錄
1. 題目
給定一個整數數組 nums 和一個正整數 k，找出是否有可能把這個數組分成 k 個非空子集，其總和都相等。
示例
1：
輸入： nums
= [4, 3, 2, 3, 5, 2, 1], k
= 4
輸出： True
說明： 有可能將其分成
4 個子集（
5），（
1,4），（
2,3），（
2,3）等于總和。提示：
1 <= k
<= len(nums
) <= 16
0 < nums
[i
] < 10000
來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/partition-to-k-equal-sum-subsets
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。
2. 解題
類似題目：LeetCode 416. 分割等和子集（動態規劃）
class Solution {int target
;bool ans
= false;int K
;
public:bool canPartitionKSubsets(vector
<int>& nums
, int k
) {int sum
= accumulate(nums
.begin(), nums
.end(), 0);K
= k
;if(sum
%k
) return false;target
= sum
/k
;vector
<int> partsum(k
, 0);dfs(nums
, partsum
, 0);return ans
;}void dfs(vector
<int>& nums
, vector
<int>& partsum
, int i
){if(ans
== true) return;if(i
== nums
.size()){bool flag
= true;for(int k
= 0; k
< K
; k
++){if(partsum
[k
] != target
){flag
= false;break;}}if(flag
)ans
= true;return;}for(int j
= 0; j
< K
; ++j
){if(partsum
[j
]+nums
[i
] <= target
){partsum
[j
] += nums
[i
];dfs(nums
, partsum
, i
+1);partsum
[j
] -= nums
[i
];}}}
};
class Solution {int target
;bool ans
= false;int K
;
public:bool canPartitionKSubsets(vector
<int>& nums
, int k
) {int sum
= accumulate(nums
.begin(), nums
.end(), 0);K
= k
;if(sum
%k
) return false;target
= sum
/k
;sort(nums
.rbegin(), nums
.rend());if(nums
[0] > target
)return false;vector
<int> partsum(k
, 0);dfs(nums
, partsum
, 0);return ans
;}void dfs(vector
<int>& nums
, vector
<int>& partsum
, int i
){if(ans
== true) return;if(i
== nums
.size()){bool flag
= true;for(int k
= 0; k
< K
; k
++){if(partsum
[k
] != target
){flag
= false;break;}}if(flag
)ans
= true;return;}for(int j
= 0; j
< K
; ++j
){if(partsum
[j
]+nums
[i
] <= target
){partsum
[j
] += nums
[i
];dfs(nums
, partsum
, i
+1);partsum
[j
] -= nums
[i
];}}}
};
168 ms 9.2 MB
我的CSDN博客地址 https://michael.blog.csdn.net/
長按或掃碼關注我的公眾號（Michael阿明），一起加油、一起學習進步！
總結
以上是生活随笔為你收集整理的LeetCode 698. 划分为k个相等的子集（回溯）的全部內容，希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。
