【C++前缀和排序】891. 子序列宽度之和|2182

本文涉及的基础知识点

C++算法：前缀和、前缀乘积、前缀异或的原理、源码及测试用例包括课程视频排序

LeetCode891. 子序列宽度之和

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

给你一个整数数组 nums ，返回 nums 的所有非空子序列的宽度之和。由于答案可能非常大，请返回对 10^9^ + 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 <= 10^5^

1 <= nums[i] <= 10^5^

排序+前缀和

对nums排序，长度为1的子序列宽带是0，故无需枚举。通过i枚举最大元素，如果有多个最大元素，以原始下标大的为准；通过j枚举最小元素，如果有多个以原始下标小的为准。排序时间复杂度：O(nlogn)，枚举时间复杂度：O(nn)。

利用前缀和优化：

cnt(i) = KaTeX parse error: Expected group after '^' at position 42: ...ums[k])*2^i-k-1^̲=KaTeX parse error: Expected group after '^' at position 24: ...0}^{i-1}2^i-k-1^̲*nums[i] - ∑ k = 0 i − 1 2 i − k − 1 n u m s [ k ] \sum_{k=0}^{i-1}2^i-k-1^nums[k] ∑k=0i−12i−k−1nums[k]

= f(i-1)nums[i]-g(i-1)
f(i) =2^0^+2^1^+ ⋯ \cdots ⋯ + 2^i^ f(0)=1 f(i)=2 f(i-1)+1

g(0)=nums[0] g(i) = g(i-1)*2+ nums[i]
时间复杂度：O(n)

代码

核心代码

cpp 复制代码

template<int MOD = 1000000007>
class C1097Int
{
public:
	C1097Int(long long llData = 0) :m_iData(llData% MOD)
	{

	}
	C1097Int  operator+(const C1097Int& o)const
	{
		return C1097Int(((long long)m_iData + o.m_iData) % MOD);
	}
	C1097Int& operator+=(const C1097Int& o)
	{
		m_iData = ((long long)m_iData + o.m_iData) % MOD;
		return *this;
	}
	C1097Int& operator-=(const C1097Int& o)
	{
		m_iData = (m_iData + MOD - o.m_iData) % MOD;
		return *this;
	}
	C1097Int  operator-(const C1097Int& o)
	{
		return C1097Int((m_iData + MOD - o.m_iData) % MOD);
	}
	C1097Int  operator*(const C1097Int& o)const
	{
		return((long long)m_iData * o.m_iData) % MOD;
	}
	C1097Int& operator*=(const C1097Int& o)
	{
		m_iData = ((long long)m_iData * o.m_iData) % MOD;
		return *this;
	}
	C1097Int  operator/(const C1097Int& o)const
	{
		return *this * o.PowNegative1();
	}
	C1097Int& operator/=(const C1097Int& o)
	{
		*this /= o.PowNegative1();
		return *this;
	}
	bool operator==(const C1097Int& o)const
	{
		return m_iData == o.m_iData;
	}
	bool operator<(const C1097Int& o)const
	{
		return m_iData < o.m_iData;
	}
	C1097Int pow(long long n)const
	{
		C1097Int iRet = 1, iCur = *this;
		while (n)
		{
			if (n & 1)
			{
				iRet *= iCur;
			}
			iCur *= iCur;
			n >>= 1;
		}
		return iRet;
	}
	C1097Int PowNegative1()const
	{
		return pow(MOD - 2);
	}
	int ToInt()const
	{
		return m_iData;
	}
private:
	int m_iData = 0;;
};

class Solution {
		public:
			int sumSubseqWidths(vector<int>& nums) {
				sort(nums.begin(), nums.end());
				C1097Int<> ans, f, g;
				for (const auto& n : nums) {
					ans += f * n - g;
					f = f * 2 + 1;
					g = g * 2 + n;
				}
				return ans.ToInt();
			}
		};

单元测试

cpp 复制代码

		vector<int> nums;
		TEST_METHOD(TestMethod11)
		{
			nums = { 2,1,3 };
			auto res = Solution().sumSubseqWidths(nums);
			AssertEx(6, res);
		}
		TEST_METHOD(TestMethod12)
		{
			nums = {2 };
			auto res = Solution().sumSubseqWidths(nums);
			AssertEx(0, res);
		}

扩展阅读

我想对大家说的话
工作中遇到的问题，可以按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
学习算法：按章节学习《喜缺全书算法册》，大量的题目和测试用例，打包下载。重视操作
有效学习：明确的目标及时的反馈拉伸区（难度合适）专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛
失败+反思=成功成功+反思=成功

视频课程

先学简单的课程，请移步CSDN学院，听白银讲师（也就是鄙人）的讲解。
https://edu.csdn.net/course/detail/38771

如何你想快速形成战斗了，为老板分忧，请学习C#入职培训、C++入职培训等课程
https://edu.csdn.net/lecturer/6176

测试环境

操作系统：win7 开发环境： VS2019 C++17

或者操作系统：win10 开发环境： VS2022 C++17

如无特殊说明，本算法用**C++**实现。

【C++前缀和 排序】891. 子序列宽度之和|2182