#587457 (C++14) No.1304 あなたは基本が何か知っていますか？私は知っています．

提出ソース

結果

問題	No.1304 あなたは基本が何か知っていますか？私は知っています．
ユーザー	fuppy_kyopro
提出日時	2020-12-01 03:58:09
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	7 ms / 2,000 ms
コード長	4,846 bytes
コンパイル時間	3,123 ms
コンパイル使用メモリ	197,860 KB
実行使用メモリ	6,948 KB
最終ジャッジ日時	2024-06-12 09:41:50
合計ジャッジ時間	14,599 ms
ジャッジサーバーID （参考情報）	judge3 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	2 ms 6,944 KB
testcase_02	AC	2 ms 6,940 KB
testcase_03	AC	3 ms 6,940 KB
testcase_04	AC	7 ms 6,940 KB
testcase_05	AC	4 ms 6,944 KB
testcase_06	AC	2 ms 6,944 KB
testcase_07	AC	6 ms 6,940 KB
testcase_08	AC	2 ms 6,944 KB
testcase_09	AC	6 ms 6,944 KB
testcase_10	AC	4 ms 6,940 KB
testcase_11	AC	6 ms 6,940 KB
testcase_12	AC	5 ms 6,940 KB
testcase_13	AC	4 ms 6,944 KB
testcase_14	AC	5 ms 6,940 KB
testcase_15	AC	4 ms 6,944 KB
testcase_16	AC	5 ms 6,944 KB
testcase_17	AC	2 ms 6,944 KB
testcase_18	AC	5 ms 6,940 KB
testcase_19	AC	4 ms 6,944 KB
testcase_20	AC	2 ms 6,944 KB
testcase_21	AC	3 ms 6,940 KB
testcase_22	AC	5 ms 6,940 KB
testcase_23	AC	5 ms 6,940 KB
testcase_24	AC	7 ms 6,944 KB
testcase_25	AC	3 ms 6,940 KB
testcase_26	AC	2 ms 6,940 KB
testcase_27	AC	2 ms 6,948 KB
testcase_28	AC	2 ms 6,940 KB
testcase_29	AC	2 ms 6,944 KB
testcase_30	AC	2 ms 6,944 KB
testcase_31	AC	3 ms 6,940 KB
testcase_32	AC	2 ms 6,944 KB
testcase_33	AC	2 ms 6,944 KB
testcase_34	AC	2 ms 6,940 KB
testcase_35	AC	3 ms 6,940 KB
testcase_36	AC	3 ms 6,944 KB
testcase_37	AC	3 ms 6,940 KB
testcase_38	AC	2 ms 6,944 KB
testcase_39	AC	3 ms 6,940 KB
testcase_40	AC	3 ms 6,944 KB
testcase_41	AC	2 ms 6,940 KB
testcase_42	AC	3 ms 6,944 KB
testcase_43	AC	3 ms 6,940 KB
testcase_44	AC	3 ms 6,940 KB
04_evil_A_01	WA	-
04_evil_A_02	WA	-
04_evil_A_03	WA	-
04_evil_A_04	WA	-
04_evil_A_05	WA	-
04_evil_A_06	RE	-
04_evil_A_07	RE	-
04_evil_A_08	RE	-
04_evil_A_09	RE	-
04_evil_A_10	RE	-
05_evil_B_01	RE	-
05_evil_B_02	RE	-
05_evil_B_03	RE	-
05_evil_B_04	RE	-
05_evil_B_05	RE	-
05_evil_B_06	RE	-
05_evil_B_07	RE	-
05_evil_B_08	RE	-
05_evil_B_09	RE	-
05_evil_B_10	RE	-
06_evil_C_01	RE	-
06_evil_C_02	RE	-
06_evil_C_03	RE	-
06_evil_C_04	RE	-
06_evil_C_05	RE	-
06_evil_C_06	RE	-
06_evil_C_07	RE	-
06_evil_C_08	RE	-
06_evil_C_09	RE	-
06_evil_C_10	RE	-

権限があれば一括ダウンロードができます

ソースコード

raw source code

//*
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
//*/
 
#include <bits/stdc++.h>
 
// #include <atcoder/all>
 
using namespace std;
// using namespace atcoder;
 
#define DEBUG(x) cerr<<#x<<": "<<x<<endl;
#define DEBUG_VEC(v) cerr<<#v<<":";for(int i=0;i<v.size();i++) cerr<<" "<<v[i]; cerr<<endl;
#define DEBUG_MAT(v) cerr<<#v<<endl;for(int i=0;i<v.size();i++){for(int j=0;j<v[i].size();j++) {cerr<<v[i][j]<<" ";}cerr<<endl;}
typedef long long ll;
// #define int ll
 
