#391508 (C++14) No.911 ラッキーソート

提出ソース

結果

問題	No.911 ラッキーソート
ユーザー	knshnb
提出日時	2019-10-18 23:32:55
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	101 ms / 2,000 ms
コード長	5,259 bytes
コンパイル時間	2,603 ms
コンパイル使用メモリ	197,824 KB
実行使用メモリ	5,180 KB
最終ジャッジ日時	2023-09-08 02:47:38
合計ジャッジ時間	7,376 ms
ジャッジサーバーID （参考情報）	judge12 / judge13

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 4,380 KB
testcase_01	AC	1 ms 4,380 KB
testcase_02	AC	2 ms 4,384 KB
testcase_03	AC	2 ms 4,384 KB
testcase_04	AC	1 ms 4,380 KB
testcase_05	AC	1 ms 4,380 KB
testcase_06	AC	1 ms 4,380 KB
testcase_07	AC	2 ms 4,380 KB
testcase_08	AC	2 ms 4,380 KB
testcase_09	AC	2 ms 4,384 KB
testcase_10	AC	2 ms 4,384 KB
testcase_11	AC	1 ms 4,380 KB
testcase_12	AC	2 ms 4,380 KB
testcase_13	AC	96 ms 4,544 KB
testcase_14	AC	96 ms 4,688 KB
testcase_15	AC	95 ms 4,612 KB
testcase_16	AC	100 ms 4,688 KB
testcase_17	AC	98 ms 4,760 KB
testcase_18	AC	99 ms 4,708 KB
testcase_19	AC	100 ms 4,708 KB
testcase_20	AC	99 ms 4,760 KB
testcase_21	AC	99 ms 4,592 KB
testcase_22	AC	97 ms 4,592 KB
testcase_23	AC	96 ms 4,704 KB
testcase_24	AC	87 ms 4,500 KB
testcase_25	AC	53 ms 4,380 KB
testcase_26	AC	44 ms 4,384 KB
testcase_27	AC	22 ms 4,380 KB
testcase_28	AC	11 ms 4,380 KB
testcase_29	AC	4 ms 4,504 KB
testcase_30	AC	2 ms 4,380 KB
testcase_31	AC	2 ms 4,384 KB
testcase_32	AC	2 ms 4,380 KB
testcase_33	AC	2 ms 4,380 KB
testcase_34	AC	1 ms 4,380 KB
testcase_35	AC	2 ms 4,384 KB
testcase_36	AC	2 ms 4,380 KB
testcase_37	AC	2 ms 4,380 KB
testcase_38	AC	97 ms 4,692 KB
testcase_39	AC	97 ms 4,544 KB
testcase_40	AC	4 ms 4,384 KB
testcase_41	AC	101 ms 4,756 KB
testcase_42	AC	65 ms 4,380 KB
testcase_43	AC	97 ms 4,756 KB
testcase_44	AC	42 ms 4,940 KB
testcase_45	AC	43 ms 5,064 KB
testcase_46	AC	42 ms 5,180 KB
testcase_47	AC	41 ms 4,912 KB
testcase_48	AC	41 ms 5,048 KB

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

ソースコード

raw source code

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T> using treap = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
using Int = long long;
#define REP(i, n) for (long long i = 0, max_i = (n); i < max_i; i++)
#define REPI(i, a, b) for (long long i = (a), max_i = (b); i < max_i; i++)
#define ALL(obj) begin(obj), end(obj)
#define RALL(obj) rbegin(obj), rend(obj)
#define fi first
#define se second
using ii = pair<Int, Int>;
vector<ii> dirs = {
    {1, 0}, {0, 1}, {-1, 0}, {0, -1},  // 4方向
    {1, 1}, {-1, 1}, {-1, -1}, {1, -1},  // 斜め
    {0, 0},  // 自身
};
Int get_msb(long long x) {
    assert(x != 0);
    return 63 - __builtin_clzll(x);
}
Int get_lsb(long long x) {
    assert(x != 0);
    return __builtin_ctzll(x);
}
template <class T> inline bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; } return false; }
template <class T> inline bool chmin(T& a, const T& b) { if (a > b) { a = b; return true; } return false; }
template <class T, class S> vector<T> make_vec(size_t n, S x) { return vector<T>(n, x); }
template <class T, class... Ts> auto make_vec(size_t n, Ts... ts) { return vector<decltype(make_vec<T>(ts...))>(n, make_vec<T>(ts...)); }

