結果

問題 No.802 だいたい等差数列
ユーザー tokoharu_procontokoharu_procon
提出日時 2019-03-17 22:17:28
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,086 ms / 2,000 ms
コード長 5,360 bytes
コンパイル時間 1,801 ms
コンパイル使用メモリ 170,452 KB
実行使用メモリ 21,052 KB
最終ジャッジ日時 2023-09-22 07:21:17
合計ジャッジ時間 17,162 ms
ジャッジサーバーID
(参考情報)
judge14 / judge12
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 85 ms
19,964 KB
testcase_01 AC 82 ms
21,052 KB
testcase_02 AC 85 ms
20,064 KB
testcase_03 AC 85 ms
19,604 KB
testcase_04 AC 86 ms
20,028 KB
testcase_05 AC 83 ms
20,288 KB
testcase_06 AC 86 ms
20,572 KB
testcase_07 AC 86 ms
20,664 KB
testcase_08 AC 84 ms
19,368 KB
testcase_09 AC 84 ms
19,900 KB
testcase_10 AC 255 ms
19,456 KB
testcase_11 AC 595 ms
19,488 KB
testcase_12 AC 571 ms
20,884 KB
testcase_13 AC 511 ms
19,436 KB
testcase_14 AC 651 ms
19,556 KB
testcase_15 AC 428 ms
19,540 KB
testcase_16 AC 607 ms
19,888 KB
testcase_17 AC 594 ms
19,860 KB
testcase_18 AC 494 ms
20,024 KB
testcase_19 AC 644 ms
19,492 KB
testcase_20 AC 1,083 ms
20,456 KB
testcase_21 AC 1,086 ms
19,856 KB
testcase_22 AC 596 ms
19,512 KB
testcase_23 AC 762 ms
20,300 KB
testcase_24 AC 599 ms
20,944 KB
testcase_25 AC 83 ms
19,808 KB
testcase_26 AC 596 ms
20,692 KB
testcase_27 AC 512 ms
20,688 KB
testcase_28 AC 756 ms
19,660 KB
testcase_29 AC 1,083 ms
19,432 KB
testcase_30 AC 82 ms
20,760 KB
testcase_31 AC 83 ms
20,996 KB
testcase_32 AC 83 ms
20,052 KB
testcase_33 AC 94 ms
21,008 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<LL, LL> PII;

// chmax, chmin
template <class T>
inline bool chmax(T& a, T b) {
  if (a < b) {
    a = b;
    return 1;
  }
  return 0;
}
template <class T>
inline bool chmin(T& a, T b) {
  if (a > b) {
    a = b;
    return 1;
  }
  return 0;
}

template <typename T>
function<T(T, T)> op_max = [](T a, T b) -> T { return max(a, b); };
template <typename T>
function<T(T, T)> op_min = [](T a, T b) -> T { return min(a, b); };

template <typename T>
function<T(T, T)> op_sum = [](T a, T b) -> T { return a + b; };

namespace detail {
template <typename T, std::size_t NDIMS>
struct vector_builder {
  using type = std::vector<typename vector_builder<T, NDIMS - 1>::type>;

  static type make(std::vector<std::size_t> dims, const T& v = {}) {
    const auto vec = std::vector<T>(dims.empty() ? 0 : dims.back(), v);
    if (!dims.empty()) dims.pop_back();
    return vector_builder<std::vector<T>, NDIMS - 1>::make(dims, vec);
  }
};

template <typename T>
struct vector_builder<T, 1> {
  using type = std::vector<T>;

  static type make(std::vector<std::size_t> dims, const T& v = {}) {
    return type(dims.empty() ? 0 : dims.back(), v);
  }
};
}  // namespace detail

template <typename T, typename... SIZE_T>
auto exvector(const T& v, SIZE_T... dims) {
  static_assert(sizeof...(dims) != 0, "invalid dimension");
  std::vector<std::size_t> vec_dims{dims...};
  return detail::vector_builder<T, sizeof...(dims)>::make(vec_dims, v);
}

using namespace std;
//********************  dumps ************************//

template <typename T>
void dump(const T& data, vector<int>& iter) {
  cout << data << " ";
}
template <typename T>
void dump(const vector<T>& data, vector<int>& iter) {
  for (auto elem : data) dump(elem, iter);
  cout << endl;
}

