結果

問題 No.823 Many Shifts Easy
ユーザー yakamotoyakamoto
提出日時 2019-08-18 18:37:48
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 11 ms / 2,000 ms
コード長 3,168 bytes
コンパイル時間 1,772 ms
コンパイル使用メモリ 171,060 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-01 14:10:25
合計ジャッジ時間 2,516 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,816 KB
testcase_03 AC 11 ms
6,816 KB
testcase_04 AC 2 ms
6,820 KB
testcase_05 AC 8 ms
6,820 KB
testcase_06 AC 10 ms
6,820 KB
testcase_07 AC 2 ms
6,820 KB
testcase_08 AC 11 ms
6,816 KB
testcase_09 AC 3 ms
6,820 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/**
 * code generated by JHelper
 * More info: https://github.com/AlexeyDmitriev/JHelper
 * @author
 */

#include <iostream>
#include <fstream>

#ifndef SOLUTION_COMMON_H

#include <bits/stdc++.h>

using namespace std;

using ll = long long;
using PI = pair<int, int>;
template<class T> using V = vector<T>;
using VI = V<int>;
#define _1 first
#define _2 second

#ifdef MY_DEBUG
# define DEBUG(x) x
#else
# define DEBUG(x)
#endif

template<class T>
inline void debug(T &A) {
  DEBUG(
      for (const auto &a : A) {
        cerr << a << " ";
      }
      cerr << '\n';
  )
}

template<class T, class Func>
inline void debug_with_format(T &A, Func f) {
  DEBUG(
      for (const auto &a : A) {
        cerr << f(a) << " ";
      }
      cerr << '\n';
  )
}

template<class T>
inline void debug_dim2(T &A) {
  DEBUG(
      for (const auto &as : A) {
        debug(as);
      }
  )
}

template<typename ... Args>
inline void debug(const char *format, Args const &... args) {
  DEBUG(
      fprintf(stderr, format, args ...);
      cerr << '\n';
  )
}

template<typename ... Args>
string format(const string &fmt, Args ... args) {
  size_t len = snprintf(nullptr, 0, fmt.c_str(), args ...);
  vector<char> buf(len + 1);
  snprintf(&buf[0], len + 1, fmt.c_str(), args ...);
  return string(&buf[0], &buf[0] + len);
}

template<class T1, class T2>
string fmtP(pair<T1, T2> a) {
  stringstream ss;
  ss << "(" << a._1 << "," << a._2 << ")";
  return ss.str();
}

#define SOLUTION_COMMON_H

#endif //SOLUTION_COMMON_H


class Comb {
public:
  int MOD;
  V<ll> F, I;

  Comb(int N, int MOD): MOD(MOD), F(N + 1), I(N + 1) {
    F[0] = 1;
    for (int i = 1; i <= N; ++i) {
      F[i] = F[i - 1] * i % MOD;
    }

    I[N] = pow_mod(F[N], MOD - 2, MOD);
    for (int i = N - 1; i >= 0; --i) {
      I[i] = I[i + 1] * (i + 1) % MOD;
    }
  }

  static ll pow_mod(ll x, int k, int MOD) {
    ll res = k >= 2 ? pow_mod(x * x % MOD, k / 2, MOD) : 1ll;
    if (k&1) res = res * x % MOD;
    return res;
  };

  ll operator()(int n, int k) {
    if (k < 0 || k > n) return 0ll;
    return F[n] * I[n - k] % MOD * I[k] % MOD;
  };

  ll rev(int x) {
    return F[x - 1] * I[x] % MOD;
  }
};

const int MOD = 1000000007;

inline void add(ll &a, ll b) {
  a += b;
  if (a >= MOD) a -= MOD;
}

inline void mul(ll &a, ll b) {
  a = a * b % MOD;
}

class B {
public:
  void solve(std::istream& in, std::ostream& out) {
    int N, K;
    in >> N >> K;
    Comb comb(N + 1, MOD);
    ll ALL = comb.F[N];
    comb(ALL, comb.I[K]);

    debug("ALL:%d", ALL);

    ll ans = 0ll;
    for (int i = 1; i <= N; ++i) {
      // i, i+1 どちらもあるケース
      ll a = 0ll;
      if (N != i) {
        a = comb(N - 2, K - 2);
        mul(a, comb.F[K]);
        mul(a, comb.rev(2));
      }

      // iがないケース
      ll b = 0ll;
      if (N - 1 - K >= 0) {
        b = comb.F[N - 1];
        mul(b, comb.I[max(0, N - 1 - K)]);
      }

      debug("i:%d a:%d b:%d", i, a, b);

      add(a, b);
      mul(a, i);
      add(ans, a);
    }

    out << ans;
  }
};


int main() {
	B solver;
	std::istream& in(std::cin);
	std::ostream& out(std::cout);
	solver.solve(in, out);
	return 0;
}
0