CF2075C.Two Colors

普及/提高-

通过率:0%

AC君温馨提醒

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

题目描述

Monocarp 决定按照以下规则粉刷围栏:

  • 每块木板必须被涂上恰好一种颜色;
  • 使用的不同颜色总数必须恰好为两种;
  • 相同颜色的木板必须形成连续序列,即对于所有被涂成同一颜色的木板对,它们之间不存在被涂成其他颜色的木板。

Monocarp 有 mm 种不同的颜料,其中第 ii 种颜色的颜料最多可以涂 aia_i 块木板。Monocarp 不会购买额外颜料。

你的任务是计算满足 Monocarp 所有描述的愿望的围栏涂色方式数目。两种涂色方式被认为是不同的,当且仅当存在至少一块木板在这两种方式中被涂成不同颜色。

输入格式

第一行包含一个整数 tt(1≤t≤1041 \le t \le 10^4)——测试用例的数量。

每个测试用例的第一行包含两个整数 nn 和 mm(2≤n,m≤2⋅1052 \le n, m \le 2 \cdot 10^5)——围栏的木板数量以及 Monocarp 拥有的不同颜色数量。

第二行包含 mm 个整数 a1,a2,…,ama_1, a_2, \dots, a_m(1≤ai≤n1 \le a_i \le n),其中 aia_i 表示第 ii 种颜色最多可涂的木板数量。

所有测试用例的 nn 之和不超过 2⋅1052 \cdot 10^5。所有测试用例的 mm 之和不超过 2⋅1052 \cdot 10^5。

输出格式

对于每个测试用例,输出满足条件的不同涂色方式数目。

输入输出样例

  • 输入#1

    3
    5 2
    2 4
    5 2
    3 4
    12 3
    5 9 8

    输出#1

    4
    6
    22

说明/提示

第一个测试案例中,存在 44 种不同的涂色方式(木板从左到右的颜色编号序列如下):

  1. [1,2,2,2,2][1, 2, 2, 2, 2];
  2. [1,1,2,2,2][1, 1, 2, 2, 2];
  3. [2,2,2,1,1][2, 2, 2, 1, 1];
  4. [2,2,2,2,1][2, 2, 2, 2, 1]。

第二个测试案例中,存在 66 种不同的涂色方式(木板从左到右的颜色编号序列如下):

  1. [1,2,2,2,2][1, 2, 2, 2, 2];
  2. [1,1,2,2,2][1, 1, 2, 2, 2];
  3. [1,1,1,2,2][1, 1, 1, 2, 2];
  4. [2,2,1,1,1][2, 2, 1, 1, 1];
  5. [2,2,2,1,1][2, 2, 2, 1, 1];
  6. [2,2,2,2,1][2, 2, 2, 2, 1]。

翻译由 DeepSeek R1 完成

输入解题思路,AI测评打分。不知道怎么写?

首页