#568098 (C++17) No.1259 スイッチ

• 問題: No.1259 スイッチ
• ユーザー: 57tggx
• 提出日時: 2020-10-16 23:23:12
• 言語: C++17 (gcc 13.2.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 544 ms / 2,000 ms
• コード長: 1,852 bytes
• コンパイル時間: 1,042 ms
• コンパイル使用メモリ: 71,620 KB
• 実行使用メモリ: 4,384 KB
• 最終ジャッジ日時: 2023-09-28 05:24:20
• 合計ジャッジ時間: 14,689 ms
• ジャッジサーバーID （参考情報）: judge14 / judge15

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 4,376 KB
testcase_01	AC	2 ms; 4,380 KB
testcase_02	AC	1 ms; 4,376 KB
testcase_03	AC	1 ms; 4,376 KB
testcase_04	AC	1 ms; 4,380 KB
testcase_05	AC	2 ms; 4,376 KB
testcase_06	AC	1 ms; 4,376 KB
testcase_07	AC	2 ms; 4,380 KB
testcase_08	AC	2 ms; 4,380 KB
testcase_09	AC	2 ms; 4,380 KB
testcase_10	AC	316 ms; 4,380 KB
testcase_11	AC	400 ms; 4,380 KB
testcase_12	AC	226 ms; 4,380 KB
testcase_13	AC	135 ms; 4,376 KB
testcase_14	AC	160 ms; 4,376 KB
testcase_15	AC	161 ms; 4,380 KB
testcase_16	AC	208 ms; 4,376 KB
testcase_17	AC	169 ms; 4,380 KB
testcase_18	AC	544 ms; 4,376 KB
testcase_19	AC	318 ms; 4,380 KB
testcase_20	AC	164 ms; 4,380 KB
testcase_21	AC	225 ms; 4,380 KB
testcase_22	AC	143 ms; 4,376 KB
testcase_23	AC	340 ms; 4,380 KB
testcase_24	AC	319 ms; 4,380 KB
testcase_25	AC	188 ms; 4,376 KB
testcase_26	AC	512 ms; 4,380 KB
testcase_27	AC	366 ms; 4,380 KB
testcase_28	AC	291 ms; 4,376 KB
testcase_29	AC	287 ms; 4,380 KB
testcase_30	AC	89 ms; 4,376 KB
testcase_31	AC	119 ms; 4,376 KB
testcase_32	AC	51 ms; 4,376 KB
testcase_33	AC	303 ms; 4,376 KB
testcase_34	AC	72 ms; 4,376 KB
testcase_35	AC	185 ms; 4,376 KB
testcase_36	AC	159 ms; 4,380 KB
testcase_37	AC	129 ms; 4,384 KB
testcase_38	AC	146 ms; 4,380 KB
testcase_39	AC	183 ms; 4,376 KB
testcase_40	AC	217 ms; 4,380 KB
testcase_41	AC	154 ms; 4,376 KB
testcase_42	AC	174 ms; 4,376 KB
testcase_43	AC	103 ms; 4,376 KB
testcase_44	AC	165 ms; 4,376 KB
testcase_45	AC	181 ms; 4,380 KB
testcase_46	AC	108 ms; 4,376 KB
testcase_47	AC	212 ms; 4,380 KB
testcase_48	AC	109 ms; 4,376 KB
testcase_49	AC	175 ms; 4,376 KB
testcase_50	AC	214 ms; 4,380 KB
testcase_51	AC	155 ms; 4,376 KB
testcase_52	AC	311 ms; 4,376 KB
testcase_53	AC	178 ms; 4,376 KB
testcase_54	AC	339 ms; 4,376 KB
testcase_55	AC	128 ms; 4,380 KB
testcase_56	AC	248 ms; 4,380 KB
testcase_57	AC	340 ms; 4,376 KB
testcase_58	AC	270 ms; 4,376 KB
testcase_59	AC	448 ms; 4,376 KB
testcase_60	AC	107 ms; 4,380 KB

ソースコード

#include <iostream>

template<class T, T MOD>
class Modint{
	T val;
public:
	Modint(T val = 0) : val(val % MOD + (val >= 0 ? 0 : MOD)) {}
	operator T(){ return val; }
	Modint &operator+=(const Modint &a){ val += a.val; if(val >= MOD) val -= MOD; return *this; }
	Modint &operator-=(const Modint &a){ val -= a.val; if(val < 0) val += MOD; return *this; }
	Modint &operator*=(const Modint &a){ val = val * a.val % MOD; return *this; }
	void inverse(){T x[2]={MOD,val},a[2]={0,1};int i;for(i=0;x[!i];i^=1){a[i]-=x[i]/x[!i]*a[!i];x[i]=x[i]%x[!i];}if(!i)a[i]+= MOD;val=a[i];}
	Modint &operator/=(Modint a){ a.inverse(); return *this *= a; }
	friend Modint modpow(Modint a, int n){ Modint ret(1); while(n){ if(n & 1) ret *= a; a *= a; n >>= 1; } return ret; }
	friend Modint operator+(Modint a, const Modint &b){ return a += b; }
	friend Modint operator-(Modint a, const Modint &b){ return a -= b; }
	friend Modint operator*(Modint a, const Modint &b){ return a *= b; }
	friend Modint operator/(Modint a, const Modint &b){ return a /= b; }
	friend std::ostream &operator<<(std::ostream &os, const Modint &a){ return os << a.val; }
};

using lint = long long;
using Mint = Modint<long long, 1000000007>;

int main(){
	lint n, k, m;
	std::cin >> n >> k >> m;
	if(m == 1){
		Mint x = modpow(Mint(n), n - 1);
		Mint y = 1;
		Mint ans = 0;
		for(int i = 1; i <= n; ++i){
			if(k % i == 0){
				ans += x * y;
			}
			x /= Mint(n);
			y *= Mint(n - i);
		}
		std::cout << ans << std::endl;
	}else{
		Mint x = modpow(Mint(n), n - 2);
		Mint y = 1;
		Mint ans = 0;
		for(int i = 1; i <= n - 1; ++i){
			ans += x * y * Mint(i);

			y *= Mint(n - i - 1);
			x /= Mint(n);
		}
		x = modpow(Mint(n), n - 2);
		y = 1;
		for(int i = 2; i <= n; ++i){
			if(k % i != 0){
				ans += x * y;
			}
			x /= Mint(n);
			y *= Mint(n - i);
		}

		std::cout << ans << std::endl;
	}
}