#9296 (C++11) No.7 プライムナンバーゲーム

• 問題: No.7 プライムナンバーゲーム
• ユーザー: masa
• 提出日時: 2015-01-19 03:12:52
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 99 ms / 5,000 ms
• コード長: 832 bytes
• コンパイル時間: 694 ms
• コンパイル使用メモリ: 72,656 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-10-01 15:28:57
• 合計ジャッジ時間: 1,757 ms
• ジャッジサーバーID （参考情報）: judge3 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	1 ms; 5,248 KB
testcase_02	AC	99 ms; 5,248 KB
testcase_03	AC	5 ms; 5,248 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	2 ms; 5,248 KB
testcase_06	AC	25 ms; 5,248 KB
testcase_07	AC	16 ms; 5,248 KB
testcase_08	AC	7 ms; 5,248 KB
testcase_09	AC	40 ms; 5,248 KB
testcase_10	AC	1 ms; 5,248 KB
testcase_11	AC	16 ms; 5,248 KB
testcase_12	AC	70 ms; 5,248 KB
testcase_13	AC	74 ms; 5,248 KB
testcase_14	AC	99 ms; 5,248 KB
testcase_15	AC	93 ms; 5,248 KB
testcase_16	AC	87 ms; 5,248 KB

ソースコード

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <utility>
#include <set>

using namespace std;

int main() {
	int n;
	cin >> n;

	vector<bool> is_prime(n + 1, true);
	is_prime[0] = is_prime[1] = false;
	vector<int> primes;

	for (int i = 2; i * i <= n; i++) {
		if (is_prime[i]) {
			for (int j = i * i; j <= n; j += i) {
				is_prime[j] = false;
			}
		}
	}
	for (int i = 2; i <= n; i++) {
		if (is_prime[i]) {
			primes.push_back(i);
		}
	}

	set<int> lose;

	for (int i = 2; i <= n; i++) {
		bool win = false;
		for (int j = 0;  j < primes.size() && primes[j] <= i; j++) {
			if (lose.count(i - primes[j]) > 0) {
				win = true;
				break;
			}
		}
		if (!win) {
			lose.insert(i);
			// printf("insert %d\n", i);
		}
	}

	cout << (lose.count(n) > 0 ? "Lose" : "Win") << endl;

	return 0;
}