组合-lintcode

这是回溯类型的简单题。
分析:
- 函数一定要传一个引用类型的vector,以便以后可以往里加数据。其他参数包括n,k,temp(这个就不可以是引用了),还有最重要的一个i,这个函数栈里应该从哪取。
- 函数里首先应该判断return的条件,即result.size()==k或者temp的最后一个元素达到最大n时return。接下来是for循环,每次重新分配一个temp2,然后往里加i,再调用函数,这样可以从小到大找完所有的可能。
C++代码:

class Solution {
public:
    /**
     * @param n: Given the range of numbers
     * @param k: Given the numbers of combinations
     * @return: All the combinations of k numbers out of 1..n
     */
    vector<vector<int> > combine(int n, int k) {
        vector<vector<int>> result;
        vector<int> temp;
        if (0==n || 0==k){
            return result;
        }
        process(result,temp,n,k,1);
        return result;
    }
    void process(vector<vector<int>>&result, vector<int> temp, int n, int k,int i) {
        if (temp.size() == k) {
            result.push_back(temp);
            return;
        }
        if (temp.size()>=1 && temp[temp.size()-1] == n) {
            return;
        }

        for ( ; i<=n; i++) {
            vector<int> temp2 = temp;
            temp2.push_back(i);
            process(result,temp2,n,k,i+1);
        }
    }
};

然后我又看了网友的代码,对每个元素,取或不取,递归调用。
他的代码:

class Solution {  
public:  
    /** 
     * @param n: Given the range of numbers 
     * @param k: Given the numbers of combinations 
     * @return: All the combinations of k numbers out of 1..n 
     */  
    vector<vector<int> > combine(int n, int k) {  
        // write your code here  
        vector<vector<int> > ret;  
        vector<int> cur;  
        combine(cur,1,n,k,ret);  
        return ret;  
    }  
    void combine(vector<int> cur,int index,int n,int k,vector<vector<int> >&ret)  
    {  
        if(cur.size()==k)  
        {  
            ret.push_back(cur);  
            return;  
        }  
        if(index>n||cur.size()>k)  
            return;  
        combine(cur,index+1,n,k,ret);  
        cur.push_back(index);  
        combine(cur,index+1,n,k,ret);  
    }  

}; 

最后三行加他的解释简直道出了回溯的本质,第一行表示不取index,第二行取该元素index,地三行表示取该元素index。
总结:
发现别人的思想总能给你启发,要多多发现并吸收;另外,感觉回溯法用于解决那些组合类型的问题,就是看看哪些组合是满足要求的,然后递归求解,八皇后问题也是如此。

你可能感兴趣的:(算法,LintCode,回溯法,LintCode)