leetcode 1286. 字母排列迭代器（C++）

请你设计一个迭代器类，包括以下内容：

• 一个构造函数，输入参数包括：一个 有序且字符唯一 的字符串 characters（该字符串只包含小写英文字母）和一个数字 combinationLength 。
• 函数 next() ，按 字典序 返回长度为 combinationLength 的下一个字母排列。
• 函数 hasNext() ，只有存在长度为 combinationLength 的下一个字母排列时，才返回 True；否则，返回 False。

 

示例：

CombinationIterator iterator = new CombinationIterator("abc", 2); // 创建迭代器 iterator

iterator.next(); // 返回 "ab"
iterator.hasNext(); // 返回 true
iterator.next(); // 返回 "ac"
iterator.hasNext(); // 返回 true
iterator.next(); // 返回 "bc"
iterator.hasNext(); // 返回 false

 

提示：

• 1 <= combinationLength <= characters.length <= 15
• 每组测试数据最多包含 10^4 次函数调用。
• 题目保证每次调用函数 next 时都存在下一个字母排列。

C++

class CombinationIterator {
public:
    void dfs(queue<string>& tmp, string characters, int count, int start, string str)
    {
        if(str.size()==count)
        {
            tmp.push(str);
            return;
        }
        for(int i=start;i<characters.size();i++)
        {
            str+=characters[i];
            dfs(tmp,characters,count,i+1,str);
            int m=str.size();
            str=str.substr(0,m-1);
        }
    }
    
    CombinationIterator(string characters, int combinationLength) 
    {
        queue<string> tmp;
        string str="";
        dfs(tmp,characters,combinationLength,0,str);
        que=tmp;
    }
    
    string next() 
    {
        auto it=que.front();
        que.pop();
        return it;
    }
    
    bool hasNext() 
    {
        if(que.size())
        {
            return true;
        }
        return false;
    }

private:
    queue<string> que;
};

/**
 * 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();
 */