結果

問題 No.1426 Got a Covered OR
ユーザー trineutrontrineutron
提出日時 2021-03-12 22:26:14
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 1,554 bytes
コンパイル時間 2,376 ms
コンパイル使用メモリ 204,876 KB
最終ジャッジ日時 2025-01-19 15:11:24
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1 MLE * 2
other TLE * 15 MLE * 9
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/modint>

using namespace std;
using namespace atcoder;

using mint = modint1000000007;

vector memo(100001, vector(30, vector(30, -1)));

vector<mint> fact, inv;

void init(int n)
{
    fact.push_back(1);
    inv.push_back(1);
    for (int i = 0; i < n; i++)
    {
        fact.push_back(fact.back() * (i + 1));
        inv.push_back(1 / fact.back());
    }
}

mint ncr(int n, int r)
{
    if (r < 0 or n - r < 0)
    {
        return 0;
    }
    return fact[n] * inv[r] * inv[n - r];
}

mint calc(int n, int pc0, int pc)
{
    if (memo.at(n).at(pc0).at(pc) != -1)
    {
        return memo.at(n).at(pc0).at(pc);
    }
    if (n == 0)
    {
        return pc0 == pc;
    }
    mint res = 0;
    for (int i = pc0; i <= pc; i++)
    {
        res += calc(n - 1, i, pc) * ((1 << pc0) * ncr(pc - pc0, i - pc0) - (i == pc0));
    }
    memo.at(n).at(pc0).at(pc) = res.val();
    return res;
}

int main(int argc, char const *argv[])
{
    int n;
    cin >> n;
    vector<int> b(n + 1);
    for (int i = 0; i < n; i++)
    {
        cin >> b.at(i + 1);
    }

    init(30);
    mint ans = 1;
    int last = 0;
    for (int i = 0; i < n; i++)
    {
        if (b.at(i + 1) != -1)
        {
            if ((b.at(last) & b.at(i + 1)) != b.at(last))
            {
                ans = 0;
                break;
            }
            ans *= calc(i + 1 - last, __builtin_popcount(b.at(last)), __builtin_popcount(b.at(i + 1)));
            last = i + 1;
        }
    }
    cout << ans.val() << endl;
    return 0;
}
0