在您提供的代码中,有几个问题需要注意:
函数
backtrace的第二个参数应该传入一个拷贝的map,而不是原始的map。因为在回溯过程中会修改这个map,所以最好操作副本。在循环遍历当前数字重复次数的时候,内层循环应该从0开始,而不是从1开始。因为您希望生成所有可能的组合,包括没有重复元素的情况。
在递归结束条件判断部分,应该使用小于等于号(<=)来比较
epoch和total的大小关系。否则会导致漏掉一些结果。
下面是修改后的代码:
class Solution {
public:
std::unordered_map<int, int> Map;
vector<vector<int>> result;
void backtrace(vector<int> path, unordered_map<int, int> map, int total, int epoch) {
if (epoch > total) {
result.push_back(path);
return;
}
auto p = *(map.begin());
map.erase(p.first);
for (int i = 0; i <= p.second; i++) {
for (int j = 0; j < i; j++) {
path.push_back(p.first);
}
backtrace(path, map, total, epoch + 1);
for (int j = 0; j < i; j++) {
path.pop_back();
}
}
}
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
for (int i = 0; i < nums.size(); i++) {
Map[nums[i]]++;
}
backtrace(vector<int>{}, Map, Map.size(), 1);
return result;
}
};
希望对您有帮助!
内容由零声教学AI助手提供,问题来源于学员提问




