結果

問題 No.917 Make One With GCD
ユーザー ningenMe
提出日時 2019-09-09 05:01:12
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,617 bytes
コンパイル時間 2,152 ms
コンパイル使用メモリ 183,100 KB
実行使用メモリ 19,200 KB
最終ジャッジ日時 2024-06-28 00:32:57
合計ジャッジ時間 7,482 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 RE * 1
other AC * 3 RE * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

//divisor O(sqrt(N))
set<long long> Divisor(long long N) {
    set<long long> ret;
    for (long long i = 1; i*i <= N; ++i) {
        if (N%i == 0) {
            ret.insert(i);
            ret.insert(N / i);
        }
    }
    return ret;
}

//Greatest Common Divisor
long long GCD(long long a, long long b) {
    return ((b == 0) ? a : GCD(b, a % b));
}

int main() {
	int N,MAX_A = 2000; cin >> N;
	assert(1<=N && N <= 40);
	vector<int> A(N);
	for(int i = 0; i < N; ++i) cin >> A[i];
	for(int i = 0; i < N; ++i) assert(1 <= A[i] && A[i] <= 1000);
	
	// gcdを前計算 O(MAX_A*MAX_A*log(MAX_A))
	vector<vector<int>> gcd(MAX_A+1,vector<int>(MAX_A+1,0));
	for(int i = 1; i <= MAX_A; ++i) for(int j = i; j <= MAX_A; ++j) gcd[i][j] = gcd[j][i] = GCD(i,j);
	
	//約数のindexをメモするテーブル
	vector<int> table(MAX_A+1,0);

	//左端決め打ち全探索 O(N*N*N*M+N*sqrt(A)+N*M*logM)
	long long ans = 0;
	for(int i = 0; i < N; ++i) {

		//約数列挙 O(sqrt(A))
		auto d = Divisor(A[i]);
		int M = d.size(),idx = 0;
		vector<int> val;

		//添え字前計算 O(M*logM)
		for(auto& e:d) {
			table[e] = idx++;
			val.push_back(e);
		}

		// dp[i][j] := (最後にA[i]を使ってgcdがval[j]になるような場合の数) 
		vector<vector<long long>> dp(N, vector<long long>(M,0));
		dp[i][table[A[i]]] = 1;

		// 遷移 O(N*N*M)
		for(int j = i; j < N; ++j) {
			for(int k = j + 1; k < N; ++k){
				for(int l = 0; l < M; ++l){
					dp[k][table[gcd[val[l]][A[k]]]] += dp[j][l];
				}
			}
			ans += dp[j][table[1]];
		}
	}
	cout << ans << endl;
	return 0;
}
0