ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

2021.8.4 力扣-组合总和

2021-08-04 21:57:38  阅读:167  来源: 互联网

标签:index int 2021.8 力扣 vector candidates ans combine 总和


题目描述:

给定一个无重复元素的正整数数组 candidates 和一个正整数 target ,找出 candidates 中所有可以使数字和为目标数 target 的唯一组合。

candidates 中的数字可以无限制重复被选取。如果至少一个所选数字数量不同,则两种组合是唯一的。 

对于给定的输入,保证和为 target 的唯一组合数少于 150 个。

方法一:

class Solution {
public:
    vector<vector<int>> ans;
    vector<int> mycan;
    int mytar , n;
    void DFS(int index , int sum , vector<int> combine)
    {
        if(index == n)
        {
            return;
        }
        if(sum==mytar)
        {
            ans.push_back(combine);
            return;
        }
        DFS(index+1 , sum , combine);       //跳过第index个数
        if(sum+mycan[index] > mytar)
        {
            return ;
        }
        combine.push_back(mycan[index]);
        DFS(index , sum+mycan[index] , combine);    //选择第index个数
    }
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        sort(candidates.begin() , candidates.end());
        mycan = candidates;
        mytar = target;
        n = candidates.size();
        vector<int> combine;        //记录过程中选择的数字
        DFS(0 , 0 , combine);
        return ans;
    }
};

自己写回溯实在写不出来,看了看官方的题解做法。 

正如官方题解所说“对于这类寻找所有可行解的问题,我们都可以使用‘搜索回溯’的方法来解决”。

但速度实在有点慢,需要112ms

方法二: 

class Solution {
public:
    vector<vector<int>> ans;
    vector<int> mycan;
    int mytar , n;
    void DFS(int index , int sum , vector<int>& combine)
    {
        if(index == n)
        {
            return;
        }
        if(sum==mytar)
        {
            ans.push_back(combine);
            return;
        }
        for(int i=index ; i<n ; i++)
        //这里i初始值为index,说明同一个数字可以被选中多次,如果初始值为index+1,就说明不能有重复数字
        {
            if(sum+mycan[i] <= mytar)
            {
                combine.push_back(mycan[i]);
                sum+=mycan[i];
                DFS(i , sum , combine);
                combine.pop_back();
                sum-=mycan[i];
            }
        }
    }
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        sort(candidates.begin() , candidates.end());
        mycan = candidates;
        mytar = target;
        n = candidates.size();
        vector<int> combine;        //记录过程中选择的数字
        DFS(0 , 0 , combine);
        return ans;
    }
};

 对比方法一,用for循环将是否选中下标为index元素的两种情况整合到一起,这样比方法一更快,时间为12ms

方法二的改良:

class Solution {
public:
    void DFS(int index , int sum , vector<int>& combine ,vector<vector<int>>& ans , vector<int>& mycan , int mytar , int n)
    {
        if(index == n)
        {
            return;
        }
        if(sum==mytar)
        {
            ans.push_back(combine);
            return;
        }
        for(int i=index ; i<n ; i++)
        //这里i初始值为index,说明同一个数字可以被选中多次,如果初始值为index+1,就说明不能有重复数字
        {
            if(sum+mycan[i] <= mytar)
            {
                combine.push_back(mycan[i]);
                sum+=mycan[i];
                DFS(i , sum , combine , ans , mycan , mytar , n);
                combine.pop_back();
                sum-=mycan[i];
            }
        }
    }
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        vector<vector<int>> ans;
        //vector<int> mycan;
        //int mytar , n;
        sort(candidates.begin() , candidates.end());
        //mycan = candidates;
        //mytar = target;
        int n = candidates.size();
        vector<int> combine;        //记录过程中选择的数字
        DFS(0 , 0 , combine , ans , candidates , target , n);
        return ans;
    }
};

对比了一下别人的解法,发现对于ans、mycan、mytar、n这样的量,将其直接作为参数传递会进一步提高速度,这时只需用4ms即可

标签:index,int,2021.8,力扣,vector,candidates,ans,combine,总和
来源: https://blog.csdn.net/weixin_52629904/article/details/119387749

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有