原题链接
思路一:
一图胜千言:
所以说要预处理出来对应的前缀和,后缀和!
代码一:附上注释!
class Solution {
public:
vector<int> leftRigthDifference(vector<int>& nums) {
int n = nums.size();
vector<int> ans(n);
vector<int> ls(n);
vector<int> rs(n);
//预处理求出前缀和
for (int i=0; i < n; i ++)
{
if (i==0) ls[i] = nums[i];
else ls[i] = ls[i-1] + nums[i];
}
//预处理求出后缀和
for (int i=n-1; i >= 0; i --)
{
if (i == n-1) rs[i] = nums[i];
else rs[i] = rs[i+1] + nums[i];
}
//构造ans数组:
for (int i=0; i < n; i ++)
ans[i] = abs(ls[i] - rs[i]);
return ans;
}
};
思路二:
- 初始化 L e f t S u m = 0 , R i g h t S u m = 元素总和 LeftSum=0,RightSum = 元素总和 LeftSum=0,RightSum=元素总和,然后从左往右依次求解;
- 每往右移动一格,意味着
L
e
f
t
S
u
m
LeftSum
LeftSum 向右新增一格,即累加当前格子元素值;
对于 R i g h t S u m RightSum RightSum 而言,每往右走一个,即丢弃当前左端点的格子,所以需要减去当前格子!
用 left_sum 保存数组的前缀和(从左往右依次累加),初始值为 0。
用 right_sum 保存数组的总和(从右往左依次减),初始值为数组的所有元素之和。
遍历数组 nums,每次迭代计算出当前元素的“左右差异”,即 abs(left_sum - right_sum),并将其保存回数组 nums 中。
更新 left_sum 和 right_sum,以继续迭代数组。
这个算法的时间复杂度是 O(N),其中 N 是输入数组的长度。因为该算法仅遍历数组两次:一次从左到右累加数组的前缀和,一次从右到左计算“左右差异”。
补充:accmulate
std::accumulate
是 C++ 中的一个标准库算法,它可以计算一个范围内(指定 by 起始和终止迭代器)元素的和或积。它是在头文件 numeric
中定义的。
函数签名:
template<class InputIt, class T>
T accumulate(InputIt first, InputIt last, T init);
其中,first
和 last
分别是要求和(或求积)的范围的开始和结束迭代器,init
是计算的初始值。
std::accumulate
函数的返回值类型与输入的初始值 init
的类型一致,例如如果 init
的类型是 int
,则结果也是 int
类型。
以下是使用 std::accumulate
函数计算一个整数数组和的示例:
#include <iostream>
#include <vector>
#include <numeric>
using namespace std;
int main() {
vector<int> nums = {1, 2, 3, 4, 5};
int sum = accumulate(nums.begin(), nums.end(), 0);
cout << "The sum of the vector is: " << sum << endl;
return 0;
}
在上面的示例中,我们将整数数组 [1, 2, 3, 4, 5]
的和计算出来,并将结果保存在变量 sum
中。函数的第三个参数 0
用来初始化 sum
的值。
std::accumulate
函数非常方便,它不仅仅适用于整数类型,也可以适用于其他数字类型,例如浮点数和复数。同时,它也支持求积运算,将第三个参数的初始值改为 1 即可。
代码二:
class Solution {
public:
vector<int> leftRigthDifference(vector<int>& nums) {
int n = nums.size();
int leftsum = 0, rightsum = accumulate(nums.begin(), nums.end(), 0);
vector<int> ans(n);
for (int i=0; i < n; i ++)
{
rightsum -= nums[i];
ans[i] = abs(leftsum - rightsum);
leftsum += nums[i];
}
return ans;
}
};