题目大意:给一个只包含-1,0,1的数列,每次操作可以让a[i]+=a[i-1],求最少操作次数使得序列单调不降
一看就是个DP,不妨用f[i][0..1..2]分别表示把第i位变成-1,0,1且保证前i个单调不降的最少次数
然后分各种情况讨论转移一下就行了
具体如下:
想让第i位变成-1,那就可以让第i-1位变成-1然后在对第i位进行操作直到变成-1
想让第i位变成0,那就要根据第i位原来是什么分三种情况讨论
假设原来是-1,那就只有在原来第i-1位是1的情况下,先让他加回来,否则解一定不是最优的
假设原来是0,那就不用改
假设原来是1,那就由i-1原来是-1或最后是-1的情况转移而来
变成1的同理,具体可以看代码
#include<iostream>
#include<cstdio>
#define N 1000010
using namespace std;
int a[N];
int f[N][3];
int main()
{
int n;
int i,j,x,y;
scanf("%d",&n);
for(i=1;i<=n;i++)
scanf("%d",&a[i]);
f[1][0]=f[1][1]=f[1][2]=707185547;
if(a[1]==-1) f[1][0]=0;
else if(a[1]==0) f[1][1]=0;
else f[1][2]=0;
for(i=2;i<=n;i++)
{
f[i][0]=f[i-1][0]+a[i]+1;
if(a[i]==-1)
{
if(a[i-1]==1) f[i][1]=min(f[i-1][0],f[i-1][1])+1;
else f[i][1]=707185547;
}
else if(a[i]==0) f[i][1]=min(f[i-1][0],f[i-1][1]);
else
{
if(a[i-1]==-1) f[i][1]=min(f[i-1][0],f[i-1][1])+1;
else f[i][1]=f[i-1][0]+1;
}
if(a[i]==-1)
{
if(a[i-1]==1) f[i][2]=min(min(f[i-1][0],f[i-1][1]),f[i-1][2])+2;
else f[i][2]=f[i-1][2]+2;
}
else if(a[i]==0)
{
if(a[i-1]==1) f[i][2]=min(min(f[i-1][0],f[i-1][1]),f[i-1][2])+1;
else f[i][2]=f[i-1][2]+1;
}
else f[i][2]=min(min(f[i-1][0],f[i-1][1]),f[i-1][2]);
}
int ans=min(min(f[n][0],f[n][1]),f[n][2]);
if(ans>=707185547) puts("BRAK");
else printf("%d",ans);
}