sgu239 分类： sgu 2015-06-24 22:18 22人阅读 评论(0) 收藏

---

如果知道第一个方格是否有雷，可以推出其他所有方格是否有雷：

记 ai为位置 i上雷的数量，bi为位置 i上的数字
bi=ai−1+ai+ai+1⇒ai=bi−1−ai−1−ai−2

---

只要枚举 a1的值，判断是否可行即可。
时间复杂度：O(n)

---

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>


const int maxn = 1005;
int n, b[maxn];
int ans, a[maxn];

int check(int x)
{
    a[1] = x;

    for(int i = 2; i <= n+1; i++)
    {
        a[i] = b[i-1] - a[i-1] - a[i-2];

        if(!(a[i] == 0 || a[i] == 1))
            return 0;
    }
    if(a[n+1]) return 0;

    return 1;
}

int main()
{
#ifndef ONLINE_JUDGE    
    freopen("sgu239.in","r",stdin);
    freopen("sgu239.out","w",stdout);
#endif

    std::cin >> n;

    for(int i = 1; i <= n; i++)
        std::cin >> b[i];

    ans = check(0) + check(1);

    std::cout << ans;

#ifndef ONLINE_JUDGE
    fclose(stdin);
    fclose(stdout);
#endif
    return 0;       
}

版权声明：本文为博主原创文章，未经博主允许不得转载。