結果

問題 No.1304 あなたは基本が何か知っていますか？私は知っています．
ユーザー leafirby
提出日時 2021-02-05 18:12:01
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 260 ms / 2,000 ms
コード長 2,105 bytes
コンパイル時間 1,613 ms
コンパイル使用メモリ 174,540 KB
実行使用メモリ 21,740 KB
最終ジャッジ日時 2025-06-22 03:06:22
合計ジャッジ時間 35,898 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 65 TLE * 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 + mod) % mod * 499122177 % mod;
            }
        }
    }
}

vector<ll> mul(vector<ll> &mt1, vector<vector<ll>> &mt2) {
    vector<ll> res(2);
    res[0] = (mt1[0] * mt2[0][0] % mod + mt1[1] * mt2[0][1] % mod) % mod;
    res[1] = (mt1[0] * mt2[1][0] % mod + mt1[1] * mt2[1][1] % mod) % mod;
    return res;
}

vector<vector<ll>> update(vector<vector<ll>> &mt) {
    vector<vector<ll>> res(2, vector<ll>(2, 0));
    rep(i, 0, 2) rep(j, 0, 2) rep(k, 0, 2) res[i][j] = (res[i][j] + mt[i][k] * mt[k][j] % mod) % mod;
    return res; 
}

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(M, vector<ll>(2));
    vector<ll> H(M, 0);
    rep(i, 0, K) {
		ll A;
		cin >> A;
		H[A] = 1;
	}
	fwt(H);
    rep(v, 0, M) dp[v][1] = H[v];
    rep(v, 0, M) dp[v][0] = (dp[v][1] * H[v] - K + mod) % mod;
	rep(v, 0, M) {
        vector<vector<ll>> mt(2, vector<ll>(2));
        mt[0] = {H[v], (1 - K + mod) % mod};
        mt[1] = {1, 0};
        ll P = N - 1;
        while(P) {
            if(P & 1) dp[v] = mul(dp[v], mt);
            mt = update(mt);
            P /= 2;
        }
    }
    vector<ll> ep(M);
    rep(v, 0, M) ep[v] = dp[v][1];
	ifwt(ep);
	ll res = 0;
	rep(v, X, Y + 1) res = (res + ep[v]) % mod;
	cout << (res + mod) % mod << endl;
}
0