leetcode刷题笔记-统计上升四元组

最新推荐文章于 2024-05-20 20:08:58 发布

Five_L

最新推荐文章于 2024-05-20 20:08:58 发布

阅读量477

点赞数

分类专栏： leetcode刷题笔记文章标签： leetcode 算法 python

本文链接：https://blog.csdn.net/Five_L/article/details/129911430

版权

leetcode刷题笔记专栏收录该内容

1 篇文章 0 订阅

订阅专栏

2552.统计上升四元组

题目链接

题解参考

基本思路：

以中间两个数j, k作为参考，对不同的j, k找出可行的i, l数量
维护两个数组
1. greater[k][x]，表示nums中位于k（索引）的右侧比x（数值）大的数的个数
2. less[j][y]，表示nums中位于j（索引）的左侧比y（数值）小的数的个数
数组更新方法，两层循环
1. $greater[k][x]=greater[k+1][x]+\left\{ \begin{matrix} 1, x<nums[k+1] \\0, otherwise \end{matrix} \right.$ ，此处k用倒序遍历，因为累计右侧数量，需要从右侧开始计算，x遍历[1, nums[k+1])
2. $less[j][y]=less[j-1][y]+\left\{ \begin{matrix} 1, y>nums[j-1] \\0, otherwise \end{matrix} \right.$ ，此处j用正序遍历，因为累计左侧数量，y遍历[nums[j-1], n+1)
结果统计

$ans=\sum \sum greater[k][nums[j]]*less[j][nums[i]]$

代码：

class Solution:
    def countQuadruplets(self, nums: List[int]) -> int:
        n = len(nums)
        great=[[0]*(n+1) for _ in range(n)]
        ans=0
        for k in range(n-2, 1, -1):
            great[k]=great[k+1][:]
            for x in range(1, nums[k+1]):
                great[k][x]+=1
        less=[0]*(n+1)
        for j in range(1,n):
            for x in range(nums[j-1]+1, n+1):
                less[x]+=1
            for k in range(j+1, n-1):
                if nums[k] < nums[j]:
                    ans+=less[nums[k]]*great[k][nums[j]]
        return ans