Leetcode周赛246 5790. 查询差绝对值的最小值

这场周赛是最有希望AK的一场,没有hard题,然而并没有成功。再接再厉!
传送门

题目描述

一个数组 a 的 差绝对值的最小值 定义为:0 <= i < j < a.length 且 a[i] != a[j] 的 |a[i] – a[j]| 的 最小值。如果 a 中所有元素都 相同 ,那么差绝对值的最小值为 -1 。

比方说,数组 [5,2,3,7,2] 差绝对值的最小值是 |2 – 3| = 1 。注意答案不为 0 ,因为 a[i] 和 a[j] 必须不相等。
给你一个整数数组 nums 和查询数组 queries ,其中 queries[i] = [li, ri] 。对于每个查询 i ,计算 子数组 nums[li…ri] 中 差绝对值的最小值 ,子数组 nums[li…ri] 包含 nums 数组(下标从 0 开始)中下标在 li 和 ri 之间的所有元素(包含 li 和 ri 在内)。

请你返回 ans 数组,其中 ans[i] 是第 i 个查询的答案。

子数组 是一个数组中连续的一段元素。

|x| 的值定义为:

如果 x >= 0 ,那么值为 x 。
如果 x < 0 ,那么值为 -x 。

样例

输入:nums = [1,3,4,8], queries = [[0,1],[1,2],[2,3],[0,3]]
输出:[2,1,4,1]
解释:查询结果如下:
– queries[0] = [0,1]:子数组是 [1,3] ,差绝对值的最小值为 |1-3| = 2 。
– queries[1] = [1,2]:子数组是 [3,4] ,差绝对值的最小值为 |3-4| = 1 。
– queries[2] = [2,3]:子数组是 [4,8] ,差绝对值的最小值为 |4-8| = 4 。
– queries[3] = [0,3]:子数组是 [1,3,4,8] ,差的绝对值的最小值为 |3-4| = 1 。
示例 2:

输入:nums = [4,5,2,2,7,10], queries = [[2,3],[0,2],[0,5],[3,5]]
输出:[-1,1,1,3]
解释:查询结果如下:
– queries[0] = [2,3]:子数组是 [2,2] ,差绝对值的最小值为 -1 ,因为所有元素相等。
– queries[1] = [0,2]:子数组是 [4,5,2] ,差绝对值的最小值为 |4-5| = 1 。
– queries[2] = [0,5]:子数组是 [4,5,2,2,7,10] ,差绝对值的最小值为 |4-5| = 1 。
– queries[3] = [3,5]:子数组是 [2,7,10] ,差绝对值的最小值为 |7-10| = 3 。

数据范围

2 <= nums.length <= 105
1 <= nums[i] <= 100
1 <= queries.length <= 2 * 104
0 <= li < ri < nums.length

思路



其实就是利用前缀和预处理后,可以在O(1)的时间复杂度内求出每个区间内1~100这100个数字是否存在。这样我们的时间复杂度刚刚好。

代码

const int N = 100010, M = 110;

int s[N][M];    //前缀和数组

class Solution {
public:
    vector<int> minDifference(vector<int>& nums, vector<vector<int>>& queries) {
        int n = nums.size(), m = queries.size();
        //预处理出前缀和
        for(int i = 1; i<=n; i++)
        {
            for(int j = 1; j<=100; j++)
            {
                int t = 0;
                if(nums[i-1] == j) t = 1;
                s[i][j] = s[i-1][j] + t;
            }
        }
        vector<int> ans;
        for(int i = 0; i<m; i++)
        {
            int l = queries[i][0]+1, r = queries[i][1]+1;   //前缀和下标从1开始
            int res = M, last = -1;
            for(int j = 1; j<=100; j++)
            {
                if(s[r][j] > s[l-1][j])
                {
                    if(last != -1) res = min(res,j-last);
                    last = j;
                }
            }
            if(res == M) ans.push_back(-1);
            else ans.push_back(res);
        }
        return ans;
    }
};

One Reply to “Leetcode周赛246 5790. 查询差绝对值的最小值”

发表回复

您的电子邮箱地址不会被公开。