CF1753C.Wish I Knew How to Sort
普及+/提高
通过率:0%
时间限制:2.00s
内存限制:256MB
AC君温馨提醒
该题目为【codeforces】题库的题目,您提交的代码将被提交至codeforces进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。
题目描述
You are given a binary array a (all elements of the array are 0 or 1) of length n. You wish to sort this array, but unfortunately, your algorithms teacher forgot to teach you sorting algorithms. You perform the following operations until a is sorted:
- Choose two random indices i and j such that i<j. Indices are chosen equally probable among all pairs of indices (i,j) such that 1≤i<j≤n.
- If ai>aj, then swap elements ai and aj.
What is the expected number of such operations you will perform before the array becomes sorted?
It can be shown that the answer can be expressed as an irreducible fraction qp, where p and q are integers and q≡0(mod998244353). Output the integer equal to p⋅q−1mod998244353. In other words, output such an integer x that 0≤x<998244353 and x⋅q≡p(mod998244353).
给你一个长度为 n 的二进制数组 a(数组中所有元素均为 0 或 1)。你想将该数组排序,但不幸的是,你的算法老师忘记教你排序算法了。你将持续执行以下操作,直到数组 a 被排好序:
- 随机选择两个下标 i 和 j,满足 i<j。所有满足 1≤i<j≤n 的下标对 (i,j) 被选中的概率相等。
- 若 ai>aj,则交换元素 ai 和 aj。
在数组变为有序之前,你所执行的此类操作的期望次数是多少?
可以证明,该答案可表示为最简分数 qp,其中 p 和 q 为整数,且 q≡0(mod998244353)。请输出整数 p⋅q−1mod998244353。换言之,请输出满足 0≤x<998244353 且 x⋅q≡p(mod998244353) 的整数 x。
输入格式
Each test contains multiple test cases. The first line contains the number of test cases t (1≤t≤105). Description of the test cases follows.
The first line of each test case contains an integer n (1≤n≤200000) — the number of elements in the binary array.
The second line of each test case contains n integers a1,a2,…,an (ai∈0,1) — elements of the array.
It's guaranteed that sum of n over all test cases does not exceed 200000.
每个测试包含多个测试用例。第一行包含测试用例的数量 t(1≤t≤105)。随后是测试用例的描述。
每个测试用例的第一行包含一个整数 n(1≤n≤200000)—— 表示二进制数组中元素的个数。
每个测试用例的第二行包含 n 个整数 a1,a2,…,an(其中 ai∈{0,1})—— 表示数组的元素。
保证所有测试用例的 n 之和不超过 200000。
输出格式
For each test case print one integer — the value p⋅q−1mod998244353.
对于每个测试用例,输出一个整数——即 p⋅q−1mod998244353 的值。
输入输出样例
输入#1
3 3 0 1 0 5 0 0 1 1 1 6 1 1 1 0 0 1
输出#1
3 0 249561107
说明/提示
Consider the first test case. If the pair of indices (2,3) will be chosen, these elements will be swapped and array will become sorted. Otherwise, if one of pairs (1,2) or (1,3) will be selected, nothing will happen. So, the probability that the array will become sorted after one operation is 31, the probability that the array will become sorted after two operations is 32⋅31, the probability that the array will become sorted after three operations is 32⋅32⋅31 and so on. The expected number of operations is i=1∑∞(32)i−1⋅31⋅i=3.
In the second test case the array is already sorted so the expected number of operations is zero.
In the third test case the expected number of operations equals to 475 so the answer is 75⋅4−1≡249561107(mod998244353).
考虑第一个测试用例。若选择下标对 (2,3),则这两个元素将被交换,数组将变为有序;否则,若选择下标对 (1,2) 或 (1,3) 中的任意一个,则数组不发生任何变化。因此,一次操作后数组变为有序的概率为 31,两次操作后变为有序的概率为 32⋅31,三次操作后变为有序的概率为 32⋅32⋅31,依此类推。期望操作次数为 i=1∑∞(32)i−1⋅31⋅i=3。
在第二个测试用例中,数组初始即为有序,因此期望操作次数为 0。
在第三个测试用例中,期望操作次数等于 475,故答案为 75⋅4−1≡249561107(mod998244353)。
输入解题思路,AI测评打分。不知道怎么写?