【Lintcode】197. Permutation Index

题目地址：

https://www.lintcode.com/problem/permutation-index/description

给定一个数组$A$，求其在所有全排列的字典序中的排名，从$1$开始计数。题目保证数组中的数两两不同。

用康托展开。康托展开$f$是从一个全排列到其字典序排名的一个映射（这个排名是从$0$开始的）。对于一个全排列$X=(x_1,x_2,...,x_n)$，有：$$f(X)=a_1(n-1)!+a_2(n-2)!+...+a_{n}0!$$其中$a_i$指的是$x_i$右边比$x_i$小的数的个数（参考https://blog.csdn.net/qq_46105170/article/details/108505684）。所以只需要对$A$中每个数求一下其右边比它小的数的个数，然后乘以对应的阶乘即可。由于阶乘中的的$n$在$f$的表达式中是从左到右依次递减的，所以为了节省时间，我们可以逆序遍历$A$。求每个数右边比其小的数的个数可以用树状数组做优化。代码如下：

class Solution {
public:
  /**
   * @param a: An array of integers
   * @return: A long integer
   */
  vector<int> tr;
  int n;
#define lowbit(x) (x & -x)

  void add(int k, int x) {
    for (; k <= n; k += lowbit(k)) tr[k] += x;
  }

  int sum(int k) {
    int res = 0;
    for (; k; k -= lowbit(k)) res += tr[k];
    return res;
  }

  using LL = long long;
  LL permutationIndex(vector<int> &a) {
    // write your code here
    n = *max_element(a.begin(), a.end());
    tr.resize(n + 1);
    LL res = 0, fact = 1;
    for (int i = a.size() - 1; i >= 0; i--) {
      add(a[i], 1);
      res += fact * sum(a[i] - 1);
      fact *= a.size() - i;
    }

    return res + 1;
  }
};

时间复杂度$O(n\log n)$，空间$O(n)$。