1 Star 0 Fork 0

yuhang2__2/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
dice-roll-simulation.cpp 1.49 KB
一键复制 编辑 原始数据 按行查看 历史
// 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;
}
};
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/yuhang2__2/LeetCode-Solutions.git
git@gitee.com:yuhang2__2/LeetCode-Solutions.git
yuhang2__2
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助