LeetCode 891.子序列宽度之和数学每日一题

最新推荐文章于 2024-09-09 00:44:52 发布

@小红花

最新推荐文章于 2024-09-09 00:44:52 发布

阅读量164

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法蓝桥杯

本文链接：https://blog.csdn.net/lishifu_/article/details/124890874

版权

LeetCode 专栏收录该内容

84 篇文章 0 订阅

订阅专栏

给定一个整数数组，计算所有非空子序列的宽度之和，即最大元素与最小元素的差值。示例中，对于输入[2,1,3]，输出为6。算法通过排序数组，计算2的幂次并累加每个元素的贡献来求解，最后返回模10^9+7的结果。

摘要由CSDN通过智能技术生成

问题描述

一个序列的宽度定义为该序列中最大元素和最小元素的差值。

给你一个整数数组 nums ，返回 nums 的所有非空子序列的宽度之和。由于答案可能非常大，请返回对 109 + 7 取余后的结果。

子序列定义为从一个数组里删除一些（或者不删除）元素，但不改变剩下元素的顺序得到的数组。例如，[3,6,2,7] 就是数组 [0,3,1,6,2,2,7] 的一个子序列。

示例 1：

输入：nums = [2,1,3]
输出：6
解释：子序列为 [1], [2], [3], [2,1], [2,3], [1,3], [2,1,3] 。
相应的宽度是 0, 0, 0, 1, 1, 2, 2 。
宽度之和是 6 。
示例 2：

输入：nums = [2]
输出：0

提示：

1 <= nums.length <= 105
1 <= nums[i] <= 105

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/sum-of-subsequence-widths
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

Java 数学

class Solution {
    public int sumSubseqWidths(int[] nums) {
        int mod = (int)Math.pow(10,9) + 7;
        int N = nums.length;
        Arrays.sort(nums);

        long[] pow2 = new long[N];
        pow2[0] = 1;
        for(int i = 1;i < N;i++){
            pow2[i] = pow2[i - 1] * 2 % mod;
        }
        long ans = 0;
        for(int i = 0;i < N;i++){
            ans = (ans + (pow2[i] - pow2[N - i - 1]) * nums[i] % mod) % mod;
        }
        return (int)ans;
    }
}