第一反应是枚举兔子起点+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;
}