結果

問題 No.2075 GCD Subsequence
ユーザー KudeKude
提出日時 2022-09-16 22:13:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 300 ms / 4,000 ms
コード長 2,955 bytes
コンパイル時間 2,596 ms
コンパイル使用メモリ 233,636 KB
実行使用メモリ 12,404 KB
最終ジャッジ日時 2024-06-01 13:17:32
合計ジャッジ時間 8,180 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 14 ms
12,296 KB
testcase_01 AC 14 ms
12,208 KB
testcase_02 AC 15 ms
12,236 KB
testcase_03 AC 15 ms
12,324 KB
testcase_04 AC 15 ms
12,312 KB
testcase_05 AC 15 ms
12,240 KB
testcase_06 AC 14 ms
12,344 KB
testcase_07 AC 15 ms
12,304 KB
testcase_08 AC 75 ms
12,316 KB
testcase_09 AC 118 ms
12,148 KB
testcase_10 AC 81 ms
12,280 KB
testcase_11 AC 101 ms
12,340 KB
testcase_12 AC 93 ms
12,404 KB
testcase_13 AC 70 ms
12,304 KB
testcase_14 AC 100 ms
12,320 KB
testcase_15 AC 72 ms
12,180 KB
testcase_16 AC 77 ms
12,360 KB
testcase_17 AC 117 ms
12,300 KB
testcase_18 AC 287 ms
12,316 KB
testcase_19 AC 295 ms
12,148 KB
testcase_20 AC 300 ms
12,188 KB
testcase_21 AC 293 ms
12,304 KB
testcase_22 AC 296 ms
12,364 KB
testcase_23 AC 292 ms
12,244 KB
testcase_24 AC 296 ms
12,356 KB
testcase_25 AC 297 ms
12,280 KB
testcase_26 AC 300 ms
12,308 KB
testcase_27 AC 291 ms
12,288 KB
testcase_28 AC 247 ms
12,288 KB
testcase_29 AC 14 ms
12,328 KB
testcase_30 AC 14 ms
12,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
namespace {
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-function"
#include<atcoder/all>
#pragma GCC diagnostic pop
using namespace std;
using namespace atcoder;
#define rep(i,n) for(int i = 0; i < (int)(n); i++)
#define rrep(i,n) for(int i = (int)(n) - 1; i >= 0; i--)
#define all(x) begin(x), end(x)
#define rall(x) rbegin(x), rend(x)
template<class T> bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; } else return false; }
template<class T> bool chmin(T& a, const T& b) { if (b < a) { a = b; return true; } else return false; }
using ll = long long;
using P = pair<int,int>;
using VI = vector<int>;
using VVI = vector<VI>;
using VL = vector<ll>;
using VVL = vector<VL>;
using mint = modint998244353;

pair<vector<int>, vector<int>> primes_lpf(const int n) {
  vector<int> primes; primes.reserve(n / 10);
  vector<int> lpf(n + 1);
  for (int i = 2; i <= n; i += 2) lpf[i] = 2;
  for (int i = 3; i <= n; i += 6) lpf[i] = 3;
  if (2 <= n) primes.push_back(2);
  if (3 <= n) primes.push_back(3);
  // 5 * x <= n, x <= floor(n / 5)
  const int n5 = n / 5;
  int x = 5;
  char add_next = 2;
  for (; x <= n5; x += add_next, add_next ^= 2 ^ 4) {
    int px = lpf[x];
    if (px == 0) {
      lpf[x] = px = x;
      primes.push_back(x);
    }
    for (int i = 2;; ++i) {
      int q = primes[i];
      int y = q * x;
      if (y > n) break;
      lpf[y] = q;
      if (q == px) break;
    }
  }
  for (; x <= n; x += add_next, add_next ^= 2 ^ 4) {
    if (lpf[x] == 0) {
      lpf[x] = x;
      primes.push_back(x);
    }
  }
  return {move(primes), move(lpf)};
}

constexpr int PSIZE = 1000000;
auto [primes, lpf] = primes_lpf(PSIZE);

VI factorize(int x) {
  int ps[10], cs[10];
  int sz = 0;
  while (x != 1) {
    int p = lpf[x], c = 0;
    do {x /= p; c++;} while (x % p == 0);
    ps[sz] = p; cs[sz] = c; sz++;
  }
  VI fs(ps, ps + sz);
  // for (int i = 0; i < sz; i++) fs.emplace_back(ps[i], cs[i]);
  return fs;
}

vector<signed char> build_mu_table(const int n) {
  assert(n < (int)lpf.size());
  vector<signed char> res(n + 1);
  for(int p: primes) {
    if (p > n) break;
    res[p] = 1;
  }
  for(int x = 2; x <= n; x++) {
    if (!res[x]) continue;
    int px = lpf[x];
    for(int q: primes) {
      if (q == px) break;
      int y = q * x;
      if (y > n) break;
      res[y] = -res[x];
    }
  }
  return res;
}

auto mu = build_mu_table(1000000);

} int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  int n;
  cin >> n;
  vector<mint> dp(1000000);
  mint ans;
  rep(_, n) {
    int x;
    cin >> x;
    auto ps = factorize(x);
    int sz = ps.size();
    mint tot;
    rep(s, 1 << sz) if (s) {
      int t = 1;
      rep(k, sz) if (s >> k & 1) t *= ps[k];
      tot += dp[t] * mu[t];
    }
    tot++;
    ans += tot;
    rep(s, 1 << sz) if (s) {
      int t = 1;
      rep(k, sz) if (s >> k & 1) t *= ps[k];
      dp[t] += tot;
    }
  }
  cout << ans.val() << '\n';
}
0