HDU 6267 Master of Random (CCPC2017杭州) 求期望 找规律+递推

http://acm.hdu.edu.cn/showproblem.php?pid=6267

题目大意：$T$组数据，每组数据$n$个点，给出每个点的权值$a_i$，第$i$个点可以随机选取$[0,i-1]$中的一个点作为它的父亲节点，这样我们可以得到一颗树，我们从树中任意的选取一个节点并计算它的子树(包括本身)的各个节点的权值之和作为答案，现在我想知道这个答案的期望。因为答案可能是个分数$p/q$，所以请输出：$p\ast q^{-1}\%mod$，其中$mod=998244353$。

思路：应该是第一次补这种题目吧，这题确实不难，但是训练赛推错了$QAQ$(太菜了)。其实在图上画一下$n=4$的情况就能找到规律辣，根据乘法原理，一共可以构造出：$1\ast 2\ast 3=(n-1)!$ 棵不同的树，因为每棵树都有$n$个节点可以选择，所以一共有$n!$ 种不同的情况，然后手算一下每个点的权值计算了多少次，可以发现：(1)节点$0$计算了$6$次；(2)节点$1$计算了$12$次；(3)节点$2$计算了$15$次；(4)节点$3$计算了$17$次；又有：(1) $6=3!$，(2) $12=3!+3!/1$，(3) $15=3!+3!/1+3!/2$，(4) $17=3!+3!/1+3!/2+3!/3$，这不就发现规律了嘛。逆元也很好求，因为$mod$是一个质数，所以$q^{-1}\%mod=q^{mod-2}$，然后这题就可以做了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<unordered_map>
#define pr pair<int,int>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;

const ll mod=998244353;
const int maxn=1e5+5;

ll jc[maxn];
int n,t;

ll qpow(ll a,ll b)//快速幂
{
	ll ans=1;
	while(b)
	{
		if(b&1)
			ans=ans*a%mod;
		a=a*a%mod;
		b>>=1;
	}
	return ans;
}

inline ll inv(ll a)//逆元
{
	return qpow(a,mod-2);
}

int main()
{
	jc[0]=1;
	for(int i=1;i<=100000;i++)//先处理出阶乘
		jc[i]=jc[i-1]*i%mod;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		ll tmp=jc[n-1],ans=0,u;
		scanf("%lld",&u);
		ans+=u*tmp%mod;
		for(int i=1;i<n;i++)//递推
		{
			scanf("%lld",&u);
			tmp=(tmp+jc[n-1]*inv(i))%mod;
			ans=(ans+u*tmp)%mod;
		}
		printf("%lld\n",ans*inv(jc[n])%mod);//一共有n!种情况
	}
	return 0;
}