#568138 (C++14) No.1259 スイッチ

提出ソース

結果

問題	No.1259 スイッチ
ユーザー	suta28407928
提出日時	2020-10-16 23:35:56
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	69 ms / 2,000 ms
コード長	4,003 bytes
コンパイル時間	1,597 ms
コンパイル使用メモリ	112,208 KB
実行使用メモリ	27,264 KB
最終ジャッジ日時	2024-07-21 00:15:19
合計ジャッジ時間	6,862 ms
ジャッジサーバーID （参考情報）	judge1 / judge4

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	52 ms 27,136 KB
testcase_01	AC	50 ms 27,136 KB
testcase_02	AC	51 ms 27,232 KB
testcase_03	AC	50 ms 27,136 KB
testcase_04	AC	54 ms 27,136 KB
testcase_05	AC	51 ms 27,136 KB
testcase_06	AC	53 ms 27,136 KB
testcase_07	AC	52 ms 27,136 KB
testcase_08	AC	52 ms 27,136 KB
testcase_09	AC	49 ms 27,136 KB
testcase_10	AC	57 ms 27,140 KB
testcase_11	AC	60 ms 27,140 KB
testcase_12	AC	56 ms 27,212 KB
testcase_13	AC	58 ms 27,136 KB
testcase_14	AC	60 ms 27,136 KB
testcase_15	AC	56 ms 27,132 KB
testcase_16	AC	56 ms 27,136 KB
testcase_17	AC	57 ms 27,136 KB
testcase_18	AC	61 ms 27,096 KB
testcase_19	AC	60 ms 27,264 KB
testcase_20	AC	54 ms 27,092 KB
testcase_21	AC	58 ms 27,136 KB
testcase_22	AC	50 ms 27,136 KB
testcase_23	AC	61 ms 27,124 KB
testcase_24	AC	60 ms 27,192 KB
testcase_25	AC	56 ms 27,264 KB
testcase_26	AC	62 ms 27,096 KB
testcase_27	AC	66 ms 27,136 KB
testcase_28	AC	55 ms 27,264 KB
testcase_29	AC	59 ms 27,136 KB
testcase_30	AC	55 ms 27,172 KB
testcase_31	AC	60 ms 27,220 KB
testcase_32	AC	58 ms 27,136 KB
testcase_33	AC	61 ms 27,068 KB
testcase_34	AC	52 ms 27,136 KB
testcase_35	AC	59 ms 27,192 KB
testcase_36	AC	64 ms 27,228 KB
testcase_37	AC	64 ms 27,136 KB
testcase_38	AC	55 ms 27,072 KB
testcase_39	AC	60 ms 27,136 KB
testcase_40	AC	56 ms 27,072 KB
testcase_41	AC	63 ms 27,132 KB
testcase_42	AC	58 ms 27,136 KB
testcase_43	AC	56 ms 27,196 KB
testcase_44	AC	65 ms 27,132 KB
testcase_45	AC	64 ms 27,132 KB
testcase_46	AC	53 ms 27,136 KB
testcase_47	AC	61 ms 27,136 KB
testcase_48	AC	55 ms 27,096 KB
testcase_49	AC	64 ms 27,264 KB
testcase_50	AC	54 ms 27,096 KB
testcase_51	AC	57 ms 27,264 KB
testcase_52	AC	69 ms 27,116 KB
testcase_53	AC	62 ms 27,136 KB
testcase_54	AC	69 ms 27,136 KB
testcase_55	AC	59 ms 27,136 KB
testcase_56	AC	63 ms 27,220 KB
testcase_57	AC	65 ms 27,136 KB
testcase_58	AC	57 ms 27,136 KB
testcase_59	AC	58 ms 27,136 KB
testcase_60	AC	56 ms 27,264 KB

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

ソースコード

raw source code

#pragma GCC target("avx")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <iostream>
#include <string>
#include <cstdlib>
#include <cmath>
#include <vector>
#include <unordered_map>
#include <map>
#include <set>
#include <algorithm>
#include <queue>
#include <stack>
#include <functional>
#include <bitset>
#include <assert.h>
#include <unordered_map>
#include <fstream>
#include <ctime>
#include <complex>
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 pair<int,int> pii;
typedef vector<P> vpl;
typedef tuple<ll,ll,ll> tapu;
#define rep(i,n) for(int 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 = 1020000;
ll dy[8] = {1,-1,0,0,1,-1,1,-1};
ll dx[8] = {0,0,1,-1,1,-1,-1,1};
const double pi = acos(-1);
const double eps = 1e-10;
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 T> inline void print(T &a){
    for(auto itr = a.begin(); itr != a.end(); itr++){
		cout << *itr << " ";
	}
    cout << "\n";
}
template<typename T1,typename T2> inline void print2(T1 a, T2 b){
	cout << "debug: " << a << " " << b << "\n";
}
template<typename T1,typename T2,typename T3> inline void print3(T1 a, T2 b, T3 c){
	cout << "debug: " << a << " " << b << " " << c << "\n";
}
void mark() {cout << "#" << "\n";}
ll pcount(ll x) {return __builtin_popcountll(x);}
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,k,m; cin >> n >> k >> m;
	comInit();
	mint ans = 0;
	for(int i=n; i>=1; i--){
		if(k % i != 0) continue;
		mint cnt = com(n-1,i-1) * fac[i-1];
		cnt *= modpow(n,n-i,mod);
		ans += cnt;
	}
	if(m == 1) cout << ans.value() << "\n";
	else{
		mint as = modpow(n,n,mod);
		ans = (as - ans) / (n-1);
		cout << ans.value() << "\n";
	}
}

yukicoder

結果

テストケース

ソースコード