1113: 数列|NFOJ-NEW

#include<bits/stdc++.h>
using namespace std;
int n,a[1111111];
int dp[1111111][3],inf,f1,flag;
int main() {
	cin>>n;
	cin>>a[1];
	for(int i=2;i<=n;i++) {
		cin>>a[i];
	}
	memset(dp,125,sizeof(dp));
	inf=dp[0][0];
	dp[1][a[1]+1]=0;
	for(int i=2;i<=n;i++) {
		if(dp[i-1][0]!=inf) {
			if(a[i]==-1) {
				dp[i][0]=min(dp[i][0],dp[i-1][0]);
			}
			if(a[i]==0) {
				dp[i][1]=min(dp[i][1],dp[i-1][0]);
				dp[i][2]=min(dp[i][2],dp[i-1][0]);
				dp[i][0]=min(dp[i][0],dp[i-1][0]+1);
			}
			if(a[i]==1) {
				dp[i][0]=min(dp[i][0],dp[i-1][0]+2);
				dp[i][1]=min(dp[i][1],dp[i-1][0]+1);
				dp[i][2]=min(dp[i][2],dp[i-1][0]);
			}
		}
		if(dp[i-1][1]!=inf) {
			if(a[i]==0) {
				dp[i][1]=min(dp[i][1],dp[i-1][1]);
				dp[i][2]=min(dp[i][2],dp[i-1][1]+1);
			}
			if(a[i]==1) {
				dp[i][2]=min(dp[i][2],dp[i-1][1]);
			}
		}
		if(dp[i-1][2]!=inf) {
			if(a[i]==-1) {
				dp[i][2]=min(dp[i][2],dp[i-1][2]+2);
			}
			if(a[i]==0) {
				dp[i][2]=min(dp[i][2],dp[i-1][2]+1);
			}
			if(a[i]==1) {
				dp[i][2]=min(dp[i][2],dp[i-1][2]);
			}
		}
	}
	for(int i=1;i<=n;i++) {
		for(int j=0;j<=2;j++) {
			cout<<dp[i][j]<<" ";
		}
		cout<<"\n";
		
	}
	if(min(dp[n][0],min(dp[n][1],dp[n][2]))<1000000) {
		cout<<min(dp[n][0],min(dp[n][1],dp[n][2]));
	}
	else {
		cout<<"BRAK";
	}
	return 0;
}