Leetcode 3193. Count the Number of Inversions

  • [Leetcode 3193. Count the Number of Inversions](#Leetcode 3193. Count the Number of Inversions)
    • [1. 解题思路](#1. 解题思路)
    • [2. 代码实现](#2. 代码实现)

1. 解题思路

这一题的话我的思路核心还是动态规划,因此核心就是迭代关系如何构建。

我们定义 f ( n , k ) f(n, k) f(n,k)表示将1到 n n n一共n个元素进行排列,一共有 k k k个逆序pair的排列数目。我们考察 f ( n + 1 , k ) f(n+1, k) f(n+1,k)的构造方式。

一种直接的构造就是基于 f ( n , k ) f(n,k) f(n,k),我们考虑如何加入元素 n + 1 n+1 n+1即可。如果将 n + 1 n+1 n+1放到最后一个元素,则必然有 k k k个序列pair,如果要有 k + p k+p k+p个逆序pair,那么我们基于 f ( n , k ) f(n,k) f(n,k)当中的任意排列进行改造,只需要将元素 n − p n-p n−p放置到最后一个元素,然后将所有原序列当中所有大于等于 n − p n-p n−p的元素加1,就可以保持前 n n n个元素依然保持 k k k个逆序pair,而总的 n + 1 n+1 n+1个元素有 k + p k+p k+p的逆序pair。

综上,我们即可整理得到迭代关系:

f ( n + 1 , k ) = ∑ i = m a x ( 0 , k − n ) k f ( n , i ) f(n+1, k) = \sum\limits_{i = \mathop{max}(0, k-n)}^{k} f(n, i) f(n+1,k)=i=max(0,k−n)∑kf(n,i)

我们将其翻译为代码语言并加入限制条件即可。

2. 代码实现

给出python代码实现如下:

python 复制代码
MOD = 10**9+7

class Solution:
    def numberOfPermutations(self, n: int, requirements: List[List[int]]) -> int:
        requirements = sorted(requirements)
        rq = {ed: cnt for ed, cnt in requirements}
        
        @lru_cache(None)
        def dp(idx, cnt):
            if idx == 0:
                return 1 if cnt == 0 else 0
            elif cnt > idx * (idx+1) // 2 or cnt < 0:
                return 0
            elif idx in rq and cnt != rq[idx]:
                return 0
            elif idx-1 in rq:
                return dp(idx-1, rq[idx-1]) if cnt-idx <= rq[idx-1] <= cnt else 0
            ans = 0
            for pre_cnt in range(cnt-idx, cnt+1):
                ans += dp(idx-1, pre_cnt)
            return ans % MOD
        
        ed, cnt = requirements[-1]
        ans = dp(ed, cnt)
        return ans

提交代码评测得到:耗时8757ms,占用内存116.3MB。

相关推荐
恣艺6 小时前
LeetCode 124:二叉树中的最大路径和
算法·leetcode·职场和发展
1白天的黑夜16 小时前
前缀和-1314.矩阵区域和-力扣(LeetCode)
c++·leetcode·前缀和
weisian1516 小时前
力扣经典算法篇-42-矩阵置零(辅助数组标记法,使用两个标记变量)
算法·leetcode·矩阵
恣艺7 小时前
LeetCode 123:买卖股票的最佳时机 III
算法·leetcode·职场和发展
Q741_1477 小时前
优选算法 力扣 202.快乐数 快慢双指针 解决带环问题 C++解题思路 每日一题
c++·算法·leetcode·快慢双指针·环形问题
UP_Continue9 小时前
算法讲解--最大连续1的个数
数据结构·算法·leetcode
设计师小聂!10 小时前
力扣热题100--------240.搜索二维矩阵
java·算法·leetcode·矩阵
岁忧19 小时前
(LeetCode 面试经典 150 题) 138. 随机链表的复制 (哈希表)
java·c++·leetcode·链表·面试·go
菜鸡nan1 天前
23th Day| 39.组合总和,40.组合总和II,131.分割回文串
算法·leetcode·职场和发展
qq_513970441 天前
力扣 hot100 Day63
数据结构·算法·leetcode