結果

問題 No.1001 注文の多い順列
ユーザー krotonkroton
提出日時 2020-03-02 16:30:17
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 1,287 ms / 2,000 ms
コード長 4,109 bytes
コンパイル時間 1,212 ms
コンパイル使用メモリ 101,000 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-04-21 22:47:46
合計ジャッジ時間 14,013 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 2 ms
5,376 KB
testcase_13 AC 3 ms
5,376 KB
testcase_14 AC 5 ms
5,376 KB
testcase_15 AC 8 ms
5,376 KB
testcase_16 AC 4 ms
5,376 KB
testcase_17 AC 4 ms
5,376 KB
testcase_18 AC 535 ms
5,376 KB
testcase_19 AC 506 ms
5,376 KB
testcase_20 AC 313 ms
5,376 KB
testcase_21 AC 291 ms
5,376 KB
testcase_22 AC 684 ms
5,376 KB
testcase_23 AC 654 ms
5,376 KB
testcase_24 AC 652 ms
5,376 KB
testcase_25 AC 629 ms
5,376 KB
testcase_26 AC 1,230 ms
5,376 KB
testcase_27 AC 1,181 ms
5,376 KB
testcase_28 AC 1,107 ms
5,376 KB
testcase_29 AC 900 ms
5,376 KB
testcase_30 AC 860 ms
5,376 KB
testcase_31 AC 932 ms
5,376 KB
testcase_32 AC 41 ms
5,376 KB
testcase_33 AC 1,287 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <algorithm>
#include <bitset>
#include <cassert>
#include <cctype>
#include <cmath>
#include <cstdint>
#include <cstdio>
#include <cstring>
#include <deque>
#include <fstream>
#include <functional>
#include <iostream>
#include <limits>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <vector>
using namespace std;
using ll = long long;
#define fst first
#define snd second

/* clang-format off */
template <class T, size_t D> struct _vec { using type = vector<typename _vec<T, D - 1>::type>; };
template <class T> struct _vec<T, 0> { using type = T; };
template <class T, size_t D> using vec = typename _vec<T, D>::type;
template <class T> vector<T> make_v(size_t size, const T& init) { return vector<T>(size, init); }
template <class... Ts> auto make_v(size_t size, Ts... rest) { return vector<decltype(make_v(rest...))>(size, make_v(rest...)); }
template <class T> inline void chmin(T &a, const T& b) { if (b < a) a = b; }
template <class T> inline void chmax(T &a, const T& b) { if (b > a) a = b; }
/* clang-format on */

template <class T>
vector<T> sumup(const vector<T>& a) {
  int N = a.size();
  vector<T> res(N + 1);
  for (int i = 0; i < N; i++) res[i + 1] = res[i] + a[i];
  return res;
}

template <std::uint_fast64_t Modulus>
class modint {
  using u64 = std::uint_fast64_t;

 public:
  u64 a;

  constexpr modint(const u64 x = 0) noexcept
      : a(x % Modulus) {
  }
  constexpr u64 &value() noexcept {
    return a;
  }
  constexpr const u64 &value() const noexcept {
    return a;
  }
  constexpr modint operator+(const modint rhs) const noexcept {
    return modint(*this) += rhs;
  }
  constexpr modint operator-(const modint rhs) const noexcept {
    return modint(*this) -= rhs;
  }
  constexpr modint operator*(const modint rhs) const noexcept {
    return modint(*this) *= rhs;
  }
  constexpr modint operator/(const modint rhs) const noexcept {
    return modint(*this) /= rhs;
  }
  constexpr modint &operator+=(const modint rhs) noexcept {
    a += rhs.a;
    if (a >= Modulus) {
      a -= Modulus;
    }
    return *this;
  }
  constexpr modint &operator-=(const modint rhs) noexcept {
    if (a < rhs.a) {
      a += Modulus;
    }
    a -= rhs.a;
    return *this;
  }
  constexpr modint &operator*=(const modint rhs) noexcept {
    a = a * rhs.a % Modulus;
    return *this;
  }
  constexpr modint &operator/=(const modint rhs) noexcept {
    return *this *= ~rhs;
  }
  constexpr modint power(u64 exp) const noexcept {
    modint v = 1, x = *this;
    while (exp) {
      if (exp & 1) {
        v *= x;
      }
      x *= x;
      exp >>= 1;
    }
    return v;
  }
  constexpr modint operator~() const noexcept {
    return power(Modulus - 2);
  }
};

using mint = modint<1000000007>;

vector<mint> fact, factInv;
void initFact(int N) {
  fact.resize(N + 1);
  factInv.resize(N + 1);
  fact[0] = 1;
  for (int i = 1; i <= N; i++) fact[i] = fact[i - 1] * i;
  factInv[N] = ~fact[N];
  for (int i = N - 1; i >= 0; i--) factInv[i] = factInv[i + 1] * (i + 1);
}

int main() {
#ifdef DEBUG
  ifstream ifs("in.txt");
  cin.rdbuf(ifs.rdbuf());
#endif
  int N;
  while (cin >> N) {
    initFact(4000);
    vector<int> X(N, 0), Y(N, 0);
    int xs = 0, ys = 0;
    for (int i = 0; i < N; i++) {
      int t, x;
      cin >> t >> x;
      --x;
      if (t == 0) {
        ++X[x];
        ++xs;
      } else {
        ++Y[x];
        ++ys;
      }
    }
    auto sumX = sumup(X), sumY = sumup(Y);
    vector<mint> dp(N + 1, 0);
    dp[0] = 1;
    for (int p = 0; p < N; p++) {
      vector<mint> ndp(N + 1, 0);
      for (int y = 0; y <= p; y++) {
        for (int add = 0; add < 2; add++) {
          int ny = y + add;
          mint way = 1;
          if (add) {
            way *= max(0, sumY[p + 1] - y);
          } 
          if (p + 1 - ny - sumX[p + 1] >= 0) {
            way *= fact[p + 1 - ny - sumX[p]];
            way /= fact[p + 1 - ny - sumX[p + 1]];
          } else {
            way = 0;
          }
          ndp[ny] += dp[y] * way;
        }
      }
      dp.swap(ndp);
    }
    cout << dp[ys].value() << endl;
  }
  return 0;
}
0