codeforces 812B. Sagheer, the Hausmeister

题目链接：http://codeforces.com/contest/812/problem/B

题意：n层楼，每层有m个房间，左右两边是楼梯，输入一个n*（m+2）的字符串，字符1代表这个房间灯亮。要想关闭上层的等，低层的灯必须全部关闭。现在问你关闭所有的灯最少移动多少次。

分析：动态规划或者dfs都可以。dp[i][0]代表到达第i层的左边楼梯最短移动，dp[i][1]代表到达第i层的右边楼梯的最短移动。初始化dp[1][0]=0,dp[1][1]=m+1;

　　　　dp[i][0]=min(dp[i-1][0]+r[i-1]*2,dp[i-1][1]+m+1)+1;

　　　　dp[i][1]=min(dp[i-1][1]+(m+1-l[i-1])*2,dp[i-1][0]+m+1)+1;

　　　　然后dp递推到需要关灯的最高层即可（如果那一层以上再没有需要关的灯即为最高层）。最高层的移动不需要再回来，只需要关灯就行。

AC代码：

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 
 4 char a[20][105];
 5 int l[20],r[20];
 6 int dp[20][2];
 7 int main() {
 8     ios_base::sync_with_stdio(0);
 9     cin.tie(0);
10     int n,m;
11     memset(a,'0',sizeof(a));
12     memset(l,0,sizeof(l));
13     memset(r,0,sizeof(r));
14     memset(dp,0,sizeof(0));
15     cin>>n>>m;
16     for(int i=n;i>=1;i--){
17         for(int j=0;j<=m+1;j++){
18             cin>>a[i][j];
19             if(a[i][j]=='1'){
20                 r[i]=j;
21                 if(l[i]==0){
22                     l[i]=j;
23                 }
24             }
25         }
26     }
27     int high=n;
28     for(int i=n;i>0;i--){
29         if(l[i]==0&&r[i]==0){
30             high--;
31         }
32         else break;
33     }
34     dp[1][0]=0;
35     dp[1][1]=m+1;
36     for(int i=2;i<=high;i++){
37         if(l[i-1]==0&&r[i-1]==0){
38             dp[i][0]=min(dp[i-1][0],dp[i-1][1]+m+1)+1;
39             dp[i][1]=min(dp[i-1][1],dp[i-1][0]+m+1)+1;
40         }
41         else {
42             dp[i][0]=min(dp[i-1][0]+r[i-1]*2,dp[i-1][1]+m+1)+1;
43             dp[i][1]=min(dp[i-1][1]+(m+1-l[i-1])*2,dp[i-1][0]+m+1)+1;
44         }
45     }
46     int result;
47     result=min(dp[high][0]+r[high],dp[high][1]+(m+1-l[high]));
48     cout<<result<<endl;
49 
50 
51 return 0;
52 }

View Code

 

转载于:https://www.cnblogs.com/ls961006/p/6931793.html