AT_kupc2024_m Mahjong 2

题目描述

共有 $3M-1$ 张写有 $1$ 到 $N$ 的整数的卡片,对于每个 $1 \leq i \leq N$,写有整数 $i$ 的卡片有 $A_i$ 张。 你需要从 $1$ 到 $N$ 的整数中选择一个 $K$,然后再额外添加一张写有 $K$ 的卡片,使得如下条件能够被满足。 - 条件:可以重复 $M$ 次,每次拿走 $3$ 张卡片,这 $3$ 张卡片要么写有相同的数字,要么写有连续的数字,最终卡片全部拿空。 请你枚举所有满足条件的 $K$。

输入格式

输入按以下格式从标准输入给出: > $N$ $M$ $A_1$ $A_2$ $\ldots$ $A_N$

输出格式

第一行输出满足条件的 $K$ 的个数 $L$。 接下来一行按升序输出所有可能的 $K$。 > $L$ $K_1$ $K_2$ $\ldots$ $K_L$

说明/提示

### 部分得分 对于满足下述约束的数据集,答对可获得 $1$ 分: - $N \leq 2000$ ### 样例解释 1 例如添加 $K=5$ 后,可以组成以下 $4$ 组:$\{2,2,2\}$、$\{2,3,4\}$、$\{3,4,5\}$、$\{5,5,5\}$。 ### 样例解释 2 有时不存在满足条件的 $K$。 ### 约束条件 - 输入均为整数 - $1 \leq N \leq 5 \times 10^5$ - $1 \leq M \leq 10^8$ - $0 \leq A_i \leq 3M-1$ - $\sum A_i = 3M-1$ 由 ChatGPT 5 翻译