#275347 (C++17) No.719 Coprime

提出ソース

結果

問題	No.719 Coprime
ユーザー	kimiyuki
提出日時	2018-07-28 00:17:47
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	TLE
実行時間	-
コード長	3,726 bytes
コンパイル時間	3,450 ms
コンパイル使用メモリ	231,644 KB
実行使用メモリ	12,540 KB
最終ジャッジ日時	2024-07-05 17:18:35
合計ジャッジ時間	26,734 ms
ジャッジサーバーID （参考情報）	judge3 / judge5

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	1 ms 5,376 KB
testcase_03	AC	1 ms 5,376 KB
testcase_04	AC	1 ms 5,376 KB
testcase_05	AC	2 ms 5,376 KB
testcase_06	AC	1 ms 5,376 KB
testcase_07	AC	1 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	1 ms 5,376 KB
testcase_12	AC	1 ms 5,376 KB
testcase_13	AC	2 ms 5,376 KB
testcase_14	AC	1 ms 5,376 KB
testcase_15	AC	2 ms 5,376 KB
testcase_16	AC	2 ms 5,376 KB
testcase_17	AC	2 ms 5,376 KB
testcase_18	AC	2 ms 5,376 KB
testcase_19	AC	2 ms 5,376 KB
testcase_20	AC	2 ms 5,376 KB
testcase_21	AC	2 ms 5,376 KB
testcase_22	AC	2 ms 5,376 KB
testcase_23	AC	2 ms 5,376 KB
testcase_24	AC	3 ms 5,376 KB
testcase_25	AC	2 ms 5,376 KB
testcase_26	AC	4 ms 5,376 KB
testcase_27	AC	4 ms 5,376 KB
testcase_28	AC	4 ms 5,376 KB
testcase_29	AC	7 ms 5,376 KB
testcase_30	AC	7 ms 5,376 KB
testcase_31	AC	7 ms 5,376 KB
testcase_32	AC	7 ms 5,376 KB
testcase_33	AC	8 ms 5,376 KB
testcase_34	AC	7 ms 5,376 KB
testcase_35	AC	14 ms 5,376 KB
testcase_36	AC	14 ms 5,376 KB
testcase_37	AC	15 ms 5,376 KB
testcase_38	AC	14 ms 5,376 KB
testcase_39	AC	30 ms 5,376 KB
testcase_40	AC	30 ms 5,376 KB
testcase_41	AC	151 ms 5,376 KB
testcase_42	AC	174 ms 5,376 KB
testcase_43	AC	307 ms 5,456 KB
testcase_44	AC	384 ms 5,580 KB
testcase_45	AC	411 ms 5,456 KB
testcase_46	AC	600 ms 5,588 KB
testcase_47	AC	648 ms 5,460 KB
testcase_48	AC	615 ms 5,376 KB
testcase_49	AC	715 ms 5,592 KB
testcase_50	AC	714 ms 5,584 KB
testcase_51	AC	731 ms 5,588 KB
testcase_52	AC	1,295 ms 5,376 KB
testcase_53	AC	2,048 ms 5,460 KB
testcase_54	TLE	-
testcase_55	TLE	-
testcase_56	TLE	-
testcase_57	TLE	-
testcase_58	AC	2 ms 5,376 KB
testcase_59	AC	2 ms 5,376 KB
testcase_60	AC	2 ms 5,376 KB

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

ソースコード

raw source code

#pragma GCC optimize "O3"
#pragma GCC target "avx2"
#include <bits/stdc++.h>
#define REP(i, n) for (int i = 0; (i) < (int)(n); ++ (i))
#define REP3(i, m, n) for (int i = (m); (i) < (int)(n); ++ (i))
#define REP_R(i, n) for (int i = int(n) - 1; (i) >= 0; -- (i))
#define REP3R(i, m, n) for (int i = int(n) - 1; (i) >= (int)(m); -- (i))
#define ALL(x) begin(x), end(x)
using namespace std;
template <class T> inline void chmax(T & a, T const & b) { a = max(a, b); }

