結果

問題 No.1035 Color Box
ユーザー kilala
提出日時 2020-04-24 22:08:41
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,091 bytes
コンパイル時間 792 ms
コンパイル使用メモリ 85,920 KB
実行使用メモリ 18,752 KB
最終ジャッジ日時 2024-10-15 02:57:23
合計ジャッジ時間 7,230 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 2
other TLE * 1 -- * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
#include <map>
using namespace std;
typedef long long ll;
vector<ll> ans(100005, 1); long long M = 1e9 + 7;
map<pair<ll, ll>, ll> cmb;

long long qpow(long long a, long long n) {
    long long ret = 1;
    a = (a % M + M) % M;
    for (; n; n >>= 1) {
        if (n & 1) ret = (ret * a) % M;
        a = (a * a) % M;
    }
    return ret;
}


ll comp(ll a,ll b,ll p)//composite num C(a,b)%p
{
    if (cmb.count(make_pair(a,b))) return cmb[make_pair(a, b)];
    if (cmb.count(make_pair(a,a-b))) return cmb[make_pair(a, a-b)];
	if(a<b)return 0;
	if(a==b)return 1;
	if(b>a-b)b=a-b;
	int ans=1,ca=1,cb=1;
	for(ll i=0;i<b;i++)
	{
		ca=(ca*(a-i))%p;
		cb=(cb*(b-i))%p;
	}
	ans=(ca*qpow(cb,p-2))%p;
	cmb[make_pair(a, b)] = ans;
    return ans;
}

int main()
{
    int n, m;
    cin >> n >> m;
    ans[1] = m;
    ans[n] = qpow(m, n);
    int k = -1;
    for (int j = 1; j < m; ++j) {
        ans[n] += k * qpow(j, n) * comp(m, j, M) % M;
        ans[n] = (ans[n] + M) % M;
        k *= -1;
    }
    cout << ans[n] << endl;
    return 0;
}
0