狡兔三窟问题(深信服提前批笔试第一题)

题意大概时这个样子的
狡兔三窟问题(深信服提前批笔试第一题)_第1张图片
因为是第一次笔试,太紧张了,实力也不够,一时没想起来,挺可惜的。当时感觉是枚举+dfs,但是没过,后来复盘的时候,惊觉dp+滚动数组完美解决。

思路:可以呆的洞为1,不可以的为0,先全部初始化为1,横坐标代表洞的标号,纵坐标代表第几天。第j天被检查的洞一定是0;如果是1号洞或是m号洞,则查看前一天的二号洞或m-1号洞是0的话,1号洞或m号洞为0;如果前一天的i-1和i+1号洞都是0的话,第i号洞为0。
填表完成之后,遍历数组的最后一行,如果有1的话,代表兔子有可能不被抓住,如果全为0,则代表一定会被抓住。
代码:

#include 
#include 
#include 

using namespace std;

int main()
{
    int m, n;
    int a[1001];
    int dp[2][1001];
    //memset(dp, 1, sizeof(dp));
    //初始化记录数组
    for (int i = 0; i < 2; i++) {
        for(int j = 0; j < 1001; j++) {
            dp[i][j] = 1;
        }
    }


    scanf("%d %d", &m, &n); //m是洞的编号,n是天数
    //第i+1天检查的洞的编号
    for (int i = 0; i < n; i++)
        scanf("%d", &a[i]);
    //填表
    for (int i = 0; i < n; i++) {
        for (int j = 1; j <= m; j++) {
            if (j == a[i])
                dp[i%2][j] = 0;
            if (dp[(i-1)%2][2] == 0 && i > 0)
                dp[i%2][1] = 0;
            if (dp[(i-1)%2][m-1] == 0 && i > 0)
                dp[i%2][m] = 0;
            if (dp[(i-1)%2][j-1] == 0 && dp[(i-1)%2][j+1] == 0 && i > 0)
                dp[i%2][j] = 0;
        }
    }

    int index = 0;
    for (int j = 1; j <= m; j++) {
        if (dp[(n-1)%2][j] == 1) {
            index = 1;
            break;
        }
    }

    if (index == 1)
        printf("No\n");
    else
        printf("Yes\n");

    return 0;
}

虽然没法再提交了,不过测试的几个用例都是可以的了。

你可能感兴趣的:(数据结构与算法)