题目:
样例输入:
2
2
2 2
10
1 2 4 8 16 32 64 128 256 512
样例输出:
8
579063023
题意:给定n个数a1,a2,……,an。每次随机选两个数ai和aj合并成ai*aj+ai+aj,直到剩余一个数为止。求剩下数字的期望。答案对998244353取模。
分析:通过对a,b,c三个数进行分析,发现先合并a和b,再合并c得到的结果和先合并b和c最后合并a的结果都是a*b*c+a*b+a*c+b*c+a+b+c,那么就可以得知最后的结果和合并顺序并没有关系,他是一个固定值,所以我们直接for循环暴力合并即可。
代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<map>
#include<queue>
#include<vector>
#include<cmath>
using namespace std;
const int mod=998244353;
int main()
{
int T;
cin>>T;
while(T--)
{
int n;
scanf("%d",&n);
long long t,ans;
scanf("%lld",&ans);
for(int i=2;i<=n;i++)
{
scanf("%lld",&t);
ans=(ans*t+ans+t)%mod;
}
printf("%lld\n",ans);
}
return 0;
}