結果

問題 No.2075 GCD Subsequence
ユーザー 沙耶花
提出日時 2022-09-16 21:59:08
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 675 ms / 4,000 ms
コード長 1,123 bytes
コンパイル時間 4,014 ms
コンパイル使用メモリ 255,556 KB
最終ジャッジ日時 2025-02-07 09:27:28
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:23:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   23 |         rep(i,n)scanf("%d",&a[i]);
      |                 ~~~~~^~~~~~~~~~~~

ソースコード

diff #

#include <stdio.h>
#include <atcoder/all>
#include <bits/stdc++.h>
using namespace std;
using namespace atcoder;
using mint = modint998244353;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf32 1000000001
#define Inf64 4000000000000000001

int n;
vector<int> a;
vector<vector<int>> ps(1000001);
vector<mint> dp;
vector<mint> sum;


int main(){
	
	cin>>n;
	
	a.resize(n);
	rep(i,n)scanf("%d",&a[i]);
	
	for(int i=2;i<ps.size();i++){
		if(ps[i].size()!=0)continue;
		for(int j=i;j<ps.size();j+=i)ps[j].push_back(i);
	}
	dp.resize(n);
	sum.resize(1000001,0);
	rep(i,n){
		rep(j,(1<<ps[a[i]].size())){
			if(j==0)continue;
			int cur = 1;
			rep(k,ps[a[i]].size()){
				if((j>>k)&1)cur *= ps[a[i]][k];
			}
			mint v = sum[cur];
			if(__builtin_popcount(j)%2==0)v *= -1;
			dp[i] += v;
			//if(i==3)cout<<v.val()<<endl;
		}
		dp[i]++;
		rep(j,(1<<ps[a[i]].size())){
			if(j==0)continue;
			int cur = 1;
			rep(k,ps[a[i]].size()){
				if((j>>k)&1)cur *= ps[a[i]][k];
			}
			sum[cur] += dp[i];
		}
	}
	mint ans = 0;
	rep(i,n){
		ans += dp[i];
	//	cout<<dp[i].val()<<endl;
	}
	
	cout<<ans.val()<<endl;
	
	
	return 0;
}
0