P2327 [SCOI2005]扫雷

题目描述

输入输出格式

输入格式：

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

输出格式：

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

输入输出样例

输入样例#1： 复制

2
1  1

输出样例#1： 复制

2

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define inf 2147483647
const ll INF = 0x3f3f3f3f3f3f3f3fll;
#define ri register int
template <class T> inline T min(T a, T b, T c)
{
    return min(min(a, b), c);
}
template <class T> inline T max(T a, T b, T c)
{
    return max(max(a, b), c);
}
template <class T> inline T min(T a, T b, T c, T d)
{
    return min(min(a, b), min(c, d));
}
template <class T> inline T max(T a, T b, T c, T d)
{
    return max(max(a, b), max(c, d));
}
#define pi acos(-1)
#define me(x, y) memset(x, y, sizeof(x));
#define For(i, a, b) for (int i = a; i <= b; i++)
#define FFor(i, a, b) for (int i = a; i >= b; i--)
#define mp make_pair
#define pb push_back
const int maxn = ;
#define mod 100003
const int N=;

// name*******************************
int f[N][];
int n;
int num[];
int ans=;
// function******************************
int counter(int x)
{
    int res=;
    while(x)
    {
        x&=(x-);
        res++;
    }
    return res;
}

//***************************************
int main()
{
//freopen("test.txt", "r", stdin);
    cin>>n;
    For(i,,n)
    cin>>num[i];
    f[][]=;
    f[][]=;
    For(i,,n)
    {
        For(j,,)
        {
            if(counter(j)!=num[i-]||counter(j>>)>num[i])continue;
            f[i][j]+=f[i-][(j<<)%]+f[i-][((j<<)%)|];
        }
    }

    For(i,,)
    {
        if(counter(i)==num[n-]&&counter(i>>)==num[n])
            ans+=f[n][i];
    }
    cout<<ans;

    return ;
}