9166 - 选择题(choice)

通过次数

3

提交次数

25

时间限制 : 1 秒
内存限制 : 128 MB

题目背景

注意:本场比赛每道题满分均为 100 分,提交显示 Accepted 不一定代表通过。

高中的许多琐碎,终将随着时间的流逝而褪色。可是,小 M 永远记得,那个夏天……

小 M 期中考试由于迷信某些玄学,写挂了。 (你以为呢)

题目描述

期中考试的玄学不灵了,小 M 很伤心,他把一张试卷扔给了你。这张试卷上有 N 道标号从 1N 的单项选择题,每道选择题都有 K 个不同种类选项,且每道题只选择一个选项。

现在小 M 想问你有多少种不同的方案,满足该试卷存在三道或以上标号连续的题目选择同一个选项(例如:第 8,9,10 题都选第 3 种选项,又例如第 3,4,5,6 题都选第 1 种选项)。

本题中,两种方案不同指在这两种方案中,存在至少一道相同标号的题目选择不同种类的选项。答案与 998244353 取模。

输入

一行两个整数 N,K

输出

一行一个非负整数,表示符合题目要求的方案数与 998244353 取模后的结果。

样例

输入

4 4

输出

28

输入

5 3

输出

63

输入

49571 10

输出

205967084

提示

样例解释

样例1:有以下几种方案:四题都选同一种,共4种;1,2,3题选同一种,第4题选另一种,共12种;2,3,4题选同一种, 第1题选另一种,共12种。全部加起来有28种。

样例2:符合 AAAAAAAAABBAAAAAAABBAAABCBAAABBAAACBBAAABCAAA 形式的都符合要求。(上述表示中,一种字母代表选择一种答案,字母排列顺序为题号顺序。)

数据范围

对于 10\% 的数据,N \leq 15K = 4

对于 50\% 的数据,N \leq 5 \times 10^4K \leq 15

另有 20\% 的数据,N \leq 5 \times 10^3

对于所有数据, 3 \leq N \leq 2 \times 10^5 2 \leq K \leq 10^4

来源

SF Round 5 暨 NOIP 信心赛