题目大意
给定一个序列 a a a ,满足 a n = 1 + 2 + 3 + . . . + n a_{n}=1+2+3+...+n an=1+2+3+...+n,根据题目给定的概率选择其中两个(可相同)的元素 v 1 , v 2 v1,v2 v1,v2 ,我们记录 a ∈ [ 1... v 1 ] a \in [1...v1] a∈[1...v1], b ∈ [ 1... v 2 ] b \in [1...v2] b∈[1...v2] , a a a 和 b b b 都等概率地选择区间内的数,求使得 a < b a<b a<b 的概率(模 998244353 998244353 998244353)。
其中,假设选择第 i i i 个元素,那么选中第 i i i 个元素的概率为:
3 i ∗ ( i + 1 ) n ( n + 1 ) ( n + 2 ) \frac{3i * (i+1)}{n(n+1)(n+2)} n(n+1)(n+2)3i∗(i+1)
样例输入:
2
样例输出:
686292993
题解
做一下简单的分析,我们 a a a 和 b b b 的生成方法相同,所以我们 a < b a<b a<b 和 a > b a>b a>b 的概率是相等的,所以假设 a = b a=b a=b 的概率为 p p p 那么答案就为 1 − p 2 \frac{1-p}{2} 21−p 。
我们考虑每个数生成的概率。可以发现,第 i i i 个元素中含有 i ( i + 1 ) 2 \frac{i(i+1)}{2} 2i(i+1) 个数,那么在第 i i i 个元素内选中某一个数的概率为:
3 i ∗ ( i + 1 ) n ( n + 1 ) ( n + 2 ) ∗ 2 i ( i + 1