AT_arc135_f [ARC135F] Delete 1, 4, 7, ...
题目描述
给定一个整数 $N$。对于整数序列 $A = (1, 2, \ldots, N)$,你需要恰好进行 $K$ 次如下操作:
- 设当前 $A$ 的项数为 $n$。对于所有满足 $1 \leq i \leq n$ 且 $i \equiv 1 \pmod{3}$ 的 $i$,同时删除 $A$ 的第 $i$ 项。
请你求出经过 $K$ 次操作后,$A$ 的所有项的总和对 $998244353$ 取模的结果。
输入格式
输入通过标准输入给出,格式如下:
> $N$ $K$
输出格式
输出经过 $K$ 次操作后,$A$ 的所有项的总和对 $998244353$ 取模的结果。
说明/提示
## 限制条件
- $1 \leq N \leq 10^{14}$
- $1 \leq K \leq 100$
## 样例解释 1
- 初始时,$A = (1, 2, 3, 4, 5, 6, 7, 8, 9, 10)$。
- 第 $1$ 次操作后,$A = (2, 3, 5, 6, 8, 9)$。
- 第 $2$ 次操作后,$A = (3, 5, 8, 9)$。
- 此时 $A$ 的项的总和为 $3 + 5 + 8 + 9 = 25$。
## 样例解释 2
- 第 $2$ 次操作后,$A = (3, 5, 8, 9)$(与样例 1 相同)。
- 第 $3$ 次操作后,$A = (5, 8)$。
- 第 $4$ 次操作后,$A = (8)$。
- 第 $5$ 次操作后,$A$ 为空。
- 第 $6$ 次及之后的操作,$A$ 仍为空,其项的总和为 $0$。
由 ChatGPT 4.1 翻译