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