template <typename T>
void dump(const vector<vector<T>>& data, vector<int>& iter) {
  for (auto elem : iter) {
    cout << "[" << elem << "]";
  }
  cout << endl;
  for (int i = 0; i < data.size(); i++) {
    iter.push_back(i);
    dump(data[i], iter);
    iter.pop_back();
  }
  cout << endl;
}

template <typename T>
void dump(const vector<T>& data, int dummy) {
  for (int i = 0; i < data.size(); i++) {
    cout << "[" << i << "] " << data[i] << endl;
  }
}
template <typename T>
void dump(const T& data) {  // T : data, U = base type val.
  vector<int> iter;
  dump(data, iter);
}
///////////////////////////////////////////////

long long mul(long long a, long long b, const long long MOD) {
  return b ? (mul(a * 2, b / 2, MOD) + (b & 1 ? a : 0)) % MOD : 0;
}

long long bpow(long long a, long long b, const long long MOD) {
  return (b ? bpow(a * a % MOD, b / 2, MOD) * (b & 1 ? a : 1) : 1) % MOD;
}

long long inv(long long a, const long long MOD) {
  return bpow(a, MOD - 2, MOD);
}

vector<long long> MODS = {1000000007};
template <int kind = 0>
class mint {
 public:
  long long v;

  mint() : v(0) {}
  mint(long long v) : v((v % MODS[kind] + MODS[kind]) % MODS[kind]) {}
  long long get_mod() { return MODS[kind]; }
  long long get_val() { return v; }
};

template <int kind>
ostream& operator<<(ostream& os, const mint<kind>& x) {
  return os << x.v;
}

template <int kind>
mint<kind>& operator+=(mint<kind>& a, mint<kind> b) {
  return a = a.v + b.v;
}
template <int kind>
mint<kind>& operator-=(mint<kind>& a, mint<kind> b) {
  return a = a.v - b.v;
}
template <int kind>
mint<kind>& operator*=(mint<kind>& a, mint<kind> b) {
  return a = a.v * b.v;
}
template <int kind>
mint<kind>& operator/=(mint<kind>& a, mint<kind> b) {
  return a = a.v * inv(b.v, a.get_mod());
}
template <int kind>
mint<kind> operator+(mint<kind> a, mint<kind> b) {
  return a += b;
}
template <int kind>
mint<kind> operator-(mint<kind> a, mint<kind> b) {
  return a -= b;
}
template <int kind>
mint<kind> operator*(mint<kind> a, mint<kind> b) {
  return a *= b;
}
template <int kind>
mint<kind> operator/(mint<kind> a, mint<kind> b) {
  return a /= b;
}
template <int kind>
mint<kind>& operator+=(mint<kind>& a, long long b) {
  return a = a.v + b;
}
template <int kind>
mint<kind>& operator-=(mint<kind>& a, long long b) {
  return a = a.v - b;
}
template <int kind>
mint<kind>& operator*=(mint<kind>& a, long long b) {
  return a = a.v * b;
}
template <int kind>
mint<kind>& operator/=(mint<kind>& a, long long b) {
  return a = a.v * inv(b, a.get_mod());
}
template <int kind>
mint<kind> operator+(mint<kind> a, long long b) {
  return a += b;
}
template <int kind>
mint<kind> operator-(mint<kind> a, long long b) {
  return a -= b;
}
template <int kind>
mint<kind> operator*(mint<kind> a, long long b) {
  return a *= b;
}
template <int kind>
mint<kind> operator/(mint<kind> a, long long b) {
  return a /= b;
}

vector<mint<>> kaijo;
void init() {
  kaijo.push_back(mint<>(1));
  for (int i = 1; i <= 2000000; i++) {
    kaijo.push_back(kaijo.back() * i);
  }
}

mint<> comb(LL N, LL K) {
  if (K < 0) return mint<>(0);
  if (K > N) return mint<>(0);
  return kaijo[N] / kaijo[N - K] / kaijo[K];
}

int main() {
  init();
  mint<> ans;
  LL N, M, d1, d2;

  cin >> N >> M >> d1 >> d2;
  LL B = M - 1 - d1 * (N - 1);

  LL K = d2 - d1;
  for (LL i = 0; i <= N - 1; i++) {
    mint<> tmp = comb(N - 1, i) * comb(B - (K + 1) * i + N, N);
    if (i % 2 == 1) tmp = mint<>(0) - tmp;
    ans += tmp;
  }

  cout << ans << endl;
  return 0;
}
0