题意:n层楼,每层楼有m个房间,m个房间的左右两边分别是楼梯。每个房间0表示灯关着,1表示灯开着,走一格需要一分钟,问最少需要几分钟把所有的灯都关了。
分析:我们先预处理出每层楼最左边的灯在什么位置,最右边的灯在什么位置,然后dp
dp[i][0]表示走到第i层的左边楼梯处,且i层以下的等全灭的最少时间,dp[i][1]则是走到第i层右边楼梯处的答案。
我们找出等未关的最高的楼层,再上面还有楼层可以不用上去了。
对于每层楼,如果有没关的灯,
dp[i - 1][0] = min(dp[i][0]+ed[i],dp[i][1]+m+1-st[i])+1(+1是因为需要往上走一层);
dp[i - 1][1] = min(dp[i][0]+m+1,dp[i][1]+(m+1-st[i])*2)+1;
还有就是 这层楼没有没关的灯,直接往上走就行了,
dp[i - 1][0] = min(dp[i][0]+1,dp[i][1]+m+2);
dp[i - 1][1] = min(dp[i][1]+1,dp[i][0]+m+2);
再在最高一层记录下答案,因为不需要网上走了。
以下是代码:
int mp[20][200];
int dp[20][2];
int st[20], ed[20], maxx;
bool vis[20];
bool flag = 0, all = 0;
string s[20];
int main() {
maxx = INF; all = 0;
int n, m;
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; ++i) {
cin >> s[i];
}
for (int i = 1; i <= n; ++i) {
flag = 0;
for (int j = 0; j <= m + 1; ++j) {
mp[i][j] = s[i][j] - '0';
if (mp[i][j] == 1) {
if (!flag)st[i] = j, flag = 1;
ed[i] = j;
}
}
if (flag == 0)vis[i] = 0;
else maxx = min(i, maxx), vis[i] = 1, all = 1;
}
if (all == 0) {
printf("0\n");
return 0;
}
memset(dp, -1, sizeof(dp));
dp[n][0] = 0;
for (int i = n; i >= maxx+1; --i) {
if (vis[i] == 0) {
dp[i - 1][0] = dp[i][0] + 1;
if(dp[i][1]!=-1)dp[i - 1][1] = dp[i][1] + 1;
else dp[i][1] = dp[i - 1][0] + m + 1 + 1;
continue;
}
int pos1, pos2;
if (dp[i][0] != -1)pos1 = ed[i] + dp[i][0];
else pos1 = INF;
if (dp[i][1] != -1)pos2 = m + 1 - st[i] + dp[i][1];
else pos2 = INF;
dp[i - 1][0] = min(pos1 + ed[i], pos2 + st[i]) + 1;
dp[i - 1][1] = min(pos1 + m + 1 - ed[i], pos2 + m + 1 - st[i]) + 1;
}
int pos1, pos2;
if (dp[maxx][0] != -1)pos1 = ed[maxx] + dp[maxx][0];
else pos1 = INF;
if (dp[maxx][1] != -1)pos2 = m + 1 - st[maxx] + dp[maxx][1];
else pos2 = INF;
printf("%d\n", min(pos1, pos2));
}