还没学,等学完再仔细写。
cpp
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll a[1000010];
ll vis[1000010];
ll f[1000010][3];
int main()
{
ll n,m;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
}
for(int i=1;i<=n;i++)
{
cin>>vis[i];
}
for(int i=1;i<=n;i++)
{
if(vis[i]==1)
{
f[i][0]=max(f[i-1][0]+a[i-1],f[i-1][1]);
f[i][1]=max(f[i-1][0],f[i-1][1])+a[i];
f[i][2]=max(f[i-1][0],max(f[i-1][1],f[i-1][2]))+a[i+1];
}
if(vis[i]==0)
{
f[i][0]=max(f[i-1][0],f[i-1][1]);
f[i][1]=f[i-1][2];
//f[i][2]=
}
}
cout<<max(f[n][0],max(f[n][1],f[n][2]))<<endl;
return 0;
}