力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
把n个骰子扔在地上,所有骰子朝上一面的点数之和为s,输入n,打印出s的所有可能出现的值出现的概率。
思路:一个骰子扔出来的点数为{1,2,3,4,5,6}。n个骰子扔出来为s,和n-1个骰子扔出来的点数密切相关。因为只要n-1个骰子扔出来了s-1,s-2,...,s-6,最后一个骰子随便扔出几都可以凑成s。所以用一个二维的dp数组来保存扔出来骰子的点数次数。
本来准备开辟n+1行,s+1列的数组,后来发现可能会越界,就开辟了s+7,这样在一开始初始化的时候就没问题了。出现的概率就直接除以n对应的所有点数,6^n就可以了。
void main() {
int n,s;
cin >> n>>s;
vector<vector<int>> dp(n + 1, vector<int>(s + 7, 0));
dp[1][1] = 1;
dp[1][2] = 1;
dp[1][3] = 1;
dp[1][4] = 1;
dp[1][5] = 1;
dp[1][6] = 1;
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= s; j++) {
for (int k = 1; k <= 6; k++) {
if (j - k < 0) continue;
dp[i][j] += dp[i - 1][j - k];
}
}
}
cout << dp[n][s];
}