【Leetcode】1286. Iterator for Combination

题目地址：

https://leetcode.com/problems/iterator-for-combination/

给定一个长$n$的字符串$s$，$s$只含不同的字符，并且按字典序已经从小到大排好。再给定一个正整数$m$，要求设计一个迭代器，每次返回长$m$的字典序大$1$的子序列。

字典序最小的子序列当然就是$s$的长$m$前缀。给定给一个子序列$t$，我们将其映射到一个二进制数，$1$表示$s$的这一位取，否则代表不取。我们考虑怎么让字典序变得稍微大一些，其实就是让那个二进制数小一点。如果$t$恰好是$s$的后缀，那已经最大了；如果不是，我们从后向前找第一个$0$，然后再继续向前找到第一个$1$，那么这个$0$之后的部分已经最小了，我们让那个$1$向后挪一步，然后将其后面的$1$全向前挪到这个$1$后面即可。代码如下：

class CombinationIterator {
 public:
  string s;
  int n, m, state;
  CombinationIterator(string characters, int combinationLength) {
    s = characters;
    n = s.size(), m = combinationLength, state = 0;
    for (int i = 0; i < m; i++) state |= 1 << i;
  }

  string next() {
    string res;
    for (int i = 0; i < n; i++)
      if (state >> i & 1) res += s[i];

    int k = n - 1;
    while (k >= 0 && (state >> k & 1)) k--;
    if (n - k - 1 == m)
      state = -1;
    else {
      int cnt = n - k;
      while (k >= 0 && !(state >> k & 1)) k--;
      for (int i = k; i < n; i++)
        if (state >> i & 1) state -= 1 << i;
      for (int i = k + 1, j = 0; j < cnt; i++, j++) state |= 1 << i;
    }

    return res;
  }

  bool hasNext() { return ~state; }
};

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

next的时间复杂度$O(n)$，hasNext时间$O(1)$，空间$O(n)$。