lintcode-198-排列序号II

198-排列序号II

给出一个可能包含重复数字的排列，求这些数字的所有排列按字典序排序后该排列在其中的编号。编号从1开始。

样例

给出排列[1, 4, 2, 2]，其编号为3。

思路

和 lintcode-197-排列序号 不同的是，这里的排序序列包含重复数字，所以需要排除重复排列。

在 lintcode-197-排列序号 的基础上，使用哈希表保存元素出现的次数，对于重复元素 A[i]，A[i] 在 [0,i] 区间中有重复元素，则用 A[i] 的 count * fact 的结果除以重复次数 dup 的阶乘。

code

class Solution {
public:
    /**
     * @param A an integer array
     * @return a long integer
     */
    long long permutationIndexII(vector<int>& A) {
        // Write your code here
        int size = A.size();
        if (size <= 0) {
            return 0;
        }

        long long result = 1, fac = 1, base = 1, dup = 1;
        map<int, int> hashRepeat;
        for (int i = size - 1; i >= 0; i--) {
            hashRepeat[A[i]]++;
            if (hashRepeat[A[i]] > 1) {
                dup *= hashRepeat[A[i]];
            }
            int count = 0;
            for (int j = i + 1; j < size; j++) {
                if (A[j] < A[i]) {
                    count++;
                }
            }
            result += count * fac / dup;
            fac *= (size - i);
        }
        return result;
    }
};