C++程序计算满足两个条件的涂色方案的数量

假设我们有三个数字N，M和K。考虑有N个块，它们排列在一行中。我们考虑以下两种涂色方式。当且仅当以下两种方式中的块以不同的颜色被涂色时，两个块的颜色不同：-

• 对于每个块，使用M种颜色中的一种来涂色（不一定使用所有颜色）

• 最多可能有K对相邻块以相同的颜色被涂色

如果答案太大，返回结果模998244353。

因此，如果输入为N = 3; M = 2; K = 1，则输出将为6，因为我们可以以以下不同的格式涂色：112、121、122、211、212和221。

步骤

为了解决这个问题，我们将按照以下步骤进行：

maxm := 2^6 + 5 p := 998244353 Define two large arrays fac and inv or size maxm Define a function ppow(), this will take a, b, p, ans := 1 mod p a := a mod p while b is non-zero, do: if b is odd, then: ans := ans * a mod p a := a * a mod p b := b/2 return ans Define a function C(), this will take n, m, if m n, then: return 0 return fac[n] * inv[m] mod p * inv[n - m] mod p From the main method, do the following fac[0] := 1 for initialize i := 1, when i = 0, update (decrease i by 1), do: inv[i] := (i + 1) * inv[i + 1] mod p ans := 0 for initialize i := 0, when i >= 1; } return ans; } long C(long n, long m){ if (m n) return 0; return fac[n] * inv[m] % p * inv[n - m] % p; } long solve(long n, long m, long k){ fac[0] = 1; for (long i = 1; i = 0; i--) inv[i] = (i + 1) * inv[i + 1] % p; long ans = 0; for (long i = 0; i