码迷,mamicode.com
首页 > 其他好文 > 详细

[ AGC002 F ] Leftmost Ball

时间:2021-05-23 23:13:50      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:iostream   main   wing   contest   ace   dia   lang   coder   ons   

题目

Atcoder

思路

技术图片
技术图片
技术图片

代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 4000010, mod = 1e9 + 7;
int n, k, fact[N], invf[N], f[2010][2010];
int qmi(int a, int b) {
	int res = 1;
	for (; b; b >>= 1, a = 1ll * a * a % mod)
		if (b & 1) res = 1ll * res * a % mod;
	return res;
}
int inv(int x) { return qmi(x, mod - 2); }
int C(int n, int m) { return 1ll * fact[n] * invf[m] % mod * invf[n - m] % mod; }
int main() {
	cin >> n >> k;
	fact[0] = 1, invf[0] = 1;
	if (k == 1) return cout << "1" << endl, 0;
	for (int i = 1; i <= 4000000; i++) 
		fact[i] = 1ll * fact[i - 1] * i % mod, invf[i] = inv(fact[i]);
	for (int i = 0; i <= n; i++) f[i][0] = 1;
	for (int i = 1; i <= n; i++) 
		for (int j = 1; j <= i; j++)
			f[i][j] = (f[i - 1][j] + 1ll * f[i][j - 1] * (n - j + 1) % mod * 
					  C(n * k - i - 1 - (j - 1) * (k - 1), k - 2) % mod) % mod;
	cout << f[n][n] << endl;
	return 0; 
}

[ AGC002 F ] Leftmost Ball

标签:iostream   main   wing   contest   ace   dia   lang   coder   ons   

原文地址:https://www.cnblogs.com/Proteinlzl/p/14727833.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!