結果

問題 No.1035 Color Box
ユーザー maimaicorgimaimaicorgi
提出日時 2022-09-13 20:54:36
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 62 ms / 2,000 ms
コード長 3,481 bytes
コンパイル時間 1,324 ms
コンパイル使用メモリ 141,032 KB
実行使用メモリ 4,800 KB
最終ジャッジ日時 2023-08-20 22:20:42
合計ジャッジ時間 4,673 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 61 ms
4,556 KB
testcase_01 AC 42 ms
4,700 KB
testcase_02 AC 41 ms
4,732 KB
testcase_03 AC 43 ms
4,644 KB
testcase_04 AC 44 ms
4,592 KB
testcase_05 AC 43 ms
4,552 KB
testcase_06 AC 42 ms
4,696 KB
testcase_07 AC 42 ms
4,732 KB
testcase_08 AC 59 ms
4,628 KB
testcase_09 AC 48 ms
4,488 KB
testcase_10 AC 42 ms
4,528 KB
testcase_11 AC 42 ms
4,600 KB
testcase_12 AC 42 ms
4,628 KB
testcase_13 AC 56 ms
4,652 KB
testcase_14 AC 59 ms
4,592 KB
testcase_15 AC 62 ms
4,596 KB
testcase_16 AC 42 ms
4,640 KB
testcase_17 AC 42 ms
4,560 KB
testcase_18 AC 42 ms
4,560 KB
testcase_19 AC 62 ms
4,588 KB
testcase_20 AC 41 ms
4,584 KB
testcase_21 AC 41 ms
4,572 KB
testcase_22 AC 42 ms
4,524 KB
testcase_23 AC 42 ms
4,700 KB
testcase_24 AC 42 ms
4,800 KB
testcase_25 AC 42 ms
4,696 KB
testcase_26 AC 42 ms
4,652 KB
testcase_27 AC 42 ms
4,644 KB
testcase_28 AC 42 ms
4,592 KB
testcase_29 AC 42 ms
4,592 KB
testcase_30 AC 42 ms
4,648 KB
testcase_31 AC 42 ms
4,492 KB
testcase_32 AC 41 ms
4,704 KB
testcase_33 AC 42 ms
4,608 KB
testcase_34 AC 41 ms
4,744 KB
testcase_35 AC 41 ms
4,672 KB
testcase_36 AC 42 ms
4,492 KB
testcase_37 AC 42 ms
4,488 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <iomanip>
#include <string>
#include <vector>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <ios>
#include <iomanip>
#include <queue>
#include <numeric>
#include <map>
#include <set>
#include <sstream>
#include <bitset>
#include <climits>
#include <stack>
#include <regex>
#include <functional>
#include <random>
#ifdef _WIN64
#  include <atcoder/all>
#endif
#ifdef _MSC_VER
#  include <intrin.h>
#  define __builtin_popcount __popcnt
#  define __builtin_popcountl __popcnt64
#endif

using namespace std;

#define ll long long
#define rep(i, init, n) for(ll i = init; i < (ll)n; i++)
#define rrep(i, init, n) for(ll i = init; i >= (ll)n; i--)
#define all(x) (x).begin(), (x).end()
#define sz(x) (ll)(x.size())
#define Out(x) cout << x << endl
#define Yes cout << "Yes" << endl
#define No cout << "No" << endl
#define Ans cout << ans << endl
#define PI 3.14159265358979
#define MOD 1000000007

const int inf32 = INT_MAX / 2;
const ll inf64 = 1LL << 60;
template<class T>bool chmax(T &a, const T &b) { if (a < b) { a = b; return true; } return false; }
template<class T>bool chmin(T &a, const T &b) { if (a > b) { a = b; return true; } return false; }

// -------------------------------------------------------------------------------------------------

struct mint
{
	ll x;

	mint(ll val = 0) { x = (val % MOD + MOD) % MOD; }

	mint operator-() const { return mint(-x); }
	mint& operator+=(const mint& other) { x += other.x; if (x >= MOD) x -= MOD; return *this; }
	mint& operator-=(const mint& other) { x -= other.x; if (x < 0) x += MOD; return *this; }
	mint& operator*=(const mint& other) { (x *= other.x) %= MOD; return *this; }
	mint& operator/=(const mint& other) { *this *= other.pow(MOD - 2); return *this; }
	mint operator+(const mint& other) const { return mint(*this) += other; }
	mint operator-(const mint& other) const { return mint(*this) -= other; }
	mint operator*(const mint& other) const { return mint(*this) *= other; }
	mint operator/(const mint& other) const { return mint(*this) /= other; }

	mint pow(ll i) const
	{
		mint ret = 1;
		for (mint tmp = x; i; tmp *= tmp, i >>= 1) if (i & 1) ret *= tmp;
		return ret;
	}

	friend ostream& operator<<(ostream& os, const mint& m) { os << m.x; return os; }
};
struct Combination
{
	vector<ll> f, finv;
	bool precalc = false;

	Combination(ll n, bool calc_inv = false)
	{
		f.resize(n + 1);
		if (calc_inv) finv.resize(n + 1), precalc = true;

		f[0] = 1;
		rep(i, 1, n + 1) f[i] = f[i - 1] * i % MOD;
		if (calc_inv) rep(i, 0, n + 1) finv[i] = mpow(f[i], MOD - 2);
	}

	ll mpow(ll x, ll y)
	{
		if (y == 0) return 1;

		x %= MOD;
		if (x == 0) return 0;

		if (y % 2 == 0) return mpow(x * x % MOD, y / 2) % MOD;
		else return x * mpow(x, y - 1) % MOD;
	}

	ll P(ll x, ll y) { return precalc ? f[x] * finv[x - y] % MOD : f[x] * mpow(f[x - y], MOD - 2) % MOD; }

	ll C(ll x, ll y)
	{
		ll a = f[x];
		ll b = precalc ? finv[y] * finv[x - y] % MOD : mpow(f[y] * f[x - y] % MOD, MOD - 2);

		return a * b % MOD;
	}

	ll H(ll x, ll y) { return C(x + y - 1, y); }
};
ll mpow(ll x, ll y)
{
	if (y == 0) return 1;

	x %= MOD;
	if (x == 0) return 0;

	if (y % 2 == 0) return mpow(x * x % MOD, y / 2) % MOD;
	else return x * mpow(x, y - 1) % MOD;
}

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

	Combination comb(100000, true);
	mint ans = mpow(m, n);

	rep(i, 1, m)
	{
		mint tmp = (mint)comb.C(m, i) * mpow(i, n);
		ans += tmp * ((m - i) % 2 != 0 ? -1 : 1);
	}

	Ans;

	return 0;
}
0