結果

問題 No.1463 Hungry Kanten
ユーザー takumattakumat
提出日時 2021-04-09 23:30:03
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,667 bytes
コンパイル時間 4,899 ms
コンパイル使用メモリ 286,816 KB
実行使用メモリ 292,192 KB
最終ジャッジ日時 2024-06-25 07:27:41
合計ジャッジ時間 8,522 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
10,624 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 TLE -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>

typedef unsigned long long ULLONG;
typedef long long LLONG;
static const LLONG MOD_NUM = 1000000007LL; //998244353LL;
static const LLONG INF_NUM = LLONG_MAX / 20LL;
template<class _T> static void get(_T& a) {
	std::cin >> a;
}
template<class _T> static void get(_T& a, _T& b) {
	std::cin >> a >> b;
}
template<class _T> static void get(_T& a, _T& b, _T& c) {
	std::cin >> a >> b >> c;
}
template <class _T> static _T tp_abs(_T a) {
	if (a < (_T)0) {
		a *= (_T)-1;
	}
	return a;
}

static void task();

int main()
{
	task();
	fflush(stdout);
	return 0;
}

template <class _T> void tp_getPrimes(_T targ, std::map<_T, _T>& primes)
{
	_T target = targ;
	_T i = 2;
	while ((i * i) <= target) {
		if ((target % i) == 0) {
			target /= i;
			primes[i]++;
		}
		else {
			i++;
		}
	}
	if (target > 1) {
		primes[target]++;
	}
}


static void task()
{
	int N, K;
	get(N, K);

	std::vector<LLONG> ai(N);
	std::vector<std::map<int, int>> prms;
	for (int i = 0; i < N; i++) {
		get(ai[i]);
		std::map<int, int> pr;
		tp_getPrimes<int>(ai[i], pr);
		prms.push_back(pr);
	}
	
	std::set<LLONG> ans;
	std::set<std::set<std::map<int, int>>> ps;
	for (int k = 0; k < (1 << N); k++) {
		std::bitset<32> bs(k);
		if (bs.count() < K) continue;

		LLONG sum = 0, mul = 1;
		std::set<std::map<int, int>> pset;
		for (int i = 0; i < N; i++) {
			if (k & (1 << i)) {
				sum += ai[i];
				mul *= ai[i];
				if (mul >= 100000) {
					mul = 100000;
				}
				pset.insert(prms[i]);
			}
		}

		ans.insert(sum);
		if (mul >= 100000) {
			ps.insert(pset);
		}
		else {
			ans.insert(mul);
		}
	}
	printf("%d\n", (int)(ans.size() + ps.size()));
}
0