题目描述
在一个叫做酱西功爷枝叶鸡树学院的地方有n只小动物,要么是鼠鼠,要么是鸭鸭,从1到n编号,每只小动物有个体重ai。
在这个学校里,存在一种神奇的魔法,可以将编号位于某个区间[l,r]内的所有鼠鼠都变为鸭鸭,鸭鸭都变为鼠鼠(魔法并不会改变体重)。
现在你可以施放这个魔法至多1次。(也可以不施放)
问最终鸭鸭的总重量最多是多少?
输入格式
第一行一个整数T表示样例个数。(1≤T≤10)
对于每个样例:
第一行一个整数n表示小动物的个数。(1≤n≤105)
第二行n个整数,表示第i个小动物的类型。0表示鼠鼠,1表示鸭鸭。
第三行n个整数,表示第i个小动物的体重ai。(1≤ai≤109)
输出格式
对于每个样例一行一个整数表示答案。
样例输入1
2
3
0 0 0
1 2 3
4
0 1 0 0
2 5 6 5
样例输出1
6
16
解释:第二个样例中,翻转区间[3,4][3,4],使得结果最大为5+6+5=165+6+5=16。
思路:计算原数组的偏移量之后求前缀和,可以快速得到一段区间内的和
示例代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+9; //N一定要大,输入样例的数组比较大
ll animal[N],weight[N],prefix[N],c[N];
void solve()
{
int n;
cin>>n;
ll ess=0;
for(int i=1;i<=n;i++)
{
cin>>animal[i];
}
for(int i=1;i<=n;i++)
{
cin>>weight[i];
}
for(int i=1;i<=n;i++)
{
c[i]=(animal[i] == 1 ? -1 : 1)*weight[i]; //计算出施加魔法后的偏移值(鸭子重量),如果原来是1(鸭子)就要减去,原来是0(鼠)就加上
}
for(int i=1;i<=n;i++)
{
prefix[i]=prefix[i-1]+c[i]; //偏移量的前缀和数组
}
for(int i=1;i<=n;i++)
{
ess+=animal[i]*weight[i]; //基础值(不进行修改)
}
ll mi=0,Fix=0; //mi表示min(prefix[j]),0<=j<i
for(int i=1;i<=n;i++) //从第一个遍历到最后,计算出最大最小值
{
Fix=max(Fix,prefix[i]-mi);
mi=min(mi,prefix[i]);
}
cout<<ess+Fix<<'\n';
}
int main()
{
ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
int a,n;
cin>>a;
while(a--)
{
solve();
}
return 0;
}
通过: