結果

問題 No.300 平方数
コンテスト
ユーザー te-sh
提出日時 2017-06-12 13:37:02
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 9 ms / 1,000 ms
コード長 1,092 bytes
コンパイル時間 741 ms
コンパイル使用メモリ 111,272 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-12 19:57:26
合計ジャッジ時間 2,147 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 43
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.algorithm, std.conv, std.range, std.stdio, std.string;

void main()
{
  auto x = readln.chomp.to!long;
  auto pi = primes(10L ^^ 6);

  long[long] fi;
  while (x > 1) {
    auto f = factor(x, pi);
    ++fi[f];
    x /= f;
  }

  auto r = 1L;
  foreach (f, c; fi)
    if (c % 2) r *= f;

  writeln(r);
}

pure T[] primes(T)(T n)
{
  import std.algorithm, std.bitmanip, std.conv, std.range;

  auto sieve = BitArray();
  sieve.length((n + 1) / 2);
  sieve = ~sieve;

  foreach (p; 1..((nsqrt(n) - 1) / 2 + 1))
    if (sieve[p])
      for (auto q = p * 3 + 1; q < (n + 1) / 2; q += p * 2 + 1)
        sieve[q] = false;

  auto r = sieve.bitsSet.map!(to!T).map!("a * 2 + 1").array;
  r[0] = 2;

  return r;
}

pure T factor(T)(T n, const T[] pi)
{
  auto ma = nsqrt(n) + 1;
  foreach (p; pi)
    if (p > ma) return n;
    else if (n % p == 0) return p;
  return n;
}

pure T nsqrt(T)(T n)
{
  import std.algorithm, std.conv, std.range, core.bitop;
  if (n <= 1) return n;
  T m = 1 << (n.bsr / 2 + 1);
  return iota(1, m).map!"a * a".assumeSorted!"a <= b".lowerBound(n).length.to!T;
}
0