#862464 (C++17) No.1035 Color Box

• 問題: No.1035 Color Box
• ユーザー: achapi
• 提出日時: 2023-04-23 23:08:30
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 20 ms / 2,000 ms
• コード長: 725 bytes
• コンパイル時間: 4,590 ms
• コンパイル使用メモリ: 257,136 KB
• 実行使用メモリ: 9,768 KB
• 最終ジャッジ日時: 2024-04-25 17:20:19
• 合計ジャッジ時間: 5,886 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	19 ms; 9,628 KB
testcase_01	AC	11 ms; 9,760 KB
testcase_02	AC	11 ms; 9,604 KB
testcase_03	AC	11 ms; 9,480 KB
testcase_04	AC	11 ms; 9,552 KB
testcase_05	AC	12 ms; 9,768 KB
testcase_06	AC	11 ms; 9,668 KB
testcase_07	AC	10 ms; 9,764 KB
testcase_08	AC	17 ms; 9,596 KB
testcase_09	AC	14 ms; 9,612 KB
testcase_10	AC	11 ms; 9,608 KB
testcase_11	AC	10 ms; 9,496 KB
testcase_12	AC	10 ms; 9,672 KB
testcase_13	AC	17 ms; 9,536 KB
testcase_14	AC	16 ms; 9,588 KB
testcase_15	AC	20 ms; 9,764 KB
testcase_16	AC	10 ms; 9,588 KB
testcase_17	AC	11 ms; 9,480 KB
testcase_18	AC	11 ms; 9,448 KB
testcase_19	AC	18 ms; 9,596 KB
testcase_20	AC	11 ms; 9,616 KB
testcase_21	AC	11 ms; 9,460 KB
testcase_22	AC	10 ms; 9,492 KB
testcase_23	AC	11 ms; 9,640 KB
testcase_24	AC	11 ms; 9,472 KB
testcase_25	AC	11 ms; 9,512 KB
testcase_26	AC	11 ms; 9,704 KB
testcase_27	AC	11 ms; 9,384 KB
testcase_28	AC	11 ms; 9,584 KB
testcase_29	AC	11 ms; 9,480 KB
testcase_30	AC	11 ms; 9,616 KB
testcase_31	AC	10 ms; 9,760 KB
testcase_32	AC	11 ms; 9,744 KB
testcase_33	AC	11 ms; 9,496 KB
testcase_34	AC	11 ms; 9,540 KB
testcase_35	AC	11 ms; 9,576 KB
testcase_36	AC	11 ms; 9,604 KB
testcase_37	AC	11 ms; 9,512 KB

ソースコード

#include <bits/stdc++.h>
#include <atcoder/all>
using mint = atcoder::modint1000000007;
using namespace std;
const int MAX = 510000;
const int MOD = 1000000007;

mint 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;
		inv[i] = MOD - inv[MOD%i] * (MOD / i);
		finv[i] = finv[i - 1] * inv[i];
	}
}

mint 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];
}

int main() {
	int n, m;
	cin >> n >> m;
	COMinit();
	vector<mint> f(n + 1);
	for (int i = 1; i <= m; i++){
		f[i] = COM(m, i) * (mint(i)).pow(n) - f[i - 1];
	}
	cout << f[m].val() << endl;
}