B擅长解密的小红同学
思路
因为每次输入会重置,求期望也就是求尝试多少次能成功,等价于求这些数字有多少种排列组合。
也就是多重排列数的问题。
多重排列数
设 S = { n 1 a 1 , n 2 a 2 , … , n k a k } S= \{n_1a_1,n_2a_2,…,n_ka_k\} S={n1a1,n2a2,…,nkak}是由 n 1 n_1 n1个 a 1 a_1 a1, n 2 n_2 n2个 a 2 a_2 a2, n k n_k nk个 a k a_k ak组成的多重集。
多重集排列数
多重集
S
S
S的全排列,不考虑相同的元素,其全排列个数为
(
∑
i
=
1
k
n
i
)
!
(\sum_{i=1}^kn_i)!
(∑i=1kni)!,但是因为存在若干个相同的元素,因此要把相同元素的贡献给除掉。
对于每种排列的每个
a
i
,
j
a_{i,j}
ai,j,我们都可以从
n
i
n_i
ni个
a
i
a_i
ai中挑选任意一个,其对答案的贡献满足乘法原理为
n
i
!
n_i!
ni!
因此方案数为
∑
i
=
1
k
n
i
!
∏
i
=
0
k
n
i
!
\frac{\sum_{i=1}^{k}n_i!}{\prod_{i=0}^{k}n_i!}
∏i=0kni!∑i=1kni!
注意: 该题
f
a
c
t
[
N
]
fact[N]
fact[N],
N
N
N能取到
1
0
7
10^7
107, 若
i
n
f
a
c
t
[
N
]
infact[N]
infact[N]也也算出来,复杂度达到
O
(
1
0
7
∗
l
o
g
(
1
0
6
)
)
O(10^7 * log(10^6))
O(107∗log(106)) 会超时, 所以infact[N]不需做枚举
代码
#include<iostream>
using namespace std;
const int mod = 1e9 + 7;
const int N = 1e7 + 10;
typedef long long LL;
int a[10];
LL fact[N], infact[N];
LL qmi(LL a, LL b){
LL res = 1;
while(b){
if(b & 1) res = (LL)res * a % mod;
a = (LL) a * a % mod;
b >>= 1;
}
return res;
}
void init(){
fact[0] = infact[0] = 1;
for(int i = 1; i < N; i ++){
fact[i] = (LL)fact[i - 1] * i % mod;
}
}
int main(){
for(int i = 0; i < 10; i ++)
cin >> a[i];
init();
LL fz = 0;
LL ans = 1;
for(int i = 0; i < 10; i ++){
fz = (LL)(fz + a[i]) % mod;
ans = (LL)ans * fact[a[i]] % mod;
}
ans = (LL)fact[fz] * qmi(ans, mod - 2) % mod;
cout << ans << endl;
return 0;
}
H 漂亮数
思路
通过两个质数相乘得到漂亮数, 可以直接打表并计算前缀和即可
代码
#include<iostream>
#include<cstring>
using namespace std;
const int N = 50000010;
typedef long long LL;
LL primes[N];
int cnt;
LL st[N];
int s[N * 2];
void init(){
for(int i = 2; i < N; i ++){
if(!st[i]) primes[cnt ++] = i;
for(int j = 0; (LL)primes[j] * i < N; j ++){
st[(LL)primes[j] * i] = 1;
if(i % primes[j] == 0) break;
}
}
for(int i = 0; i < cnt; i ++)
for(int j = 0; j < cnt; j ++){
if((LL)primes[i] * primes[j] >= N * 2) break;
s[(LL)primes[i] * primes[j]] = 1;
}
for(int i = 1; i <= N * 2; i ++)
s[i] += s[i - 1];
}
int main(){
init();
int T;
cin >> T;
while(T --){
int l, r;
cin >> l >> r;
cout << s[r] - s[l - 1] << endl;
}
return 0;
}