题目大意
有一个数组,由标记为R或B的数字组成,现在这个数组丢了。但是给你了两组数组R跟B,顺序为他们在原数组的出现顺序,现在让你把他们排回去,找出一种排序方式使得f(a)=max(0,a1,(a1+a2),(a1+a2+a3),…,(a1+a2+a3+⋯+an+m))最大。
观察后可以发现各数组的顺序是独立的,不会互相影响,所以a的最大值就是r的最大值加上b的最大值。因此我们分别遍历这两个数组,使用sum来记录和,max来更新最大值,之后把两组max相加即可。
这里有个坑点就是如果数组里的值都是负的或者sum一直<0,那么max就应该是0,所以我们max初始值为0,而不是-99999或者是数组R或B的第一个元素值。
例子
a 2 4 -3 4
b -1 2 7 -5
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
void solve(){
int n,m;
cin>>n;
vector<int> r(n);
for(int &x:r) cin>>x;
cin>>m;
vector<int> b(m);
for(int &x:b) cin>>x;
partial_sum(r.begin(),r.end(),r.begin());
partial_sum(b.begin(),b.end(),b.begin());
int A=max(0,*max_element(r.begin(),r.end()));
int B=max(0,*max_element(b.begin(),b.end()));
cout<<(A+B)<<endl;
}
int main(){
ios::sync_with_stdio(false);
int t;
cin>>t;
while(t--) solve();
return 0;
}