1273 天才的记忆(RMQ-区间最值)

1. 问题描述:

从前有个人名叫 WNB,他有着天才般的记忆力,他珍藏了许多许多的宝藏。在他离世之后留给后人一个难题(专门考验记忆力的啊!),如果谁能轻松回答出这个问题,便可以继承他的宝藏。题目是这样的:给你一大串数字(编号为 1 到 N,大小可不一定哦!),在你看过一遍之后,它便消失在你面前,随后问题就出现了,给你 M 个询问,每次询问就给你两个数字 A,B,要求你瞬间就说出属于 A 到 B 这段区间内的最大数。一天,一位美丽的姐姐从天上飞过,看到这个问题,感到很有意思(主要是据说那个宝藏里面藏着一种美容水,喝了可以让这美丽的姐姐更加迷人),于是她就竭尽全力想解决这个问题。但是,她每次都以失败告终,因为这数字的个数是在太多了!于是她请天才的你帮他解决。如果你帮她解决了这个问题,可是会得到很多甜头的哦!

输入格式

第一行一个整数 N 表示数字的个数。接下来一行为 N 个数,表示数字序列。第三行读入一个 M,表示你看完那串数后需要被提问的次数。接下来 M 行,每行都有两个整数 A,B。

输出格式

输出共 M 行,每行输出一个数,表示对一个问题的回答。

数据范围

1 ≤ N ≤ 2 × 10 ^ 5,
1 ≤ M ≤ 10 ^ 4,
1 ≤ A ≤ B ≤ N。

输入样例:

6
34 1 8 123 3 2
4
1 2
1 5
3 4
2 3

输出样例:

34
123
123
8
来源:https://www.acwing.com/problem/content/1275/

2. 思路分析:

分析题目可以知道是区间最值问题,可以使用线段树或者RMQ来解决,由于在查询的过程中不需要修改,所以我们可以使用RMQ算法进行求解(如果在查询的过程中需要修改那么则需要使用线段树来求解),时间复杂度为O(nlogn),代码比较容易编写,RMQ算法分为两个步骤进行处理,第一个步骤是预处理的步骤,预处理本质上是一个递推的过程,需要声明一个二维数组dp,其中dp[i][j]表示从下标为i开始的长度为2 ^ j的区间最值,根据当前区间[i,j]的分割点2 ^ (j - 1)来求解最值,也即max(dp[i: 2 ^ (j - 1) - 1],dp[i + 2 ^ (j - 1)]: 2 ^ j - 1),这样就可以使用两层循环递推出当前的dp[i][j],第二个是查询区间最值得步骤,在查询区间最值的时候也是分段处理的思想,若2 ^ k <= l,则2 ^ (k + 1) > l,我们可以使用log函数计算得到当前长度为l的k,当前区间为[i: j],那么以i开始长度为2 ^ k的区间和以j为右端点的长度为2 ^ k的区间那么一定可以覆盖当前长度为2 ^ k的区间那么就可以求解出当前的[i: j]的最值,主要是预处理的时间复杂度,有两层循环为O(nlogn),查询的时间复杂度为O(1)。

3. 代码如下:

import math
from typing import List


class Solution:
    # 预处理
    def init(self, n: int, nums: List[int]):
        dp = [[0] * 18 for i in range(n + 10)]
        for j in range(n + 1):
            i = 0
            while i + (1 << j) - 1 < n:
                if j == 0:
                    dp[i][j] = nums[i]
                else:
                    dp[i][j] = max(dp[i][j - 1], dp[i + (1 << j - 1)][j - 1])
                i += 1
        return dp
    
    # 查询区间[a:b]的最大值
    def get(self, a: int, b: int, dp: List[List[int]]):
        _len = b - a + 1
        k = int(math.log(_len, 2))
        return max(dp[a][k], dp[b - (1 << k) + 1][k])

    def process(self):
        n = int(input())
        nums = list(map(int, input().split()))
        m = int(input())
        dp = self.init(n, nums)
        for i in range(m):
            a, b = map(int, input().split())
            # 预处理的时候下标是从0开始的, 输入的是从1开始的所以先将其转化为0开始
            a -= 1
            b -= 1
            print(self.get(a, b, dp))


if __name__ == '__main__':
    Solution().process()
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值