AT_pakencamp_2023_day2_g Reducing x K

题目描述

给定一个正整数 $N$ 和一个长度为 $N$ 的非负整数列 $A$。请考虑恰好重复以下操作 $K$ 次: - 从 $A$ 中选择一个正的元素。记选中的元素为 $x$,将其替换为满足 $0 \leq y < x$ 的某个整数 $y$。 请你计算,经过 $K$ 次操作后,所有可能得到的 $A$ 的种类数。答案对 $998244353$ 取模。

输入格式

输入按照以下格式从标准输入中给出。 > $N\ K\ A_1\ A_2\ \cdots\ A_N$

输出格式

请输出答案一行。

说明/提示

## 配点 本题按下面三个子任务划分分值。 1. (200分) $N, K \leq 100$ 2. (400分) $N, K \leq 1000$ 3. (200分) 没有额外限制。 ## 样例解释 1 列举出经过 $K$ 次操作后所有可能的 $A$ 有:$(1, 2, 1), (1, 2, 0), (1, 1, 2), (1, 1, 1), (1, 1, 0), (1, 0, 3), (1, 0, 2), (1, 0, 1), (1, 0, 0), (0, 2, 2), (0, 2, 1), (0, 2, 0), (0, 1, 3), (0, 0, 3) $ 共 $14$ 种。 # 数据范围 - $1 \leq N \leq 10^5$ - $1 \leq K \leq 10^5$ - $0 \leq A_i \leq 10^9$ - 所有输入都为整数。 由 ChatGPT 5 翻译