腾讯马拉松的一道题(hdu oj 4503)
湫湫系列故事——植树节
Time Limit: 1000/500 MS(Java/Others) Memory Limit: 65535/32768 K (Java/Others)
Total Submission(s): 445 Accepted Submission(s): 272
Problem Description
今天是一年一度的植树节,腾讯幼儿园要求每个老师在班里选出几个小朋友一起去野外种植小树苗,根据学校的整体安排,湫湫老师的班里要选出3个小朋 友。已知湫湫的班里共有n个孩子,每个孩子有Bi个朋友(i从1到n),且朋友关系是相互的,如果a小朋友和b小朋友是朋友,那么b小朋友和a小朋友 也一定是好朋友。为了选择的公平性,湫湫老师会随机抽取3个小朋友出来(每个人被抽到的概率相同),但是她很希望这3个小朋友之间的关系完全相同,湫湫老 师想请你帮她算算抽到的3个小朋友正好关系相同的概率是多少?
PS. 关系相同就是指要么3个人互相是好朋友,要么3个人互相都不是好朋友。
Input
输入数据第一行是一个整数T(1<=T<=1000),表示输入数据的组数;每组数据的第一行是一正整数n表示孩子的总数(2<n<=1000),第二行有n个数Bi (i从1到n),分别代表每个小朋友的朋友的个数。
Output
对于每组数据,请输出抽到的3个小朋友关系相同的概率,结果保留3位小数。
Sample Input
1
5
3 3 3 3 4
Sample Output
0.400
解法:先求关系不完全相同的组合个数,可以把每个人看做一个节点,则其朋友的个数看做度,则构成无向图,则3个关系不完全相同就是3个点有1条或者
两条边。小孩i认识data[i]人,不认识的人数(n-data[i]-1);那么选出不满足情况(i小孩认识1人认识,不认识1人)的种类数就是:data[i]*(n-data[i]-1);但是这样被重复计算过一次,所以总数要除以2
Java代码:
import java.io.BufferedInputStream;
import java.util.Scanner;
publicclassProblem4503 {
/**
* author : luoyang
* 2013-3-29
* @param args
*/
publicstaticvoid main(String[] args) {
// TODO Auto-generated method stub
Scannerinput=newScanner(newBufferedInputStream( System.in));
int n=input.nextInt();
int a[]=newint [1010];
while((n--)!=0)
{
int k=input.nextInt();
for(int i=0;i<k;i++)
{
a[i]=input.nextInt();
}
int sum=0;
for(int i=0;i<k;i++)
{
sum+=(a[i]*(k-1-a[i]));
}
sum=sum/2;
int all=k*(k-1)*(k-2)/6;
double r=1-1.0*sum/all;
System.out.printf("%.3f",r);
System.out.println();
}
}
}
注意:读入时使用Scanner(system.in)会报超时的错误,因为这里读入的数据量较大,所以要换为Scanner(newBufferedInputStream( System.in))