2018深信服笔试-抓兔子 DP

第一反应是枚举兔子起点+DFS,不过同学说过不了。

dp的思路是:第i天第j个洞是否可能存在兔子,当天被访问的洞标位0,其余能到的洞为1,最后一天如果全是0的就一定能捉到

dp[i][j]==1 ==> dp[i+1][j-1]=1 dp[i+1][j+1]=1(需特判边界)

且可以优化至一维滚动数组(懒,用的二维)

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int n,k;
int dp[1002][1001];
int visit[1001];
int main(){
    cin>>n>>k;
    for(int i=0;i<k;i++){
        cin>>visit[i];
    }
    memset(dp,0,sizeof(dp));
    for(int i=0;i<n;i++){
        dp[0][i]=1;
    }
    dp[0][visit[0]-1]=0;
    for(int i=0;i<k;i++){
        for(int j=0;j<n;j++){
            if(dp[i][j]==1){
                if(j==0){
                    dp[i+1][j+1]=1;
                }
                else if(j==n-1){
                    dp[i+1][j-1]=1;
                }
                else{
                    dp[i+1][j-1]=1;
                    dp[i+1][j+1]=1;
                }
            }
            dp[i+1][visit[i+1]-1]=0;
        }
    }
    int flag=1;
    for(int i=0;i<n;i++){
        if(dp[k][i]==1){
            flag=0;
            break;
        }
    }
    if(flag){
        puts("yes");
    }
    else{
        puts("no");
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/q295657451/article/details/80803178
DP
DP?