結果

問題 No.917 Make One With GCD
ユーザー leaf_1415leaf_1415
提出日時 2019-10-25 22:44:04
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 8 ms / 2,000 ms
コード長 1,166 bytes
コンパイル時間 671 ms
コンパイル使用メモリ 78,404 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-24 11:05:23
合計ジャッジ時間 1,759 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
#include <algorithm>
#include <unordered_map>
#define llint long long

using namespace std;

llint n;
llint a[55];
vector<llint> vec[55];
llint dp[55][1<<10];
unordered_map<llint, llint> mp;

int main(void)
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	cin >> n;
	for(int i = 1; i <= n; i++) cin >> a[i];
	
	for(int i = 1; i <= n; i++){
		llint t = a[i];
		for(int j = 2; j < 10005; j++){
			bool flag = false;
			while(t % j == 0){
				if(!flag) vec[i].push_back(j);
				flag = true;
				t /= j;
			}
		}
		if(t > 1) vec[i].push_back(t);
	}
	
	llint ans = 0;
	for(int s = 1; s <= n; s++){
		llint m = vec[s].size(), M = 1<<m;
		mp.clear();
		for(int i = 0; i < vec[s].size(); i++) mp[vec[s][i]] = i;
		
		for(int i = 1; i <= n; i++){
			for(int j = 0; j < M; j++){
				dp[i][j] = 0;
			}
		}
		dp[s][M-1] = 1;
		
		for(int i = s; i < n; i++){
			llint mask = 0;
			for(int j = 0; j < vec[i+1].size(); j++){
				if(mp.count(vec[i+1][j])) mask |= 1<<mp[vec[i+1][j]];
			}
			for(int j = 0; j < M; j++){
				dp[i+1][j] += dp[i][j];
				dp[i+1][j&mask] += dp[i][j];
			}
		}
		ans += dp[n][0];
	}
	cout << ans << endl;
	
	return 0;
}
0