题目
不能多想系列,手玩几组数据简略证明出最终序列也只会存在\(0,1,-1\)三个数
做法
这样就好做了嘛
\(dp[i][c]\)表示前\(i\)个数,第\(i\)个数最终变成\(c\),然后枚举前一个数转移就好
My complete code
#include<bits/stdc++.h>
using namespace std;
typedef int LL;
const LL maxn=1e6+9,v=1,inf=0x3f3f3f3f;
LL n,ans;
LL a[maxn],dp[maxn][5];
int main(){
cin>>n;
for(LL i=1;i<=n;++i) cin>>a[i];
memset(dp,inf,sizeof(dp));
dp[1][a[1]+v]=0;
for(LL i=2;i<=n;++i){
if(a[i]==-1){
dp[i][-1+v]=dp[i-1][-1+v];
dp[i][1+v]=dp[i-1][1+v]+2;
}
if(a[i]==0){
dp[i][-1+v]=dp[i-1][-1+v]+1;
dp[i][0+v]=min(dp[i-1][0+v],dp[i-1][-1+v]);
dp[i][1+v]=dp[i-1][1+v]+1;
}
if(a[i]==1){
dp[i][-1+v]=dp[i-1][-1+v]+2;
dp[i][0+v]=dp[i-1][-1+v]+1;
dp[i][1+v]=min(dp[i-1][-1+v],min(dp[i-1][0+v],dp[i-1][1+v]));
}
}
ans=min(dp[n][-1+v],min(dp[n][0+v],dp[n][1+v]));
if(ans<inf)
cout<<ans;
else cout<<"BRAK";
return 0;
}