#469535 (C++14) No.1035 Color Box

提出ソース

結果

問題	No.1035 Color Box
ユーザー	suta28407928
提出日時	2020-04-24 22:08:44
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	23 ms / 2,000 ms
コード長	3,370 bytes
コンパイル時間	647 ms
コンパイル使用メモリ	93,524 KB
実行使用メモリ	15,296 KB
最終ジャッジ日時	2024-04-27 03:11:05
合計ジャッジ時間	2,192 ms
ジャッジサーバーID （参考情報）	judge3 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	23 ms 15,140 KB
testcase_01	AC	16 ms 15,260 KB
testcase_02	AC	16 ms 15,176 KB
testcase_03	AC	15 ms 15,264 KB
testcase_04	AC	15 ms 15,116 KB
testcase_05	AC	15 ms 15,240 KB
testcase_06	AC	16 ms 15,156 KB
testcase_07	AC	15 ms 15,144 KB
testcase_08	AC	21 ms 15,132 KB
testcase_09	AC	18 ms 15,132 KB
testcase_10	AC	16 ms 15,124 KB
testcase_11	AC	15 ms 15,188 KB
testcase_12	AC	15 ms 15,240 KB
testcase_13	AC	21 ms 15,196 KB
testcase_14	AC	20 ms 15,148 KB
testcase_15	AC	21 ms 15,152 KB
testcase_16	AC	15 ms 15,148 KB
testcase_17	AC	15 ms 15,156 KB
testcase_18	AC	15 ms 15,192 KB
testcase_19	AC	23 ms 15,176 KB
testcase_20	AC	16 ms 15,176 KB
testcase_21	AC	15 ms 15,180 KB
testcase_22	AC	15 ms 15,192 KB
testcase_23	AC	17 ms 15,156 KB
testcase_24	AC	16 ms 15,252 KB
testcase_25	AC	17 ms 15,240 KB
testcase_26	AC	15 ms 15,120 KB
testcase_27	AC	16 ms 15,208 KB
testcase_28	AC	16 ms 15,188 KB
testcase_29	AC	16 ms 15,208 KB
testcase_30	AC	16 ms 15,208 KB
testcase_31	AC	18 ms 15,116 KB
testcase_32	AC	15 ms 15,296 KB
testcase_33	AC	15 ms 15,148 KB
testcase_34	AC	16 ms 15,172 KB
testcase_35	AC	15 ms 15,192 KB
testcase_36	AC	15 ms 15,124 KB
testcase_37	AC	16 ms 15,204 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <iostream>
#include <string>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <map>
#include <set>
#include <algorithm>
#include <queue>
#include <stack>
#include <functional>
#include <bitset>
#include <assert.h>
using namespace std;
typedef long long ll;
typedef vector<ll> vl;
typedef vector<vl> vvl;
typedef vector<char> vc;
typedef vector<string> vs;
typedef vector<bool> vb;
typedef vector<double> vd;
typedef pair<ll,ll> P;
typedef vector<P> vpl;
typedef tuple<ll,ll,ll> tapu;
#define rep(i,n) for(ll i=0; i<(n); i++)
#define REP(i,a,b) for(int i=(a); i<(b); i++)
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
const int inf = 1<<30;
const ll linf = 1LL<<62;
const int MAX = 510000;
ll dy[8] = {0,1,0,-1,1,-1,1,-1};
ll dx[8] = {1,0,-1,0,1,-1,-1,1};
const double pi = acos(-1);
const double eps = 1e-7;
template<typename T1,typename T2> inline bool chmin(T1 &a,T2 b){
	if(a>b){
		a = b; return true;
	}
	else return false;
}
template<typename T1,typename T2> inline bool chmax(T1 &a,T2 b){
	if(a<b){
		a = b; return true;
	}
	else return false;
}
template<typename T1,typename T2> inline void print2(T1 a, T2 b){cout << a << " " << b << "\n";}
template<typename T1,typename T2,typename T3> inline void print3(T1 a, T2 b, T3 c){
	cout << a << " " << b << " " << c << "\n";
}
const int mod = 1e9 + 7;
//const int mod = 998244353;

template <std::uint_fast64_t Modulus> class modint {
  using u64 = std::uint_fast64_t;

public:
  u64 a;

  constexpr modint(const u64 x = 0) noexcept : a(x % Modulus) {}
  constexpr u64 &value() noexcept { return a; }
  constexpr const u64 &value() const noexcept { return a; }
  constexpr modint operator+(const modint rhs) const noexcept {
	return modint(*this) += rhs;
  }
  constexpr modint operator-(const modint rhs) const noexcept {
	return modint(*this) -= rhs;
  }
  constexpr modint operator*(const modint rhs) const noexcept {
	return modint(*this) *= rhs;
  }
  constexpr modint operator/(const modint rhs) const noexcept {
	return modint(*this) /= rhs;
  }
  constexpr modint &operator+=(const modint rhs) noexcept {
	a += rhs.a;
	if (a >= Modulus) {
	  a -= Modulus;
	}
	return *this;
  }
  constexpr modint &operator-=(const modint rhs) noexcept {
	if (a < rhs.a) {
	  a += Modulus;
	}
	a -= rhs.a;
	return *this;
  }
  constexpr modint &operator*=(const modint rhs) noexcept {
	a = a * rhs.a % Modulus;
	return *this;
  }
  constexpr modint &operator/=(modint rhs) noexcept {
	u64 exp = Modulus - 2;
	while (exp) {
	  if (exp % 2) {
		*this *= rhs;
	  }
	  rhs *= rhs;
	  exp /= 2;
	}
	return *this;
  }
};

using mint = modint<mod>;

vector<ll> fac(MAX), finv(MAX), inv(MAX);

void comInit(){
	fac[0] = fac[1] = 1;
	finv[0] = finv[1] = 1;
	inv[1] = 1;
	for(ll 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;
	}
}


ll com(ll n, ll k){
	if(n < k) return 0;
	if(n < 0 || k < 0) return 0;
	return fac[n] * (finv[k] * finv[n-k] % mod) % mod;
}

ll modpow(ll x, ll n, ll mod){
	ll res = 1;
	while(n > 0){
		if(n & 1) res = res * x % mod;
		x = x * x % mod;
		n >>= 1;
	}
	return res;
}

int main(){
	ll n,m; cin >> n >> m;
	mint ans = 0;
	comInit();
	rep(i,m+1){
		mint tmp = com(m,i) * modpow(i,n,mod);
		if((m-i) & 1) ans -= tmp;
		else ans += tmp;
	}
	cout << ans.value() << endl;
}

yukicoder

結果

テストケース

ソースコード