結果

問題 No.271 next_permutation (2)
ユーザー te-shte-sh
提出日時 2017-08-03 10:06:27
言語 D
(dmd 2.107.1)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,972 bytes
コンパイル時間 334 ms
コンパイル使用メモリ 126,804 KB
最終ジャッジ日時 2023-09-03 15:36:09
合計ジャッジ時間 703 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ
（要ログイン）
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。（リジャッジ後のコンパイルエラーは公開されます）
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
/dmd2/linux/bin64/../../src/phobos/std/format/internal/write.d(164): Error: cannot implicitly convert expression `obj` of type `const(FactorRing!(1000000007, false))` to `int`
/dmd2/linux/bin64/../../src/phobos/std/format/write.d(1239): Error: template instance `std.format.internal.write.formatValueImpl!(LockingTextWriter, FactorRing!(1000000007, false), char)` error instantiating
/dmd2/linux/bin64/../../src/phobos/std/format/write.d(632):        instantiated from here: `formatValue!(LockingTextWriter, FactorRing!(1000000007, false), char)`
/dmd2/linux/bin64/../../src/phobos/std/stdio.d(1722):        instantiated from here: `formattedWrite!(LockingTextWriter, char, FactorRing!(1000000007, false))`
/dmd2/linux/bin64/../../src/phobos/std/stdio.d(4237):        instantiated from here: `write!(FactorRing!(1000000007, false), char)`
Main.d(19):        instantiated from here: `writeln!(FactorRing!(1000000007, false))`

ソースコード

diff # 

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

const mod = 10 ^^ 9 + 7;
alias FactorRing!mod mint;

void main()
{
  auto rd = readln.split, n = rd[0].to!size_t, k = rd[1].to!long;
  auto a = readln.split.to!(int[]);

  auto sfn = toFactorNumberSystem(a);
  auto efn = addToFactorNumberSystem(sfn, k);

  auto sr = calcDp(sfn), er = calcDp(efn[0..$-1]);

  auto lfn = new long[](n); lfn[$-1] = 1;
  auto lr = calcDp(lfn);

  writeln(er - sr + lr * mint(efn[$-1]));
}

auto toFactorNumberSystem(int[] a)
{
  auto n = a.length, bt = BiTree!int(n), r = new long[](n-1);
  bt[a[$-1]] += 1;

  foreach (i; 0..n-1) {
    auto ai = a[$-i-2];
    r[i] = bt[0..ai];
    bt[ai] += 1;
  }

  return r;
}

auto addToFactorNumberSystem(long[] fn, long k)
{
  auto n = fn.length, r = new long[](n+1);
  r[0..$-1][] = fn[];
  r[0] += k;

  foreach (i; 0..n) {
    r[i+1] += r[i]/(i+2);
    r[i] %= i+2;
  }

  return r;
}

auto calcDp(long[] fn)
{
  auto n = fn.length, dp1 = new mint[][](n+1, 2), dp2 = new mint[](n+1);

  foreach (i; 0..n) {
    auto fni = fn[$-i-1], dm = n-i;
    dp1[i+1][0] = dp1[i][0] * mint(dm+1) + dp2[i] * mint(dm*(dm+1)/2) + dp1[i][1] * fni + mint(fni*(fni-1)/2);
    dp1[i+1][1] = dp1[i][1] + fni;
    dp2[i+1] = dp2[i] * (dm+1) + fni;
  }

  return dp1[n][0];
}

struct BiTree(T)
{
  const size_t n;
  T[] buf;

  this(size_t n)
  {
    this.n = n;
    this.buf = new T[](n + 1);
  }

  void opIndexOpAssign(string op: "+")(T val, size_t i)
  {
    ++i;
    for (; i <= n; i += i & -i)
      buf[i] += val;
  }

  pure T opSlice(size_t r, size_t l) const
  {
    return get(l) - get(r);
  }

  pure size_t opDollar() const { return n; }
  pure T opIndex(size_t i) const { return opSlice(i, i+1); }

private:

  pure T get(size_t i) const
  {
    auto s = T(0);
    for (; i > 0; i -= i & -i)
      s += buf[i];
    return s;
  }
}

struct FactorRing(int m, bool pos = false)
{
  long v;

  @property int toInt() { return v.to!int; }
  alias toInt this;

  this(T)(T _v) { v = mod(_v); }

  ref FactorRing!(m, pos) opAssign(int _v)
  {
    v = mod(_v);
    return this;
  }

  pure auto mod(long _v) const
  {
    static if (pos) return _v % m;
    else return (_v % m + m) % m;
  }

  pure auto opBinary(string op: "+")(long rhs) const { return FactorRing!(m, pos)(v + rhs); }
  pure auto opBinary(string op: "-")(long rhs) const { return FactorRing!(m, pos)(v - rhs); }
  pure auto opBinary(string op: "*")(long rhs) const { return FactorRing!(m, pos)(v * rhs); }

  pure auto opBinary(string op)(FactorRing!(m, pos) rhs) const
    if (op == "+" || op == "-" || op == "*") { return opBinary!op(rhs.v); }

  auto opOpAssign(string op: "+")(long rhs) { v = mod(v + rhs); }
  auto opOpAssign(string op: "-")(long rhs) { v = mod(v - rhs); }
  auto opOpAssign(string op: "*")(long rhs) { v = mod(v * rhs); }

  auto opOpAssign(string op)(FactorRing!(m, pos) rhs)
    if (op == "+" || op == "-" || op == "*") { return opOpAssign!op(rhs.v); }
}
0