1286. Iterator for Combination

问题：

设计一个组合指针类，通过给定组合元素characters，要求构成组合的大小combinationLength，实现以下方法

构造方法：CombinationIterator(string characters, int combinationLength)

返回下一个组合：next() 

是否存在下一个组合：hasNext() 

Example 1:
Input
["CombinationIterator", "next", "hasNext", "next", "hasNext", "next", "hasNext"]
[["abc", 2], [], [], [], [], [], []]
Output
[null, "ab", true, "ac", true, "bc", false]

Explanation
CombinationIterator itr = new CombinationIterator("abc", 2);
itr.next();    // return "ab"
itr.hasNext(); // return True
itr.next();    // return "ac"
itr.hasNext(); // return True
itr.next();    // return "bc"
itr.hasNext(); // return False
 

Constraints:
1 <= combinationLength <= characters.length <= 15
All the characters of characters are unique.
At most 104 calls will be made to next and hasNext.
It's guaranteed that all calls of the function next are valid.

　　

解法：

保存构造体：vector<string>

解法一：Backtracking（回溯算法）

• 状态：到当前位置为止，构成的组合path。
• 选择：从给定组合元素的第pos个开始（前面的元素，path已经选择过），到最后一个都可选择。
• 递归退出条件：path.size=combinationLength(要求的组合大小)

 

代码参考：

class CombinationIterator {
public:
    vector<string> comb;
    int itr;
    void dfs(string characters, int combinationLength, int pos, string path) {
        if(combinationLength == path.length()) {
            comb.push_back(path);
            return;
        }
        for(int i=pos; i<characters.size(); i++) {
            dfs(characters, combinationLength, i+1, path+characters[i]);
        }
    }
    CombinationIterator(string characters, int combinationLength) {
        itr=0;
        dfs(characters, combinationLength, 0, "");
    }
    
    string next() {
        return itr!=comb.size()?comb[itr++]:"";
    }
    
    bool hasNext() {
        return itr!=comb.size();
    }
};

/**
 * Your CombinationIterator object will be instantiated and called as such:
 * CombinationIterator* obj = new CombinationIterator(characters, combinationLength);
 * string param_1 = obj->next();
 * bool param_2 = obj->hasNext();
 */

 

解法二：

保存构造体：set<string>（由题意，有顺序要求）

bit组合法：

每个元素代表一个bit位，选择表示1，不选表示0

那么，题意中要求选择长度combinationLength的元素构成组合，即：选择所有元素构成bit中含有2个1的组合。

再将这两个1对应的元素拿出来，拼成解，加入解list中。

 

参考：LeetCode Thought

解法关联

Suppose Input string - "abc" and combinationLength =2
    now mask = 8 which is nothing but 2 power length of of "abc"
    now Itarate from 1 to 8 

   Num             bit_rep           no_of_set_bit       a   b   c            hold_set
    1               0 0 1                 1              0   0   1              nothing   <= because combinationLength != no_of_set_bit
    2               0 1 0                 1              0   1   0              same case like upper one
    3               0 1 1                 2              0   1   1              "bc" <= combinationLength ==no_of
    4               1 0 0                 1              1   0   0              nothing
    5               1 0 1                 2              1   0   1               "ac"
    6               1 1 0                 2              1   1   0               "ab"
    7               1 1 1                 3              1   1   1              nothing <= because combinationLength != no_of_set_bit

    return that set in lexi order ["ab","ac","bc"]
   
Now We Done Our Main Work

　　

代码参考：

class CombinationIterator {
public:
    //vector<string> comb;
    set<string> comb;
    //int itr;
    set <string> :: iterator itr;
    /*void dfs(string characters, int combinationLength, int pos, string path) {
        if(combinationLength == path.length()) {
            comb.push_back(path);
            return;
        }
        for(int i=pos; i<characters.size(); i++) {
            dfs(characters, combinationLength, i+1, path+characters[i]);
        }
    }*/
    void creatcomb(string s, int len) {
        int maxn = 1<<s.length();
        for(int cmb=1; cmb<maxn; cmb++) {
            //count the num of 1:
            if(__builtin_popcount(cmb)!=len) continue;
            string newcomb;
            //the bit 1 in cmb means the alphbet in s would be selected.
            for(int i=cmb, idx=0; i>0; i>>=1, idx++) {
                if(i&1) newcomb+=s[idx];
            }
            comb.insert(newcomb);
        }
        return;
    }
    CombinationIterator(string characters, int combinationLength) {
        //itr=0;
        creatcomb(characters, combinationLength);
        itr=comb.begin();
        //dfs(characters, combinationLength, 0, "");
    }
    
    string next() {
        return itr!=comb.end()?*itr++:"";
    }
    
    bool hasNext() {
        return itr!=comb.end();
    }
};

/**
 * Your CombinationIterator object will be instantiated and called as such:
 * CombinationIterator* obj = new CombinationIterator(characters, combinationLength);
 * string param_1 = obj->next();
 * bool param_2 = obj->hasNext();
 */