// debug
template <class T> ostream& operator<<(ostream& s, const vector<T>& d) { REP (i, d.size()) s << d[i] << (i == d.size() - 1 ? "" : " "); return s; }
template <class T> ostream& operator<<(ostream& s, const vector<vector<T>>& d) { REP (i, d.size()) s << d[i] << (i == d.size() - 1 ? "" : "\n"); return s; }
template <class T, class S> ostream& operator<<(ostream& s, const pair<T, S>& p) { s << "{" << p.first << ", " << p.second << "}"; return s; }
template <class T> ostream& operator<<(ostream& s, const set<T> m) { for (auto it = m.begin(); it != m.end(); it++) { s << *it << (next(it) == m.end() ? "" : "\n"); } return s; }
template <class T> ostream& operator<<(ostream& s, const multiset<T> m) { for (auto it = m.begin(); it != m.end(); it++) { s << *it << (next(it) == m.end() ? "" : "\n"); } return s; }
template <class T, class S> ostream& operator<<(ostream& s, const map<T, S> m) { for (auto it = m.begin(); it != m.end(); it++) { s << *it << (next(it) == m.end() ? "" : "\n"); } return s; }
template <class T, class S> ostream& operator<<(ostream& s, const unordered_map<T, S> m) { for (auto it = m.begin(); it != m.end(); it++) { s << *it << (next(it) == m.end() ? "" : "\n"); } return s; }
#ifdef _MY_DEBUG
    #define dump(...) cerr << "/* " << #__VA_ARGS__ << " :[" << __LINE__ << ":" << __FUNCTION__ << "]" << endl, dump_func(__VA_ARGS__), cerr << "*/\n\n";
#else
    #define dump(...)
    #define endl "\n"
#endif
void dump_func() { cerr << endl; }
template <class Head, class... Tail> void dump_func(Head&& h, Tail&&... t) { cerr << h << (sizeof...(Tail) == 0 ? "" : ", "), dump_func(forward<Tail>(t)...); }

struct Fast { Fast() { cin.tie(nullptr); ios::sync_with_stdio(false); } } fast;
mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());
constexpr Int MOD = 1000000007;
// *************** TEMPLATE END *************** 

void no() {
    cout << 0 << endl;
    exit(0);
}
constexpr Int K = 62;
signed main() {
    Int n, L, R; cin >> n >> L >> R;
    vector<Int> a(n); REP (i, n) cin >> a[i];
    vector<Int> bits(K, -1);
    vector<bool> ok(n);
    ok[n - 1] = true;
    for (Int k = K - 1; k >= 0; k--) {
        Int i = 0;
        Int prv = -1;
        while (i < n) {
            Int state = -1;
            while (1) {
                Int bit = a[i] >> k & 1;
                if (state == -1) {
                    state = bit;
                } else if (state == 0) {
                    if (bit == 1) {
                        state = 3;
                        ok[i - 1] = true;
                        if (prv != -1 && prv != state) no();
                        prv = state;
                    }
                } else if (state == 1) {
                    if (bit == 0) {
                        state = 2;
                        ok[i - 1] = true;
                        if (prv != -1 && prv != state) no();
                        prv = state;
                    }
                } else if (state == 2) {
                    if (bit == 1) no();
                } else if (state == 3) {
                    if (bit == 0) no();
                } else assert(0);
                if (ok[i++]) break;
            }
        }
        if (prv == 2) bits[k] = 1;
        if (prv == 3) bits[k] = 0;
    }
    REP (i, n) {
        if (!ok[i]) no();
    }

    auto dp = make_vec<Int>(K + 1, 2, 2, 0);
    dp[K][0][0] = 1;
    for (Int k = K - 1; k >= 0; k--) {
        Int l = L >> k & 1;
        Int r = R >> k & 1;
        REP (x, 2) {
            REP (y, 2) {
                REPI (d, x ? 0 : l, (y ? 1 : r) + 1) {
                    if (bits[k] == 0 && d == 1) continue;
                    if (bits[k] == 1 && d == 0) continue;
                    dp[k][x | d > l][y | d < r] += dp[k + 1][x][y];
                }
            }
        }
    }
    Int ans = 0;
    REP (x, 2) {
        REP (y, 2) {
            ans += dp[0][x][y];
        }
    }
    cout << ans << endl;
}

yukicoder

結果

テストケース

ソースコード