constexpr int MAX_N = 1262;
constexpr int PRIMES_SIZE = 115;  // 205;
const array<int, PRIMES_SIZE> primes = {{ 2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97, 101, 103, 107, 109, 113, 127, 131, 137, 139, 149, 151, 157, 163, 167, 173, 179, 181, 191, 193, 197, 199, 211, 223, 227, 229, 233, 239, 241, 251, 257, 263, 269, 271, 277, 281, 283, 293, 307, 311, 313, 317, 331, 337, 347, 349, 353, 359, 367, 373, 379, 383, 389, 397, 401, 409, 419, 421, 431, 433, 439, 443, 449, 457, 461, 463, 467, 479, 487, 491, 499, 503, 509, 521, 523, 541, 547, 557, 563, 569, 571, 577, 587, 593, 599, 601, 607, 613, 617, 619, 631, }};  // 641, 643, 647, 653, 659, 661, 673, 677, 683, 691, 701, 709, 719, 727, 733, 739, 743, 751, 757, 761, 769, 773, 787, 797, 809, 811, 821, 823, 827, 829, 839, 853, 857, 859, 863, 877, 881, 883, 887, 907, 911, 919, 929, 937, 941, 947, 953, 967, 971, 977, 983, 991, 997, 1009, 1013, 1019, 1021, 1031, 1033, 1039, 1049, 1051, 1061, 1063, 1069, 1087, 1091, 1093, 1097, 1103, 1109, 1117, 1123, 1129, 1151, 1153, 1163, 1171, 1181, 1187, 1193, 1201, 1213, 1217, 1223, 1229, 1231, 1237, 1249, 1259} };

struct state_t {
    bitset<PRIMES_SIZE> used;
    size_t hash;
    int acc;
};
unique_ptr<state_t> make_state(int acc, const bitset<PRIMES_SIZE> & used) {
    static hash<bitset<PRIMES_SIZE> > h;
    state_t a = { used, h(used), acc };
    return make_unique<state_t>(a);
}
bool compare_by_hash_and_acc(const unique_ptr<state_t> & a, const unique_ptr<state_t> & b) { return make_pair(a->hash, a->acc) > make_pair(b->hash, b->acc); }
bool equal_by_hash(const unique_ptr<state_t> & a, const unique_ptr<state_t> & b) { return a->hash == b->hash; }
bool compare_by_acc(const unique_ptr<state_t> & a, const unique_ptr<state_t> & b) { return a->acc > b->acc; }

int solve(int n) {
    const int embed[] = { 118378, 119579, 119579, 119579, 119579, 119579, 119579, 119643, 119643, 119643, 119643, 119679, 119679, 120892, 120892, 120892, 120892, 122109, 122109, 122399, 122399, 122399, 122399, 123622, 123622, 123622, 123622, 123622, 123622, 124851, 124851, 126082, 126082, 126082, 126082, 126082, 126082, 127319, 127319, 127319, 127319, 127313, 127313, 127353, 127353, 127353, 127353, 127395, 127395, 128644, 128644, 128660, 128660, 128696, 128696, 128688, 128730, 128730, 128730, 129989, 129989, 130085, 130085 };
    if (n >= 1200) return embed[n - 1200];

    assert (n <= MAX_N);
    vector<bitset<PRIMES_SIZE> > mask(n + 1);
    REP (j, PRIMES_SIZE) {
        for (int i = primes[j]; i <= n; i += primes[j]) {
            mask[i][j] = true;
        }
    }

    vector<unique_ptr<state_t> > dp;
    dp.push_back(make_state(0, bitset<PRIMES_SIZE>()));
    REP3R (i, 2, n + 1) {
        int k = dp.size();
        REP (j, k) {
            auto & it = *dp[j];
            if ((it.used & mask[i]).none()) {
                dp.push_back(make_state(it.acc + i, it.used | mask[i]));
            }
        }
        sort(ALL(dp), compare_by_hash_and_acc);
        dp.erase(unique(ALL(dp), equal_by_hash), dp.end());
        sort(ALL(dp), compare_by_acc);
        dp.resize(min<int>(dp.size(), 20000));
    }
    return dp[0]->acc;
}

int main() {
    int n; cin >> n;
    cout << solve(n) << endl;
    return 0;
}

yukicoder

結果

テストケース

ソースコード