鼠鼠我鸭（前缀和c++）

题目描述

在一个叫做酱西功爷枝叶鸡树学院的地方有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;
 } 

通过：