hpu 【1190】確率（组合数学）

1190: 確率 [数学]

时间限制: 1 Sec  内存限制: 128 MB

提交: 595  解决: 278  统计

题目描述

话说当年语文考满分的你认识标题第一个字吗？

HPU有很多内湖，湖里面放养了很多金鱼（还有黑白天鹅？），每天都会有很多小伙伴在湖边看金鱼在水中嬉戏（黑白天鹅打架？）。

Ocean是一个善于思考的好孩子，今天他给你出了一道题目：
假设湖中一共有 N N条金鱼，现在Ocean随机挑出两条金鱼，问这两条金鱼颜色不同的概率？

为了降低题目难度，Ocean认为金鱼只会有 6 6种不同的颜色，即 1，2，3，4，5，6。 1，2，3，4，5，6。

PS：在挑出来第二条鱼之前，Ocean是不会将第一条鱼放入湖中的。

输入

第一行输入一个整数 T T，代表有 T T组测试数据。
每组数据占两行，第一行输入一个整数 N N代表上面提到的信息。
下面一行输入 N N个整数 colori colori，代表第 i i条金鱼的颜色。
当 colori=1 colori=1，代表第 i i条金鱼的颜色为 1 1，其它依次类推。

注： 1<=T<=100，2<=N<=100，1<=colori<=6。 1<=T<=100，2<=N<=100，1<=colori<=6。

输出

对每组测试数据，输出一个浮点数代表最后的结果，要求保留两位小数。

样例输入

2
4
1 1 2 2
2
2 2

样例输出

0.67
0.00

开始不想写这道题的，因为其实这完全就是一道纯数学题，组合数学吧，记着是，哈哈，很长时间没做过都快忘了怎么写了，总之，最后算是历经波折吧这道题AC了，主要是开始思路有问题，改了好几遍（唉，一道这种稍微复杂一点的思维题就卡壳QAQ），代码如下（还是注意数据类型的选择）：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
long long a[10];
int zuhe(int x)
{
	if(x<2)
	return 0;
	int ans=1;
	for(int i=1;i<=2;i++)
	{
		ans=ans*(x-2+i)/i;
	}
	return ans;
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{   
	    memset(a,0,sizeof(a));
	   	int n;
	   	int k,o;
	   	scanf("%d",&n);
	   	for(int i=0;i<n;i++)
	   	{
	   	   int x;
	   	   scanf("%d",&x);
	   	   a[x]++;
		}
		k=zuhe(n);
		o=0;
		for(int i=1;i<=6;i++)
		{
			o+=zuhe(a[i]);
		}
		printf("%.2f\n",1-1.0*o/k);
    }
	return 0;
}