Sagheer, the Hausmeister

Sagheer, the Hausmeister

题意 :

有一个n行m+2列的01矩阵，最初在左下角的位置，0表示灯关，1表示灯开，问你关闭所有的灯所需要的时间，最后不需要回到原位置。

分析：

考虑dp，dp[i][0]表示关完i层从左边楼梯走，dp[i][1]表示从右边走。

代码：

#include<bits/stdc++.h>
using namespace std;
char s[20][110];
int dp[110][2];
int l[30],r[30];
int main(){
	int n,m;
	memset(l,0,sizeof(l));
	memset(r,0,sizeof(r));
	scanf("%d%d",&n,&m);
	for(int i=n-1;i>=0;i--){
		scanf("%s",s[i]);
		for(int j=0;j<m+2;j++){
			if(s[i][j]=='1'&&l[i]==0)
				l[i]=j;
			if(s[i][j]=='1')
				r[i]=j;
		}
	}
	for(int i=n-1;i>=0;i--)
		if(l[i]==0&&r[i]==0)
			n--;
		else break;
	if(n==0){
		puts("0");
		return 0;
	}
	if(n==1){
		printf("%d\n",r[0]);
		return 0;
	}
	dp[0][0]=2*r[0]+1;
	dp[0][1]=m+2;
	for(int i=1;i<n-1;i++){
		if(l[i]==0&&r[i]==0){
			dp[i][0]=dp[i-1][0]+1;
			dp[i][1]=dp[i-1][1]+1;
		}
		else{
			dp[i][0]=min(dp[i-1][0]+2*r[i],dp[i-1][1]+m+1)+1;
			dp[i][1]=min(dp[i-1][0]+m+1,dp[i-1][1]+2*(m-l[i]+1))+1;
		}
	}
	printf("%d\n",min(dp[n-2][0]+r[n-1],dp[n-2][1]+(m+1-l[n-1])));
	return 0;
}