#996584 (C++17) No.1126 SUM

• 問題: No.1126 SUM
• ユーザー: elphe
• 提出日時: 2024-07-20 16:50:17
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 76 ms / 1,000 ms
• コード長: 743 bytes
• コンパイル時間: 547 ms
• コンパイル使用メモリ: 69,704 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-07-20 16:50:23
• 合計ジャッジ時間: 2,366 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	58 ms; 5,248 KB
testcase_01	AC	33 ms; 5,376 KB
testcase_02	AC	21 ms; 5,376 KB
testcase_03	AC	6 ms; 5,376 KB
testcase_04	AC	31 ms; 5,376 KB
testcase_05	AC	30 ms; 5,376 KB
testcase_06	AC	23 ms; 5,376 KB
testcase_07	AC	50 ms; 5,376 KB
testcase_08	AC	46 ms; 5,376 KB
testcase_09	AC	41 ms; 5,376 KB
testcase_10	AC	9 ms; 5,376 KB
testcase_11	AC	47 ms; 5,376 KB
testcase_12	AC	58 ms; 5,376 KB
testcase_13	AC	32 ms; 5,376 KB
testcase_14	AC	12 ms; 5,376 KB
testcase_15	AC	11 ms; 5,376 KB
testcase_16	AC	44 ms; 5,376 KB
testcase_17	AC	18 ms; 5,376 KB
testcase_18	AC	42 ms; 5,376 KB
testcase_19	AC	25 ms; 5,376 KB
testcase_20	AC	23 ms; 5,376 KB
testcase_21	AC	24 ms; 5,376 KB
testcase_22	AC	23 ms; 5,376 KB
testcase_23	AC	39 ms; 5,376 KB
testcase_24	AC	41 ms; 5,376 KB
testcase_25	AC	76 ms; 5,376 KB

ソースコード

#include <iostream>

#define MOD   1'000'000'007

using namespace std;

constexpr unsigned int pow_mod(const unsigned long long a, const unsigned long long b, const unsigned int mod)
{
	return (b == 0 ? (mod == 1 ? 0 : 1) : (b & 1 ? a % mod : 1) * pow_mod((a % mod) * (a % mod) % mod, b >> 1, mod) % mod);
}

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

	int N, M, i, cur = 1, ans, div;
	cin >> N >> M;
	for (i = 1; i <= N; ++i)
		cur = static_cast<long long>(cur) * i % MOD;

	div = ans = cur;
	for (i = N + 1; i <= M; ++i)
		ans = (ans + (cur = static_cast<long long>(cur) * pow_mod(i - N, MOD - 2, MOD) % MOD * i % MOD)) % MOD;

	cout << static_cast<long long>(ans) * pow_mod(div, MOD - 2, MOD) % MOD << '\n';
	return 0;
}