BZOJ1088: [SCOI2005]扫雷Mine

Description
相信大家都玩过扫雷的游戏。那是在一个n*m的矩阵里面有一些雷，要你根据一些信息找出雷来。万圣节到了，“余”人国流行起了一种简单的扫雷游戏，这个游戏规则和扫雷一样，如果某个格子没有雷，那么它里面的数字表示和它8连通的格子里面雷的数目。现在棋盘是n×2的，第一列里面某些格子是雷，而第二列没有雷，如下图：
由于第一列的雷可能有多种方案满足第二列的数的限制，你的任务即根据第二列的信息确定第一列雷有多少种摆放方案。
Input
第一行为N，第二行有N个数，依次为第二列的格子中的数。（1<= N <= 10000）
Output
一个数，即第一列中雷的摆放方案数。
Sample Input
2
1 1
Sample Output
2
HINT
Source

显然答案只能为0、1、2。因为只要第一行确定了，那其余各行就都确定了。所以简单枚举一下第一行是0还是1，看看有没有冲突即可。

#include<cstdio>
#include<string.h>
using namespace std;
int sum[10010],pre1,pre2=1,now=2;
int num[10010],ans,n;
void read(int &x)
{
    x=0;int f=1;char t=getchar();
    while((t<48)or(t>57)){if(t=='-')f=-1;t=getchar();}
    while((t>=48)and(t<=57)){x=(x*10)+t-48;t=getchar();}
    x*=f;
}
int main()
{
    read(n);
    for (int i=1;i<=n;++i) read(sum[i]);
    for (int op=0;op<=1;++op)
    {
        memset(num,0,sizeof(num));
        num[1]=op;bool pd=1;
        for (int i=2;i<=n;++i)
        {
            num[i]=sum[i-1]-(num[i-1]+num[i-2]);
            if ((num[now]>1)or(num[now]<0))
            {pd=0;break;}
        }
        if (num[n]+num[n-1]!=sum[n]) pd=0;
        if (pd) ++ans;
    }
    printf("%d",ans);
    return 0;
}