結果

問題 No.1035 Color Box
ユーザー wunderkammer2wunderkammer2
提出日時 2020-06-30 13:41:02
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 40 ms / 2,000 ms
コード長 2,892 bytes
コンパイル時間 992 ms
コンパイル使用メモリ 104,852 KB
実行使用メモリ 15,560 KB
最終ジャッジ日時 2024-07-23 18:54:24
合計ジャッジ時間 2,994 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
15,404 KB
testcase_01 AC 16 ms
15,472 KB
testcase_02 AC 14 ms
15,484 KB
testcase_03 AC 16 ms
15,284 KB
testcase_04 AC 18 ms
15,272 KB
testcase_05 AC 16 ms
15,464 KB
testcase_06 AC 15 ms
15,404 KB
testcase_07 AC 15 ms
15,560 KB
testcase_08 AC 35 ms
15,292 KB
testcase_09 AC 24 ms
15,364 KB
testcase_10 AC 15 ms
15,384 KB
testcase_11 AC 14 ms
15,388 KB
testcase_12 AC 14 ms
15,384 KB
testcase_13 AC 34 ms
15,320 KB
testcase_14 AC 37 ms
15,332 KB
testcase_15 AC 39 ms
15,296 KB
testcase_16 AC 15 ms
15,400 KB
testcase_17 AC 16 ms
15,416 KB
testcase_18 AC 16 ms
15,272 KB
testcase_19 AC 39 ms
15,320 KB
testcase_20 AC 16 ms
15,380 KB
testcase_21 AC 16 ms
15,320 KB
testcase_22 AC 16 ms
15,376 KB
testcase_23 AC 16 ms
15,288 KB
testcase_24 AC 16 ms
15,288 KB
testcase_25 AC 16 ms
15,268 KB
testcase_26 AC 15 ms
15,400 KB
testcase_27 AC 15 ms
15,312 KB
testcase_28 AC 16 ms
15,448 KB
testcase_29 AC 17 ms
15,468 KB
testcase_30 AC 16 ms
15,440 KB
testcase_31 AC 17 ms
15,280 KB
testcase_32 AC 17 ms
15,496 KB
testcase_33 AC 16 ms
15,428 KB
testcase_34 AC 16 ms
15,300 KB
testcase_35 AC 16 ms
15,432 KB
testcase_36 AC 16 ms
15,248 KB
testcase_37 AC 18 ms
15,368 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<algorithm>
#include<algorithm>     //sort,二分探索,など
#include<bitset>        //固定長bit集合
#include<cmath>         //pow,logなど
#include<complex>       //複素数
#include<deque>         //両端アクセスのキュー
#include<functional>    //sortのgreater
#include<iomanip>       //setprecision(浮動小数点の出力の誤差)
#include<iostream>      //入出力
#include<iterator>      //集合演算(積集合,和集合,差集合など)
#include<map>           //map(辞書)
#include<numeric>       //iota(整数列の生成),gcdとlcm(c++17)
#include<queue>         //キュー
#include<set>           //集合
#include<stack>         //スタック
#include<string>        //文字列
#include<unordered_map> //イテレータあるけど順序保持しないmap
#include<unordered_set> //イテレータあるけど順序保持しないset
#include<utility>       //pair
#include<vector>        //可変長配列

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef vector<ld> vld;
typedef vector<string> vs;
typedef vector<bool> vb;
typedef vector<vector<int>> vvi;
typedef vector<vector<ll>> vvll;
typedef vector<vector<ld>> vvld;
typedef vector<vector<string>> vvs;
typedef vector<vector<bool>> vvb;
const ll MOD = 1000000007;
const ll INF = 1000000000000000000;
#define rep(i,n) for(int i=0;i<n;i++)
#define repl(i,s,e) for(int i=s;i<e;i++)
#define reple(i,s,e) for(int i=s;i<=e;i++)
#define revrep(i,n) for(int i=n-1;i>=0;i--)
#define all(x) (x).begin(),(x).end()

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

ll pmod(ll n, ll p) {
    ll m = n % MOD;
    if (p == 0) return 1;
    ll rtn = pmod((m * m) % MOD, p / 2) % MOD;
    if (p % 2 == 1) rtn = rtn * m % MOD;
    return rtn;
}


const int MAX = 510000;

long long fac[MAX], finv[MAX], inv[MAX];

// テーブルを作る前処理
void COMinit() {
    fac[0] = fac[1] = 1;
    finv[0] = finv[1] = 1;
    inv[1] = 1;
    for (int i = 2; i < MAX; i++) {
        fac[i] = fac[i - 1] * i % MOD;
        inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD;
        finv[i] = finv[i - 1] * inv[i] % MOD;
    }
}

// 二項係数計算
long long COM(int n, int k) {
    if (n < k) return 0;
    if (n < 0 || k < 0) return 0;
    return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int N, M;
    cin >> N >> M;

    ll ans = 0;
    ans += pmod(M, N);

    COMinit();

    reple(i, 1, M - 1)
    {
		int a =  i % 2 == 0 ? 1 : -1;
		ans += a * COM(M, i) * pmod(M - i, N);
		ans %= MOD;
    }

	ans += MOD;
	ans %= MOD;

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