CF2075C Two Colors
题目描述
Monocarp 决定按照以下规则粉刷围栏:
- 每块木板必须被涂上恰好一种颜色;
- 使用的不同颜色总数必须恰好为两种;
- 相同颜色的木板必须形成连续序列,即对于所有被涂成同一颜色的木板对,它们之间不存在被涂成其他颜色的木板。
Monocarp 有 $m$ 种不同的颜料,其中第 $i$ 种颜色的颜料最多可以涂 $a_i$ 块木板。Monocarp 不会购买额外颜料。
你的任务是计算满足 Monocarp 所有描述的愿望的围栏涂色方式数目。两种涂色方式被认为是不同的,当且仅当存在至少一块木板在这两种方式中被涂成不同颜色。
输入格式
第一行包含一个整数 $t$($1 \le t \le 10^4$)——测试用例的数量。
每个测试用例的第一行包含两个整数 $n$ 和 $m$($2 \le n, m \le 2 \cdot 10^5$)——围栏的木板数量以及 Monocarp 拥有的不同颜色数量。
第二行包含 $m$ 个整数 $a_1, a_2, \dots, a_m$($1 \le a_i \le n$),其中 $a_i$ 表示第 $i$ 种颜色最多可涂的木板数量。
所有测试用例的 $n$ 之和不超过 $2 \cdot 10^5$。所有测试用例的 $m$ 之和不超过 $2 \cdot 10^5$。
输出格式
对于每个测试用例,输出满足条件的不同涂色方式数目。
说明/提示
第一个测试案例中,存在 $4$ 种不同的涂色方式(木板从左到右的颜色编号序列如下):
1. $[1, 2, 2, 2, 2]$;
2. $[1, 1, 2, 2, 2]$;
3. $[2, 2, 2, 1, 1]$;
4. $[2, 2, 2, 2, 1]$。
第二个测试案例中,存在 $6$ 种不同的涂色方式(木板从左到右的颜色编号序列如下):
1. $[1, 2, 2, 2, 2]$;
2. $[1, 1, 2, 2, 2]$;
3. $[1, 1, 1, 2, 2]$;
4. $[2, 2, 1, 1, 1]$;
5. $[2, 2, 2, 1, 1]$;
6. $[2, 2, 2, 2, 1]$。
翻译由 DeepSeek R1 完成