代码拉取完成,页面将自动刷新
// Time: O(m * n), m is the max of rollMax
// Space: O(m)
class Solution {
public:
int dieSimulator(int n, vector<int>& rollMax) {
static const int MOD = 1e9 + 7;
vector<vector<int>> dp;
for (int i = 0; i < 6; ++i) {
dp.emplace_back(rollMax[i]);
dp[i][0] = 1; // 0-indexed
}
while (--n) {
vector<vector<int>> new_dp;
for (int i = 0; i < 6; ++i) {
new_dp.emplace_back(rollMax[i]);
}
for (int i = 0; i < 6; ++i) {
for (int k = 0; k < rollMax[i]; ++k) {
for (int j = 0; j < 6; ++j) {
if (i == j) {
if (k < rollMax[i] - 1) { // 0-indexed
new_dp[j][k + 1] = (new_dp[j][k + 1] + dp[i][k]) % MOD;
}
} else {
new_dp[j][0] = (new_dp[j][0] + dp[i][k]) % MOD;
}
}
}
}
dp = move(new_dp);
}
uint64_t result = 0;
for (const auto& row : dp) {
const auto& total =
accumulate(row.cbegin(), row.cend(),
0ull,
[&](const auto& a, const auto& b) {
return (a + b) % MOD;
});
result = (result + total) % MOD;
}
return result;
}
};
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。