剑指Offer - 九度1360 - 乐透之猜数游戏
2014-02-05 19:54
- 题目描述：
六一儿童节到了,YZ买了很多丰厚的礼品,准备奖励给JOBDU里辛劳的员工。为了增添一点趣味性,他还准备了一些不同类型的骰子,打算以掷骰子猜数字的方式发放奖品。例如,有的骰子有6个点数(点数分别为1~6),有的骰子有7个(点数分别为1~7),还有一些是8个点数(点数分别为1~8) 。他每次从中拿出n个同一类型的骰子(假设它们都是拥有m个点数并且出现概率相同)投掷,然后让员工在纸上按优先级(从高到低)的顺序写下3个数上交,表示他们认为这些骰子最有可能的点数之和是多少。第一个数就猜对的人,是一等奖;第二个数才猜对的人是二等奖;如果三个数都不是正确答案,别灰心！YZ还准备了很多棒棒糖。ZL很聪明,他想了想,打算把概率(以保留两位小数的概率计)最高的三个数找出来,如果有概率相同,则选择其中点数和最小的那个数。你觉得ZL会依次写下哪三个数？
- 输入：
输入有多组数据。
每组数据一行,包含2个整数n(0<=n<=10),m(6<=m<=8),n表示YZ拿出的骰子数,m表示骰子拥有的点数。如果n=0,则结束输入。
- 输出：
对应每组数据,输出ZL最可能依次写下的点数,以及其对应的概率值。概率值按4舍5入要求保留2位小数。每组数据之间空一行,注意:最后一组数据末尾无空行。
- 样例输入：
1 6
4 6
3 7
0
- 样例输出：
1 0.17
2 0.17
3 0.17 13 0.11
14 0.11
15 0.11 12 0.11
10 0.10
11 0.10
题意分析： 扔骰(tou，不是shai)子求点数应该也是很常见的概率问题了。对于一个点数范围1~m的骰子，扔了n次，求出点数之和概率最大的三个值，以及对应的概率。
首先，扔一次获得各个点数的概率是相等的，因此各点数之和的概率分布一定是对称的，所以概率最大的一定就是正中间的三个点数和。因此，均值n*(1+m)/2就是概率最大的。考虑到奇偶和取整问题，在n奇m偶的情况下，概率最大的点数有两个。
如果你做过另一个题：求x+y+z=n的非负整数解，应该会想在这题上试试用组合数学来搞出个巧妙解法。但由于这道题中限制每个数都在1~m的范围内，需要分情况处理。所以我还是决定用n^2*m规模的动态规划来解决。
k个骰子能扔出的点数之和最小是k，最大是m*k，因此只要知道了k-1个骰子的所有扔法，就能推导出k个骰子的所有扔法的概率分布。
递推公式为：
1. 用a[n][k]表示n个骰子扔出的点数和为k的扔法个数。
2. a[1][j]=1, j=[1,m]
3. a[i][j]=sigma(a[i-1][k]), j=[i,i*m], k∈[i-1,(i-1)*m]∩[j-m,j-1]
由于递推需要进行n轮，每轮的点数之和从i到i*m。所以时间复杂度是O(n*(n*m))，也就是O(n^2*m)，空间复杂度可以优化为O(n*m)，不过本题中n和m都比较小，所以我没有进一步优化。
最后需要注意的，是对算出来的概率四舍五日到小数点后两位，然后再比较概率。
// 689492 zhuli19901106 1360 Accepted 点击此处查看所有case的执行结果 1024KB 1230B 10MS
//
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std; typedef struct st{
public:
int x;
int y;
st(int _x = , int _y = ): x(_x), y(_y) {};
}st; bool comparator(const st &a, const st &b)
{
if (a.y == b.y) {
return a.x < b.x;
} else {
return a.y > b.y;
}
} int main()
{
int n, m;
int i, j, k;
int sum;
int a[][];
vector<st> v;
int cc; cc = ;
while (scanf("%d", &n) == && n) {
scanf("%d", &m);
if (cc > ) {
printf("\n");
}
++cc;
memset(a, , * * sizeof(int));
for (i = ; i <= m; ++i) {
a[][i] = ;
}
sum = m;
for (i = ; i <= n; ++i) {
for (j = ; j <= m; ++j) {
for (k = i - ; k <= (i - ) * m; ++k) {
a[i][k + j] += a[i - ][k];
}
}
sum *= m;
} for (i = n; i <= n * m; ++i) {
v.push_back(st(i, (int)(100.0 * a[n][i] / sum + 0.5)));
} sort(v.begin(), v.end(), comparator); for (i = ; i < ; ++i) {
printf("%d %.2f\n", v[i].x, v[i].y / 100.0);
}
v.clear();
} return ;
}