BZOJ 1088 扫雷Mine 枚举初始状态

题目链接：

https://www.lydsy.com/JudgeOnline/problem.php?id=1088

题目大意：

现在棋盘是n×2的，第一列里面某些格子是雷，而第二列没有雷，如下图： 由于第一列的雷可能有多种方案满足第二列的数的限制，你的任务即根据第二列的信息确定第一列雷有多少种摆放方案。

思路：

只需要枚举第一行是否有雷就ok了。

判断的时候除了判断相等，还要判断每a个f[i]范围必须在0-1之间

#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(false);//不可再使用scanf printf
#define Max(a, b) ((a) > (b) ? (a) : (b))//禁用于函数，会超时
#define Min(a, b) ((a) < (b) ? (a) : (b))
#define Mem(a) memset(a, 0, sizeof(a))
#define Dis(x, y, x1, y1) ((x - x1) * (x - x1) + (y - y1) * (y - y1))
#define MID(l, r) ((l) + ((r) - (l)) / 2)
#define lson ((o)<<1)
#define rson ((o)<<1|1)
#define Accepted 0
#pragma comment(linker, "/STACK:102400000,102400000")//栈外挂
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

typedef long long ll;
const int maxn = 10000 + 10;
const int MOD = 1000000007;//const引用更快，宏定义也更快
const int INF = 1e9 + 7;
const double eps = 1e-6;
int a[maxn];
int f[maxn];//f[i]表示i是否有雷
int n;
bool judge()
{
    for(int i = 2; i <= n; i++)f[i] = a[i - 1] - f[i - 1] - f[i - 2];
    for(int i = 1; i <= n; i++)
    {
        if(a[i] != f[i - 1] + f[i] + f[i + 1] || f[i] < 0 || f[i] > 2)return false;
    }
    return true;
}
int main()
{
    cin >> n;
    for(int i = 1; i <= n; i++)scanf("%d", &a[i]);
    int ans = 0;
    f[1] = 0;//枚举第一行
    if(judge())ans++;
    Mem(f);
    f[1] = 1;
    if(judge())ans++;
    cout<<ans<<endl;
    return Accepted;
}