【2019上海网络赛:D】Counting Sequences I(dfs+多重集合排列)
题目地址:https://nanti.jisuanke.com/t/41412
题目:
2≤n≤3000,求满足的排列的数目
解题思路:
通过简单的打表(n≤10),可以确定≤n,且必定存在的一个排列为,所以乘积最大为2n。
多重集合排列公式:,其实表示每个数重复的次数,n是数字的总数。
dfs枚举2~3000每个数字在整个序列中出现的次数(2*3000中最多有13个2,所以每个数字出现的次数很小),总出现次数记为num,乘积为mul,和为sum,那么满足题目要求的序列中1出现的次数num1=mul-sum,且只有当num1≥0&&num1+num≤3000时才能确定一个满足条件的序列, 该序列的n=num1+num,再利用多重集合排列公式求的排列数。
注意:排列式是除法的形式,会存在取模后分母大于分子的情况,需要用到逆元
ac代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 1e9+7;
const ll maxn = 3005;
int t, n;
ll f[maxn], INV[maxn], ans[maxn];
ll qpow(ll a, ll b)
{
ll ans = 1;
a = a % mod;
while(b)
{
if(b&1) ans = ans * a % mod;
a = a*a % mod;
b >>= 1;
}
return ans;
}
bool dfs(ll x, ll mul, ll sum, ll num, ll inv)//num:非1的位数
{
if(x==1)
{
ll num1 = mul - sum;
if(num1 >= 0 && num+num1 <= 3000)
{
ans[num+num1] = (ans[num+num1] + ((f[num+num1]*inv)%mod * INV[num1])%mod)%mod;
return true;
}
else return false;
}
dfs(x-1, mul, sum, num, inv);//无x
for(int i = 1; mul*pow(x,i) <= 2*3000; i++)//6000最多2^13
{
bool flag = dfs(x-1, mul*pow(x,i), sum+x*i, num+i, INV[i]*inv%mod);//i是x出现的次数
if(!flag) break;
}
return true;
}
int main()
{
//freopen("/Users/zhangkanqi/Desktop/11.txt","r",stdin);
f[0] = 1;
for(int i = 1; i <= 3000; i++) f[i] = f[i-1] * i % mod;
for(int i = 0; i <= 3000; i++) INV[i] = qpow(f[i], mod-2);//逆元,个数可能为0,也要处理INV[0]
dfs(3000, 1, 0, 0, 1);
scanf("%d", &t);
while(t--)
{
scanf("%d", &n);
printf("%lld\n", ans[n]);
}
return 0;
}