leetcode#53 最大子序列和
题目:
给定一个整数数组 nums ,找到一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。
示例:
输入:nums = [-2,1,-3,4,-1,2,1,-5,4]
输出:6
解释:连续子数组 [4,-1,2,1] 的和最大,为 6 。
思路:
1、动态规划:
当前这个数,要不这个区间从当前这个数重新开始,要么这个数继续算在前面的区间里面,取两者最大值。
代码:
class Solution {
public:
int maxSubArray(vector<int>& nums) {
int siz=nums.size(),ans=nums[0];
for(int i=1;i<siz;++i)
{
nums[i]=max(nums[i],nums[i]+nums[i-1]);
ans=max(ans,nums[i]);
}
return ans;
}
};
2、线段树:
因为只需要查询操作,而且只查询一次,不需要保存线段树结构,所以直接写query就可以了。
查询维护4个值:
lsum :表示 [l,r] 内以 l 为左端点的最大子段和。
rsum :表示 [l,r] 内以 r 为左端点的最大子段和。
msum:表示 [l,r] 内的最大子段和。
sum:表示[l,r]的区间和。
lsum:要么是左子区间的lsum,要么是左子区间的sum+右子区间的lsum,两者取最大值。
rsum:要么是右子区间的rsum,要么是右子区间的sum+左子区间的rsum,两者取最大值。
msum:要么左右子区间的msum取最大,要么左子区间rsum+右子区间lsum。两者取最大值。
sum:左右子区间和。
代码:
class Solution
{
public:
struct node
{
int lsum, rsum, msum, sum;
};
node push_up(node l, node r)
{
int sum = l.sum + r.sum;
int lsum = max(l.lsum, l.sum + r.lsum);
int rsum = max(r.rsum,r.sum + l.rsum);
int msum = max(l.msum, max(r.msum, l.rsum + r.lsum));
return (node){lsum, rsum, msum, sum};
}
node get(vector<int> &nums, int l, int r)
{
if (l == r)
return (node){nums[l], nums[l], nums[l], nums[l]};
int mid = (l + r) >> 1;
node left = get(nums, l, mid);
node right = get(nums, mid + 1, r);
return push_up(left, right);
}
int maxSubArray(vector<int> &nums)
{
return get(nums, 0, nums.size() - 1).msum;
}
};