#798812 (C++14) No.2075 GCD Subsequence

• 問題: No.2075 GCD Subsequence
• ユーザー: tnakao0123
• 提出日時: 2022-09-18 12:11:29
• 言語: C++14 (gcc 15.2.0 + boost 1.89.0)
• 結果: WA
• 実行時間: -
• コード長: 3,012 bytes
• コンパイル時間: 1,419 ms
• コンパイル使用メモリ: 111,376 KB
• 実行使用メモリ: 11,184 KB
• 最終ジャッジ日時: 2024-12-22 01:14:56
• 合計ジャッジ時間: 5,294 ms
• ジャッジサーバーID （参考情報）: judge1 / judge5

ファイルパターン	結果
sample	AC * 3
other	AC * 3 WA * 25

ソースコード

/* -*- 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;
}