結果

問題 No.300 平方数
ユーザー te-sh
提出日時 2017-06-12 13:23:55
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 8 ms / 1,000 ms
コード長 1,065 bytes
コンパイル時間 599 ms
コンパイル使用メモリ 101,676 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-12 19:57:23
合計ジャッジ時間 2,170 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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;
  const pi = primes(nsqrt(x));

  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.conv;

  auto sieve = new bool[]((n + 1) / 2);
  sieve[] = true;

  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;

  T[] r;
  foreach (i, s; sieve)
    if (s) r ~= (i * 2 + 1).to!T;
  r = 2 ~ r[1..$];

  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