暴搜每一个点是否是雷 优化:每次搜之前判断一下可行性
#include <cstdio> #include <algorithm> using namespace std; const int N=10010; int a[N],mine[N],n,cnt; int check(int x){//计算这个位置现在已经有多少颗雷了 int tmp=0; for(int i=x-1;i<=x+1;i++) tmp+=mine[i]; return tmp; } void dfs(int x){ if(x>n) {if(check(x-1)==a[x-1]) cnt++;return;} for(int i=0;i<=1;i++){ mine[x]=i; if(x==1 || (check(x-1)==a[x-1] && check(x)<=a[x])) dfs(x+1); mine[x]=0; } } int main(){ //freopen("1088.in","r",stdin);freopen("1088.out","w",stdout); scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&a[i]); dfs(1); printf("%d",cnt); return 0; }