#define vi vector<int>
#define vl vector<ll>
#define vii vector< vector<int> >
#define vll vector< vector<ll> >
#define vs vector<string>
#define pii pair<int,int>
#define pis pair<int,string>
#define psi pair<string,int>
#define pll pair<ll,ll>
template<class S, class T> pair<S, T> operator+(const pair<S, T> &s, const pair<S, T> &t) { return pair<S, T>(s.first + t.first, s.second + t.second); }
template<class S, class T> pair<S, T> operator-(const pair<S, T> &s, const pair<S, T> &t) { return pair<S, T>(s.first - t.first, s.second - t.second); }
template<class S, class T> ostream& operator<<(ostream& os, pair<S, T> p) { os << "(" << p.first << ", " << p.second << ")"; return os; }
#define X first
#define Y second
#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define rep1(i,n) for(int i=1;i<=(int)(n);i++)
#define rrep(i,n) for(int i=(int)(n)-1;i>=0;i--)
#define rrep1(i,n) for(int i=(int)(n);i>0;i--)
#define REP(i,a,b) for(int i=a;i<b;i++)
#define in(x, a, b) (a <= x && x < b)
#define all(c) c.begin(),c.end()
void YES(bool t=true) {cout<<(t?"YES":"NO")<<endl;}
void Yes(bool t=true) {cout<<(t?"Yes":"No")<<endl;}
void yes(bool t=true) {cout<<(t?"yes":"no")<<endl;}
void NO(bool t=true) {cout<<(t?"NO":"YES")<<endl;}
void No(bool t=true) {cout<<(t?"No":"Yes")<<endl;}
void no(bool t=true) {cout<<(t?"no":"yes")<<endl;}
template<class T> bool chmax(T &a, const T &b) { if (a<b) { a = b; return 1; } return 0; }
template<class T> bool chmin(T &a, const T &b) { if (a>b) { a = b; return 1; } return 0; }
#define UNIQUE(v) v.erase(std::unique(v.begin(), v.end()), v.end());
const ll inf = 1000000001;
const ll INF = (ll)1e18 + 1;
const long double pi = 3.1415926535897932384626433832795028841971L;
#define Sp(p) cout<<setprecision(25)<< fixed<<p<<endl;
// int dx[4] = {1, 0, -1, 0}, dy[4] = {0, 1, 0, -1};
// int dx2[8] = { 1,1,0,-1,-1,-1,0,1 }, dy2[8] = { 0,1,1,1,0,-1,-1,-1 };
vi dx = {0, 1, 0, -1}, dy = {-1, 0, 1, 0};
// vi dx2 = { 1,1,0,-1,-1,-1,0,1 }, dy2 = { 0,1,1,1,0,-1,-1,-1 };
#define fio() cin.tie(0); ios::sync_with_stdio(false);
// const ll MOD = 1000000007;
const ll MOD = 998244353;
// #define mp make_pair
//#define endl '\n'

int dp[41][1024][1024];

signed main() {
    int n, k, x, y;
    cin >> n >> k >> x >> y;
    assert(n % 2 == 0);
    assert(n <= 40);
    assert(k <= 200000);
    ll times = 1;
    rep (i, n) times *= k;
    assert(times <= 2.1 * 1e15);

    vl a(k);
    rep (i, k) cin >> a[i];
    sort(all(a));
    UNIQUE(a);
    
    if (n >= 10) {
        rep (i, a.size()) {
            dp[1][i][a[i]] = 1;
        }

        rep1 (i, n - 1) {
            rep (j, a.size()) rep (k, 1024) {
                if (dp[i][j][k] == 0) continue;
                rep (l, a.size()) {
                    if (j == l) continue;
                    (dp[i + 1][l][k ^ a[l]] += dp[i][j][k]) %= MOD;
                }
            }
        }

        ll ans = 0;
        rep (i, 1024) {
            if (i < x or i > y) continue;
            rep (j, a.size()) {
                ans += dp[n][j][i];
            }
        }
        ans %= MOD;
        cout << ans << endl;
    }
    else {
        vii dp(n + 1, vi(1024));
        dp[0][0] = 1;
        rep (i, n) {
            rep (j, 1024) {
                for (int u: a) {
                    (dp[i + 1][j ^ u] += dp[i][j]) %= MOD;
                }
            }
        }

        ll ans = 0;
        rep (i, 1 << (n - 1)) {
            int cnt = 0;
            int num = 0;
            int now = 1;
            ll bonus = 1;
            rep (j, n - 1) {
                if (i & (1 << j)) {
                    cnt++;
                    now = 1 - now;
                }
                else {
                    if (now == 0) (bonus *= a.size()) %= MOD;
                    else num += now;
                    now = 1;
                }
            }
            if (now == 0) (bonus *= a.size()) %= MOD;
            else num += now;

            rep (i, 1024) {
                if (in(i, x, y + 1)) {
                    if (cnt % 2 == 0) {
                        ans += dp[num][i] * bonus % MOD;
                    }
                    else {
                        ans -= dp[num][i] * bonus % MOD;
                    }
                    ans %= MOD;
                }
            }
        }
        cout << (ans + MOD) % MOD << endl;
    }
}

yukicoder

結果

テストケース

ソースコード