CF1753C.Wish I Knew How to Sort

普及/提高-

通过率:0%

AC君温馨提醒

该题目为【codeforces】题库的题目,您提交的代码将被提交至codeforces进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。

题目描述

You are given a binary array aa (all elements of the array are 00 or 11 ) of length nn . You wish to sort this array, but unfortunately, your algorithms teacher forgot to teach you sorting algorithms. You perform the following operations until aa is sorted:

  1. Choose two random indices ii and jj such that i<ji < j . Indices are chosen equally probable among all pairs of indices (i,j)(i, j) such that 1i<jn1 \le i < j \le n .
  2. If ai>aja_i > a_j , then swap elements aia_i and aja_j .

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 pq\frac{p}{q} , where pp and qq are integers and q≢0(mod998244353)q \not \equiv 0 \pmod{998\,244\,353} . Output the integer equal to pq1mod998244353p \cdot q^{-1} \bmod 998\,244\,353 . In other words, output such an integer xx that 0x<9982443530 \le x < 998\,244\,353 and xqp(mod998244353)x \cdot q \equiv p \pmod{998\,244\,353} .

输入格式

Each test contains multiple test cases. The first line contains the number of test cases tt ( 1t1051 \le t \le 10^5 ). Description of the test cases follows.

The first line of each test case contains an integer nn ( 1n2000001 \le n \le 200\,000 ) — the number of elements in the binary array.

The second line of each test case contains nn integers a1,a2,,ana_1, a_2, \ldots, a_n ( ai{0,1}a_i \in \{0, 1\} ) — elements of the array.

It's guaranteed that sum of nn over all test cases does not exceed 200000200\,000 .

输出格式

For each test case print one integer — the value pq1mod998244353p \cdot q^{-1} \bmod 998\,244\,353 .

输入输出样例

  • 输入#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)(2, 3) will be chosen, these elements will be swapped and array will become sorted. Otherwise, if one of pairs (1,2)(1, 2) or (1,3)(1, 3) will be selected, nothing will happen. So, the probability that the array will become sorted after one operation is 13\frac{1}{3} , the probability that the array will become sorted after two operations is 2313\frac{2}{3} \cdot \frac{1}{3} , the probability that the array will become sorted after three operations is 232313\frac{2}{3} \cdot \frac{2}{3} \cdot \frac{1}{3} and so on. The expected number of operations is i=1(23)i113i=3\sum \limits_{i=1}^{\infty} \left(\frac{2}{3} \right)^{i - 1} \cdot \frac{1}{3} \cdot 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 754\frac{75}{4} so the answer is 7541249561107(mod998244353)75 \cdot 4^{-1} \equiv 249\,561\,107 \pmod {998\,244\,353} .

首页