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 翻译