A15534 | Wish I Knew How to Sort
时间限制1s
内存限制256MB
通过 / 提交0/0
题目描述
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:
1. 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 \le i < j \le n$ .
2. If $a_i > a_j$ , then swap elements $a_i$ and $a_j$ .
What is the [expected number](https://en.wikipedia.org/wiki/Expected_value) 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 $\frac{p}{q}$ , where $p$ and $q$ are integers and $q \not \equiv 0 \pmod{998\,244\,353}$ . Output the integer equal to $p \cdot q^{-1} \bmod 998\,244\,353$ . In other words, output such an integer $x$ that $0 \le x < 998\,244\,353$ and $x \cdot q \equiv p \pmod{998\,244\,353}$ .
1. 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 \le i < j \le n$ .
2. If $a_i > a_j$ , then swap elements $a_i$ and $a_j$ .
What is the [expected number](https://en.wikipedia.org/wiki/Expected_value) 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 $\frac{p}{q}$ , where $p$ and $q$ are integers and $q \not \equiv 0 \pmod{998\,244\,353}$ . Output the integer equal to $p \cdot q^{-1} \bmod 998\,244\,353$ . In other words, output such an integer $x$ that $0 \le x < 998\,244\,353$ and $x \cdot q \equiv p \pmod{998\,244\,353}$ .
输入格式
Each test contains multiple test cases. The first line contains the number of test cases $t$ ( $1 \le t \le 10^5$ ). Description of the test cases follows.
The first line of each test case contains an integer $n$ ( $1 \le n \le 200\,000$ ) — the number of elements in the binary array.
The second line of each test case contains $n$ integers $a_1, a_2, \ldots, a_n$ ( $a_i \in \{0, 1\}$ ) — elements of the array.
It's guaranteed that sum of $n$ over all test cases does not exceed $200\,000$ .
The first line of each test case contains an integer $n$ ( $1 \le n \le 200\,000$ ) — the number of elements in the binary array.
The second line of each test case contains $n$ integers $a_1, a_2, \ldots, a_n$ ( $a_i \in \{0, 1\}$ ) — elements of the array.
It's guaranteed that sum of $n$ over all test cases does not exceed $200\,000$ .
输出格式
For each test case print one integer — the value $p \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)$ 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 $\frac{1}{3}$ , the probability that the array will become sorted after two operations is $\frac{2}{3} \cdot \frac{1}{3}$ , the probability that the array will become sorted after three operations is $\frac{2}{3} \cdot \frac{2}{3} \cdot \frac{1}{3}$ and so on. The expected number of operations is $\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 $\frac{75}{4}$ so the answer is $75 \cdot 4^{-1} \equiv 249\,561\,107 \pmod {998\,244\,353}$ .
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 $\frac{75}{4}$ so the answer is $75 \cdot 4^{-1} \equiv 249\,561\,107 \pmod {998\,244\,353}$ .
C++ 编辑器
输入
输出
可保存默认模板;新题优先使用已保存模板。
当前快捷键仅展示,暂不支持修改。
- 撤销
Ctrl / ⌘ + Z - 重做
Ctrl / ⌘ + Y - 查找
Ctrl / ⌘ + F - 全选
Ctrl / ⌘ + A - 复制
Ctrl / ⌘ + C - 剪切
Ctrl / ⌘ + X - 粘贴
Ctrl / ⌘ + V - 自动排版
工具栏排版按钮 - 草稿保存
编辑时自动保存到本机
历史
提交记录
状态说明时间源码
AI
作答助手
你好,我是作答助手。可以问思路、复杂度、样例含义或代码报错原因;不会直接给出完整 AC 代码。
确定要清空代码吗?
提交通过
评测结果:Accepted