結果

問題 No.1304 あなたは基本が何か知っていますか？私は知っています．
ユーザー leafirbyleafirby
提出日時 2021-02-03 00:47:10
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 84 ms / 2,000 ms
コード長 1,417 bytes
コンパイル時間 2,016 ms
コンパイル使用メモリ 175,736 KB
実行使用メモリ 70,880 KB
最終ジャッジ日時 2024-06-12 16:38:33
合計ジャッジ時間 7,975 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 65 RE * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#pragma GCC target("avx")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
using namespace std;
#define ll long long
#define rep(i, m, n) for (ll i = m; i < n; i++)
#define mod 998244353

void fwt(vector<ll>& f) {
    int n = f.size();
    for(int i = 1; i < n; i <<= 1) {
        for(int j = 0; j < n; j++) {
            if((j & i) == 0) {
                ll x = f[j], y = f[j | i];
                f[j] = x + y, f[j | i] = x - y;
            }
        }
    }
}

void ifwt(vector<ll>& f) {
    int n = f.size();
    for(int i = 1; i < n; i <<= 1) {
        for(int j = 0; j < n; j++) {
            if((j & i) == 0) {
                ll x = f[j], y = f[j | i];
                f[j] = (x + y) * 499122177 % mod, f[j | i] = (x - y) * 499122177 % mod;
            }
        }
    }
}

int main(){
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	ll N, K, X, Y; 
    cin >> N >> K >> X >> Y;
	ll M = 262144;
	vector<vector<ll>> dp(N + 2, vector<ll>(M, 1));
    vector<ll> H(M, 0);
    rep(i, 0, K) {
		ll A;
		cin >> A;
		H[A] = 1;
	}
	fwt(H);
    rep(v, 0, M) dp[1][v] = dp[0][v] * H[v];
    rep(v, 0, M) dp[2][v] = dp[1][v] * H[v] - K * dp[0][v];
	rep(i, 3, N + 1) rep(v, 0, M) dp[i][v] = (dp[i - 1][v] * H[v] - (K - 1) * dp[i - 2][v] + mod) % mod;
	ifwt(dp[N]);
	ll res = 0;
	rep(v, 0, M) if(X <= v & v <= Y) res = (res + dp[N][v]) % mod;
	cout << (res + mod) % mod << endl;
}
0