Leetcode#47 Permutations II

原题地址

 

字典序模拟问题，枚举num的所有字典序。

假设已知序列seq，求下一个字典序序列：

1. 从序列seq右侧向左寻找第一个顺序对，即对于位置i，有seq[i] < seq[i+1]。如果找不到i，则说明已经没有下一个序列了，算法退出
2. 从序列seq右侧向左寻找第一个比seq[i]大的数，即seq[j] > seq[i]。肯定能找到这样一个j，因为j至少可以是i+1
3. 交换seq[i]和seq[j]
4. 将seq在位置i之后的部分倒置，即将seq[i+1..n-1]翻转

以上过程实际上就是在模拟手算下一个字典序的过程，如果不明白，模拟几组数据就明白了

PS：STL自带生成字典序的函数std::next_permutation

 

代码：排序+生成字典序

int findLastAscendingPos(vector<int> &num) {
  int i = num.size() - 2;
  while (i >= 0 && num[i] >= num[i + 1])
    i--;
  return i;
}

int findLastGreaterPos(vector<int> &num, int target) {
  int i = num.size() - 1;
  while (i >= 0 && num[i] <= target)
    i--;
  return i;
}

vector<vector<int> > permuteUnique(vector<int> &num) {
  vector<vector<int> > pms;
  int n = num.size();
  sort(num.begin(), num.end());

  pms.push_back(num);
  while (true) {
    vector<int> next = pms.back();
    int i = findLastAscendingPos(next);
    if (i < 0) break;
    int j = findLastGreaterPos(next, next[i]);
    swap(next[i], next[j]);
    reverse(next.begin() + i + 1, next.end());
    pms.push_back(next);
  }

  return pms;
}