1010 Sum Plus Product
题意
𝑛个数𝑎1, 𝑎2, . . . , 𝑎𝑛。每次随机选两个数𝑎, 𝑏合并成为𝑎𝑏 + 𝑎 + 𝑏,直到剩下一个数为止,求剩下数字的期望。答案对998244353取模𝑖。
分析
直接两两合并。
代码如下
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const LL maxn = 510, mod = 998244353;
int main(){
ios::sync_with_stdio(0), cin.tie(0),cout.tie(0);
int T; cin>>T;
while(T--){
int n; cin>>n;
LL x, y; cin>>x;
for(int i = 2; i <= n; i++){
cin>>y; x = (x+y+x*y%mod)%mod;
}
cout<<x<<"\n";
}
return 0;
}
1007 Matryoshka Doll
分析
𝑑𝑝(𝑥,𝑦)表示将前𝑥个套娃分成𝑦组的方案数。
转移时考虑第x个套娃放在前面的某一组(x-1,y),或者单独一组(x-1,y-1)。 即𝑑𝑝(𝑥,𝑦) = 𝑑𝑝(𝑥 − 1,𝑦 − 1) + 𝑑𝑝(𝑥 − 1,𝑦) · max{0,𝑦 − 𝑓 (𝑥)}, 𝑓(𝑥)表示满足1 ≤ 𝑧 <𝑥且𝑎𝑥 − 𝑟 < 𝑎𝑧 ≤ 𝑎𝑥的𝑧的个数, 即y组中不能放入x进行转移的组的数量。
只要知道有几个组可以转移就行了,不需要知道前面哪些组可以转移,也不需要维护前面每一组的信息,只要知道能转移的组的个数就行,不能转移的个数可以维护前面最后一个a[p]+r<=a[i]的数的位置p得到。
分成y组不需要知道每组的最大值,因为反正dp是考虑了所有情况的,肯定会包含那些不能进行转移的情况,只需要y把这些情况减掉就行了。
代码如下
//1007
#include<bits/stdc++.h>
typedef long long LL;
using namespace std;
const int maxn = 5050, mod = 998244353;
int a[maxn], dp[maxn][maxn]; //前i个数分成j组的方案数
int main(){
int T; cin>>T;
while(T--){
memset(dp,0,sizeof(dp));
int n, k, r; cin>>n>>k>>r;
for(int i = 1; i <= n; i++)cin>>a[i];
int p = 0; dp[0][0] = 1;
for(int i = 1; i <= n; i++){
while(p+1<i && a[p+1]+r<=a[i])p++; //p维护前面最后一个a[p]+r<=a[i]的数的位置
int s = i-p-1; //前面的不能转移的数的个数
for(int j = s+1; j <= i; j++){ //单独成组,或放在前面的某一组
dp[i][j] = (dp[i-1][j-1]+1ll*dp[i-1][j]*(j-s))%mod;
}
}
cout<<dp[n][k]<<"\n";
}
return 0;
}