結果

問題 No.7 プライムナンバーゲーム
コンテスト
ユーザー Luna
提出日時 2018-09-05 11:42:55
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 21 ms / 5,000 ms
コード長 1,258 bytes
コンパイル時間 709 ms
コンパイル使用メモリ 72,608 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-01 16:14:31
合計ジャッジ時間 1,421 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
#include <cmath>


std::vector<int> primeNumbers(const int n) {
    std::vector<int> prime_numbers;

    for (int i = 2; i < n+1; i++) {
        prime_numbers.push_back(i);
    }

    for (auto itr = prime_numbers.begin(); *itr < sqrt(n); itr++) {
        int p = *itr; auto start = itr; start++;
        auto end = std::remove_if(start, prime_numbers.end(), [=](int x) { return x % p == 0; });
        prime_numbers.erase(end, prime_numbers.end());
    }

    return prime_numbers;
}


std::vector<int> vectorLessThanN(std::vector<int> vec, const int n) {
    auto end = std::remove_if(vec.begin(), vec.end(), [=](int x) { return x > n; });
    vec.erase(end, vec.end());
    return vec;
}


int main(int argc, char const* argv[]) {
    int n; std::cin >> n;
    std::vector<bool> dp(n+10, false);

    dp[0] = true;
    dp[1] = true;

    auto prime_numbers = primeNumbers(n+10);
    for (int i = 2; i < n+1; i++) {
        for (auto&& p : vectorLessThanN(prime_numbers, i)) {
            dp[i] = dp[i] + !dp[i-p];
            if (dp[i]) break;
        }
    }

    if (dp[n]) {
        std::cout << "Win" << std::endl;
    } else {
        std::cout << "Lose" << std::endl;
    }
    return 0;
}
0