結果

問題 No.2075 GCD Subsequence
ユーザー tnakao0123tnakao0123
提出日時 2022-09-18 12:11:29
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,012 bytes
コンパイル時間 1,009 ms
コンパイル使用メモリ 106,616 KB
実行使用メモリ 11,196 KB
最終ジャッジ日時 2023-08-23 18:06:23
合計ジャッジ時間 5,412 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 13 ms
8,408 KB
testcase_01 AC 14 ms
8,352 KB
testcase_02 AC 13 ms
8,268 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 AC 96 ms
9,268 KB
testcase_29 AC 13 ms
8,276 KB
testcase_30 AC 14 ms
8,344 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 2075.cc:  No.2075 GCD Subsequence - yukicoder
 */

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<deque>
#include<algorithm>
#include<numeric>
#include<utility>
#include<complex>
#include<functional>
 
using namespace std;

/* constant */

const int MAX_N = 200000;
const int MAX_A = 1000000;
const int INF = 1 << 30;
const int MOD = 998244353;

/* typedef */

typedef vector<int> vi;
typedef set<int> si;

template<const int MOD>
struct MI {
  int v;
  MI(): v() {}
  MI(int _v): v(_v % MOD) {}
  MI(long long _v): v(_v % MOD) {}

  MI operator+(const MI m) const { return MI(v + m.v); }
  MI operator-(const MI m) const { return MI(v + MOD - m.v); }
  MI operator*(const MI m) const { return MI((long long)v * m.v); }

  MI &operator+=(const MI m) { return (*this = *this + m); }
  MI &operator-=(const MI m) { return (*this = *this - m); }
  MI &operator*=(const MI m) { return (*this = *this * m); }

  MI pow(int n) const {  // a^n % MOD
    MI pm = 1, a = *this;
    while (n > 0) {
      if (n & 1) pm *= a;
      a *= a;
      n >>= 1;
    }
    return pm;
  }

  MI inv() const { return pow(MOD - 2); }
  MI operator/(const MI m) const { return *this * m.inv(); }
  MI &operator/=(const MI m) { return (*this = *this / m); }
};

typedef MI<MOD> mi;
typedef pair<int,mi> pimi;
typedef map<int,pimi> mipimi;

/* global variables */

int pdivs[MAX_A + 1], as[MAX_N];
mi dp[MAX_N];

/* subroutines */

int gen_primes(int maxp, vi &pnums) {
  pdivs[0] = 0, pdivs[1] = 1;

  int p;
  for (p = 2; p * p <= maxp; p++)
    if (! pdivs[p]) {
      pnums.push_back(p);
      pdivs[p] = p;
      for (int q = p * p; q <= maxp; q += p)
	if (! pdivs[q]) pdivs[q] = p;
    }
  for (; p <= maxp; p++)
    if (! pdivs[p]) pnums.push_back(p), pdivs[p] = p;
  return (int)pnums.size();
}

void prime_decomp(int n, vi &pds) {
  pds.clear();
  while (n > 1) {
    int p = pdivs[n];
    pds.push_back(p);
    while (n % p == 0) n /= p;
  }
}

/* main */

int main() {
  vi pnums;
  int pn = gen_primes(MAX_A, pnums);
  //printf("pn=%d\n", pn);

  int n;
  scanf("%d", &n);
  for (int i = 0; i < n; i++) scanf("%d", as + i);

  mipimi las;
  mi sum = 0;

  for (int i = 0; i < n; i++) {
    vi pds;
    prime_decomp(as[i], pds);
    //printf("as[%d]=%d:", i, as[i]);
    //for (auto p: pds) printf(" %d", p); putchar('\n');

    dp[i] = 1;
    si prvs;
    for (auto p: pds) {
      mipimi::iterator mit = las.find(p);
      if (mit != las.end()) {
	int la = mit->second.first;
	if (prvs.find(la) == prvs.end()) {
	  dp[i] += mit->second.second;
	  prvs.insert(la);
	}
      }
    }

    for (auto p: pds) {
      mipimi::iterator mit = las.find(p);
      if (mit != las.end())
	mit->second = pimi(i, mit->second.second + dp[i]);
      else
	las[p] = pimi(i, dp[i]);
    }

    sum += dp[i];
  }

  printf("%d\n", sum.v);
  return 0;
}
0