美团【避开炸弹】

大致题目

给n个房间，有m个炸弹，每次炸弹扎一个房间，小美先在第1号房间呆着，且保证第1个炸弹不会扎到1号房间。然后小美可以再爆炸的瞬间转移到另一个房间，但每次会消耗一个能量。请问小美最少消耗几个能量能安全避开所有炸弹。

能测试题目的同学帮我测试一下，我当时没做完，感谢感谢。

目前2个样例都过了。

思路：动态规划。

设dp[i][j]表示 “当前第i个炸弹要爆炸时，在第j个房间呆着所需要的消耗的最小能量”。

代码

#include <iostream>
#include <cstring>
#include <cstdlib>
using namespace std;
int n, m;
int boom[10005];
int dp[10005][15];
const int INF = 0x70000000;
int ans = INF;
int main() {
    // n: room,  m: bomb number
    cin >> n >> m;
    for (int i = 0; i < m; i++) {
        cin >> boom[i];
    }
    //dp初始状态
    for (int j = 1; j <= n; j++) {
        if (j == 1) dp[0][j] = 0;
        else dp[0][j] = INF;
    }
    for (int i = 1; i < m; i++) {   //boom
        //初始状态置为较大值
        for (int j = 1; j <= n; j++) {
            dp[i][j] = INF;
        }
        // dp[i][j]表示“第i个炸弹要爆炸时，在第j个房间所需要的能量值 */
        for (int j = 1; j <= n; j++) {   //room
            if (boom[i] == j) {     //如果第i个炸弹在第j个房间爆炸
                dp[i][j] = INF;
            } else {                //看是跳转房间消耗的能量小还是不跳转房间消耗小。
                for (int k = 1; k <= n; k++) {  //k表示上次在那个房间
                    if (k == j) {
                        dp[i][j] = min(dp[i][j], dp[i - 1][k]); //房间不变
                    } else {
                        dp[i][j] = min(dp[i][j], dp[i - 1][k] + 1);  //房间变了
                    }
                }
            }
            //当所有炸弹都爆炸完后看在那个房间跳转的收益最小。遍历一遍
            if (i == m - 1) {
                ans = min(ans, dp[i][j]);
            }
        }
    }
    cout << ans << endl;
    return 0;
}


/*
input1:
2 4
2 1 1 2
output1:
2

input1:
3 10
2 3 1 3 2 1 1 2 3 1
output2:
3
*/