CF2075C.Two Colors
普及/提高-
通过率:0%
AC君温馨提醒
该题目为【codeforces】题库的题目,您提交的代码将被提交至codeforces进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。
题目描述
Monocarp 决定按照以下规则粉刷围栏:
- 每块木板必须被涂上恰好一种颜色;
- 使用的不同颜色总数必须恰好为两种;
- 相同颜色的木板必须形成连续序列,即对于所有被涂成同一颜色的木板对,它们之间不存在被涂成其他颜色的木板。
Monocarp 有 m 种不同的颜料,其中第 i 种颜色的颜料最多可以涂 ai 块木板。Monocarp 不会购买额外颜料。
你的任务是计算满足 Monocarp 所有描述的愿望的围栏涂色方式数目。两种涂色方式被认为是不同的,当且仅当存在至少一块木板在这两种方式中被涂成不同颜色。
输入格式
第一行包含一个整数 t(1≤t≤104)——测试用例的数量。
每个测试用例的第一行包含两个整数 n 和 m(2≤n,m≤2⋅105)——围栏的木板数量以及 Monocarp 拥有的不同颜色数量。
第二行包含 m 个整数 a1,a2,…,am(1≤ai≤n),其中 ai 表示第 i 种颜色最多可涂的木板数量。
所有测试用例的 n 之和不超过 2⋅105。所有测试用例的 m 之和不超过 2⋅105。
输出格式
对于每个测试用例,输出满足条件的不同涂色方式数目。
输入输出样例
输入#1
3 5 2 2 4 5 2 3 4 12 3 5 9 8
输出#1
4 6 22
说明/提示
第一个测试案例中,存在 4 种不同的涂色方式(木板从左到右的颜色编号序列如下):
- [1,2,2,2,2];
- [1,1,2,2,2];
- [2,2,2,1,1];
- [2,2,2,2,1]。
第二个测试案例中,存在 6 种不同的涂色方式(木板从左到右的颜色编号序列如下):
- [1,2,2,2,2];
- [1,1,2,2,2];
- [1,1,1,2,2];
- [2,2,1,1,1];
- [2,2,2,1,1];
- [2,2,2,2,1]。
翻译由 DeepSeek R1 完成
输入解题思路,AI测评打分。不知道怎么写?