#469789 (C++14) No.1035 Color Box

• 問題: No.1035 Color Box
• ユーザー: どらら
• 提出日時: 2020-04-24 22:45:23
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 2,133 bytes
• コンパイル時間: 1,622 ms
• コンパイル使用メモリ: 170,336 KB
• 実行使用メモリ: 6,824 KB
• 最終ジャッジ日時: 2024-10-15 03:16:56
• 合計ジャッジ時間: 2,915 ms
• ジャッジサーバーID （参考情報）: judge4 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	6 ms; 6,816 KB
testcase_01	WA	-
testcase_02	WA	-
testcase_03	AC	6 ms; 6,820 KB
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	AC	5 ms; 6,820 KB
testcase_17	AC	6 ms; 6,820 KB
testcase_18	AC	6 ms; 6,820 KB
testcase_19	AC	5 ms; 6,816 KB
testcase_20	AC	6 ms; 6,816 KB
testcase_21	AC	5 ms; 6,820 KB
testcase_22	AC	6 ms; 6,820 KB
testcase_23	WA	-
testcase_24	WA	-
testcase_25	AC	5 ms; 6,820 KB
testcase_26	WA	-
testcase_27	AC	6 ms; 6,816 KB
testcase_28	WA	-
testcase_29	WA	-
testcase_30	AC	5 ms; 6,816 KB
testcase_31	AC	6 ms; 6,820 KB
testcase_32	WA	-
testcase_33	WA	-
testcase_34	WA	-
testcase_35	WA	-
testcase_36	WA	-
testcase_37	WA	-

ソースコード

#include <bits/stdc++.h>
using namespace std;
#define REP(i,a,n) for(int i=(a); i<(int)(n); i++)
#define rep(i,n) REP(i,0,n)
#define FOR(it,c) for(__typeof((c).begin()) it=(c).begin(); it!=(c).end(); ++it)
#define ALLOF(c) (c).begin(), (c).end()
typedef long long ll;
typedef unsigned long long ull;


static const long long mod = 1000000007;
struct mint {
  long long x;
  mint(ll x = 0):x(x%mod) {}
  mint& operator+=(const mint a) {
    (x += a.x) %= mod;
    return *this;
  }
  mint& operator-=(const mint a) {
    (x += mod-a.x) %= mod;
    return *this;
  }
  mint& operator*=(const mint a) {
    (x *= a.x) %= mod;
    return *this;
  }
  mint operator+(const mint a) const {
    mint ret(*this);
    return ret += a;
  }
  mint operator-(const mint a) const {
    mint ret(*this);
    return ret -= a;
  }
  mint operator*(const mint a) const {
    mint ret(*this);
    return ret *= a;
  }
  mint pow(ll t) const {
    if(t==0) return mint(1);
    mint a = pow(t>>1);
    a *= a;
    if(t&1) a *= *this;
    return a;
  }
  
  //for prime mod
  mint inv() const {
    return pow(mod-2);
  }
  mint& operator/=(const mint a) {
    return (*this) *= a.inv();
  }
  mint operator/(const mint a) const {
    mint ret(*this);
    return ret /= a;
  }
};
ostream &operator<<(ostream& os, const mint& x) {
  os << x.x;
  return os;
}
class modutils {
  vector<mint> fact, invfact;
public:
  modutils(int n = 200005):fact(n+1),invfact(n+1) {
    fact[0] = 1;
    for(int i=1; i<=n; i++) fact[i] = fact[i-1] * i;
    invfact[n] = fact[n].inv();
    for(int i=n; i>=1; i--) invfact[i-1] = invfact[i] * i;
  }
  mint pow(mint x, ll n) {
    return x.pow(n);
  }
  mint comb(ll n, ll k) {
    if(n<0 || k<0 || n<k) return 0;
    return fact[n] * invfact[k] * invfact[n-k];
  }
  mint perm(ll n, ll k) {
    if(n<0 || k<0 || n<k) return 0;
    return fact[n] * invfact[n-k];
  }
  mint fac(ll n) {
    return fact[n];
  }
};
modutils mutil;


int main(){
  int N, M;
  cin >> N >> M;

  mint ret = mutil.fac(M);
  ret *= mutil.comb(N, M);
  ret *= mutil.pow(M, N-M);
  ret *= mint(N-M+1).inv();

  cout << ret << endl;
  
  return 0;
}