結果

問題 No.2075 GCD Subsequence
ユーザー たたき@競プロたたき@競プロ
提出日時 2023-08-06 07:29:53
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 815 ms / 4,000 ms
コード長 1,813 bytes
コンパイル時間 5,948 ms
コンパイル使用メモリ 318,184 KB
実行使用メモリ 22,812 KB
最終ジャッジ日時 2024-11-06 10:27:17
合計ジャッジ時間 20,996 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 19 ms
11,772 KB
testcase_01 AC 20 ms
11,784 KB
testcase_02 AC 20 ms
11,656 KB
testcase_03 AC 19 ms
11,780 KB
testcase_04 AC 19 ms
11,640 KB
testcase_05 AC 19 ms
11,640 KB
testcase_06 AC 19 ms
11,704 KB
testcase_07 AC 19 ms
11,724 KB
testcase_08 AC 487 ms
19,744 KB
testcase_09 AC 815 ms
22,812 KB
testcase_10 AC 507 ms
19,900 KB
testcase_11 AC 715 ms
21,588 KB
testcase_12 AC 599 ms
20,736 KB
testcase_13 AC 415 ms
19,048 KB
testcase_14 AC 668 ms
21,584 KB
testcase_15 AC 444 ms
19,328 KB
testcase_16 AC 491 ms
19,748 KB
testcase_17 AC 807 ms
22,600 KB
testcase_18 AC 729 ms
13,416 KB
testcase_19 AC 774 ms
13,408 KB
testcase_20 AC 727 ms
13,540 KB
testcase_21 AC 734 ms
13,536 KB
testcase_22 AC 713 ms
13,540 KB
testcase_23 AC 751 ms
13,536 KB
testcase_24 AC 732 ms
13,416 KB
testcase_25 AC 754 ms
13,544 KB
testcase_26 AC 735 ms
13,544 KB
testcase_27 AC 751 ms
13,536 KB
testcase_28 AC 533 ms
12,412 KB
testcase_29 AC 20 ms
11,712 KB
testcase_30 AC 19 ms
11,708 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using mint=modint998244353; //1000000007;
using ll=long long;
using pp=pair<int,int>;
#define sr string 
#define vc vector
#define fi first
#define se second
#define rep(i,n) for(int i=0;i<(int)n;i++)
#define pb push_back
#define all(v) v.begin(),v.end()
#define pque priority_queue
#define bpc(a) __builtin_popcount(a)
struct divs{
  int n;
  vector<int> f, primes;
  divs(int n=1):n(n), f(n+1) {
    f[0] = f[1] = -1;
    for (ll i = 2; i <= n; ++i) {
      if (f[i]) continue;
      primes.push_back(i);
      f[i] = i;
      for (ll j = i*i; j <= n; j += i) {
        if (!f[j]) f[j] = i;
      }
    }
  }
  bool isPrime(int x) { return f[x] == x;}
  vector<int> factorList(int x) {
    vector<int> res;
    while (x != 1) {
      res.push_back(f[x]);
      x /= f[x];
    }
    return res;
  }
  vector<pair<int,int>> factor(int x) {
    vector<int> fl = factorList(x);
    if (fl.size() == 0) return {};
    vector<pair<int,int>> res(1, pair<int,int>(fl[0], 0));
    for (int p : fl) {
      if (res.back().first == p) {
        res.back().second++;
      } else {
        res.emplace_back(p, 1);
      }
    }
    return res;
  }
  vc<int>mev(int x){
  	auto fl=factor(x);
  	vc<int>res;
  	for(auto [a,z]:fl){
  		for(int i=res.size()-1;i>=0;i--)res.pb(res[i]*a);
  		res.pb(a);
  	}
  	return res;
  }
};
int main(){
	int n;cin>>n; int m=1e6;
	vc<int>v(n); rep(i,n)cin>>v[i];
	map<int,mint>mp;
	vc<int>mv(m+1,-1);
	divs d(m);
	for(ll au:d.primes)for(ll i=au*au;i<=m;i+=au*au)mv[i]=0;
	for(auto au:d.primes)for(int i=au;i<=m;i+=au)mv[i]*=-1;
	mint ans=0;
	rep(i,n){
		int a=v[i];
		mint now=1;
		for(auto au:d.mev(a))now+=mv[au]*mp[au];
		for(auto au:d.mev(a))mp[au]+=now;
		ans+=now;
	}
	cout<<ans.val();
}
0