【西交OJ】Problem A -- 扫雷

【西交OJ】Problem A -- 扫雷

http://202.117.21.117/xjoj/problem_html/105.html

Description

相信大家都玩过扫雷的游戏。那是在一个n * m的矩阵里面有一些雷，要你根据一些信息找出雷来。万圣节到了，"余"人国流行起了一种简单的扫雷游戏，这个游戏规则和扫雷一样，如果某个格子没有雷，那么它里面的数字表示和它8连通的格子里面雷的数目。现在棋盘是n * 2的，第一列里面某些格子是雷，而第二列没有雷，如下图：

由于第一列的雷可能有多种方案满足第二列的数的限制，你的任务即根据第二列的信息确定第一列雷有多少种摆放方案。

Input

第一行为N（N <= 10000），第二行有N个数，依次为第二列的格子中的数。

Output

一个数，即第一列中雷的摆放方案数。

Sample Input

2
1 1

Sample Output

2

对于两竖的这种情况，其实这要前面两个格子确定了，后面的格子就唯一确定了。

    所以说，只要我们对前面两个格子进行枚举就行了。

  a[1]=2，那么前面两个格子必然就是都有雷。

  a[1]=3    前面只有两个格子，这是不可能的，直接输出0

  a[1]=1   要么第一个格子是雷，要么第二个是雷。

  a[1]=0   两个都不是雷

import java.io.BufferedInputStream;
import java.io.IOException;
import java.util.Arrays;
import java.util.Scanner;

//author:pz

public class Main {
    private static final int MAX_N = 10005;
    static int ans = 0;

    public static void main(String[] args) throws IOException {
        Scanner in = new Scanner(new BufferedInputStream(System.in));
        final int n = in.nextInt();
        int[] a = new int[MAX_N];
        for (int i = 1; i <= n; i++) {
            a[i] = in.nextInt();
        }
        work(a, n);
        System.out.print(ans);
    }

    private static void work(int[] a, int n) {
        int[] f = new int[MAX_N];
        if (a[1] == 3) {
            return;
        }
        if (a[1] == 2) {
            f[1] = 1;
            f[2] = 1;
            dp(a, f, n);
        }
        if (a[1] == 1) {
            f[1] = 1;
            dp(a, f, n);
            Arrays.fill(f, 0);
            f[2] = 1;
            dp(a, f, n);
        }
        if (a[1] == 0)
            dp(a, f, n);
        return;
    }

    private static void dp(int[] a, int[] f, int n) {
        for (int i = 3; i <= n; i++)
            f[i] = a[i - 1] - f[i - 1] - f[i - 2];
        if (f[n] + f[n - 1] == a[n])
            ans++;
    }
}