結果

問題 No.1463 Hungry Kanten
コンテスト
ユーザー Example0911
提出日時 2021-04-03 10:32:34
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,097 bytes
コンパイル時間 2,274 ms
コンパイル使用メモリ 212,896 KB
最終ジャッジ日時 2025-01-20 10:22:50
ジャッジサーバーID
（参考情報） 		judge3 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 19 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
//#define int long long
using namespace std;
using ll = long long;
using P = pair<ll, ll>;
const ll INF = (1LL << 61);
ll mod = 1000000007;
map<ll, ll> prime_factor(ll n) {
	map<ll, ll> ret;
	for (ll i = 2; i * i <= n; i++) {
		while (n % i == 0) {
			ret[i]++;
			n /= i;
		}
	}
	if (n != 1) ret[n] = 1;
	return ret;
}
signed main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	ll N, K; cin >> N >> K;
	vector<ll>A(N); for (int i = 0; i < N; i++)cin >> A[i];
	vector<map<ll, ll>>t(N);for(int i = 0; i < N; i++) t[i] = prime_factor(A[i]);
	map<map<ll, ll>, bool>mp;
	for (ll bit = 0; bit < (1LL << N); bit++) {
		int cnt = 0;
		for (ll i = 0; i < N; i++) {
			if (bit >> i & 1) {
				cnt++;
			}
		}
		if (cnt < K)continue;
		ll sum = 0;
		for (int i = 0; i < N; i++) {
			if (bit >> i & 1) {
				sum += A[i];
			}
		}
		auto p = prime_factor(sum);
		mp[p] = true;
		map<ll, ll>tmp;
		for (int i = 0; i < N; i++) {
			if (bit >> i & 1) {
				for (auto y : t[i]) {
					tmp[y.first] += y.second;
				}
			}
		}
		mp[tmp] = true;
	}
	cout << mp.size() << endl;
	return 0;
}
0