#102370 (C++11) No.391 CODING WAR

• 問題: No.391 CODING WAR
• ユーザー: tossy
• 提出日時: 2016-07-09 00:55:00
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 47 ms / 2,000 ms
• コード長: 1,742 bytes
• コンパイル時間: 637 ms
• コンパイル使用メモリ: 81,668 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-10-13 07:56:21
• 合計ジャッジ時間: 1,606 ms
• ジャッジサーバーID （参考情報）: judge3 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,248 KB
testcase_02	AC	1 ms; 5,248 KB
testcase_03	AC	2 ms; 5,248 KB
testcase_04	AC	1 ms; 5,248 KB
testcase_05	AC	2 ms; 5,248 KB
testcase_06	AC	1 ms; 5,248 KB
testcase_07	AC	2 ms; 5,248 KB
testcase_08	AC	1 ms; 5,248 KB
testcase_09	AC	47 ms; 5,248 KB
testcase_10	AC	45 ms; 5,248 KB
testcase_11	AC	2 ms; 5,248 KB
testcase_12	AC	2 ms; 5,248 KB
testcase_13	AC	42 ms; 5,248 KB
testcase_14	AC	34 ms; 5,248 KB
testcase_15	AC	38 ms; 5,248 KB
testcase_16	AC	23 ms; 5,248 KB
testcase_17	AC	28 ms; 5,248 KB
testcase_18	AC	19 ms; 5,248 KB
testcase_19	AC	19 ms; 5,248 KB

ソースコード

#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <cassert>
#include <iostream>
#include <algorithm>
#include <stack>
#include <queue>
#include <vector>
#include <set>
#include <map>
#include <bitset>
#include <fstream>
using namespace std;

#define repl(i,a,b) for(int i=(int)(a);i<(int)(b);i++)
#define rep(i,n) repl(i,0,n)
#define mp(a,b) make_pair(a,b)
#define pb(a) push_back(a)
#define all(x) (x).begin(),(x).end()
#define dbg(x) cout<<#x"="<<x<<endl
typedef long long ll;

#define INF 1000000000
#define MOD 1000000007

//long dp[100000];
long frac[100000];

// x^n mod
long mod_pow(long x, long n, long mod){
  long res=1;
  while(n>0){
    if(n&1) res=res*x%mod;
    x=x*x%mod;
    n>>=1;
  }
  return res;
}

/*
long comb(long n, long r){
    long p=1;
    for (long i=1; i<=r; i++) p=p*(n-i+1)/i;
    return p%MOD;
}
*/

long comb(long n, long m){
    long res =1;
    //for(long i=m+1;i<=n;i++) res = res*i%MOD;
    //long tmp=1;
    //for(long i=1; i<=n-m; i++) tmp = tmp*i%MOD; // (n-m)!
    res = frac[n]*mod_pow(frac[m], MOD-2, MOD)%MOD*mod_pow(frac[n-m], MOD-2, MOD)%MOD;
    return res;
}
/*
long calc(long n, long m){
    if(dp[m]!=-1) return dp[m];
    long res = mod_pow(m,n,MOD);
    for(long i=m-1; i>=1; i--){
        res = (res - comb(m,i)*calc(n, i) + MOD)%MOD;
    }
    return dp[m] = res;
}
*/

long calc(long n, long m){
    long res = 0;
    rep(i,m){
        res = (res+((i%2)?(-1):1)*comb(m,i)*mod_pow(m-i, n, MOD)+MOD)%MOD;
    }
    return res;
}

int main(){
    long n,m;
    cin>>n>>m;

    if(m>n){ cout<<0<<endl; return 0; }

    frac[0]=1;
    rep(i,m) frac[i+1]=(i+1)*frac[i]%MOD;

    /*fill(dp, dp+m+1, -1);
    dp[1]=1;*/
    cout<<calc(n,m)<<endl;

    return 0;
}