SWUN 1431 - 伊邪那美(Ⅱ)

 

伊邪那美(Ⅱ) 时间限制(普通/Java) : 1000 MS/ 3000 MS          运行内存限制 : 65536 KByte 总提交 : 30            测试通过 : 5 Special Judge 描述 伊邪那美是一种究极幻术，中招的人会陷入一个无限自循环的术中世界。   术中世界一共有n条通道，其中的一些通道会让你回到现实世界，另一些通道会让你再次回到术中世界，你只记得最近走过的k个通道。（例如k=1,你第一次走4号通道，并且回到了术中世界，那么第二次选择时，你记得4号通道是错误的，你就不会走4号，假如第二次你选择2号通道，并且又回到的术中世界，那么你只记得2号通道是错误的，而忘记了4号通道是否正确）   你每次都会等概率的选择某一通道进入，并且穿过第i条通道需要花费的时间为ti。   现在，Snow_storm想知道，中招的人回到现实世界的期望时间是多少。   输入     多组测试数据（测试组数<=10000）。     对于每组测试数据，第一行为正整数n,k( 0<n<=100,0<=k<=2 )     接下来的一行，有n个整数ti，若ti>0，则表示第i条通道会让你回到现实世界，且花费的时间为ti；若ti<0，则表示第i条通道会让你再次回到术中世界，且花费的时间为|ti|。( 0<|ti|<=10000,且保证至少存在一个ti大于0 ) 输出     对于每组测试数据，输出回到现实世界的期望时间(误差需小于10-6)。 样例输入 3 0 3 -6 -9 3 1 3 -6 -9 样例输出 18.0000000000 13.0000000000 题目来源 YB

 

题目地址：　http://218.194.91.48/acmhome/problemdetail.do?&method=showdetail&id=1431

 

伊邪那美(I) ： http://blog.youkuaiyun.com/diannaok/article/details/7803445  的扩展题。

 

我的做法很简单，在 第一版 的题目得到的公式下，推出 k = 1 时 的公式，接下去 k = 2 时，直接例举出所有 k=1 时的情况，算出结果。

 

AC之后找B哥，他说这题有个 通式，可以推出来。。。 听起来就觉得很麻烦，囧。。。

 

不过效率上确实比较高。

 

#include<stdio.h>

int main(){
	int n,k;
	double res,sum1,sum2,a,flg[110];
	int m,i,num;
	while(~scanf("%d%d",&n,&k)){
		m=num=0;
		sum1=sum2=0;
		for(i=0;i<n;i++){
			scanf("%lf",&a);
			if(a<0){
				flg[m]=-a;
				m++;
				sum1-=a;
				sum2-=a;
			}
			else{
				num++;
				sum1+=a;
			}
		}
		if(k==0 || n==1){
			res=sum1/num;
		}
		else if(k==1 || n==2){
			res=(sum1+(m*sum1-sum2)/num)/n;
		}
		else{
			res=sum1;
			for(i=0;i<m;i++){
				res+=(sum1-flg[i]+((m-1)*(sum1-flg[i])-(sum2-flg[i]))/num)/(n-1);
			}
			res/=n;
		}
		printf("%.10lf\n",res);
	}
	return 0;
}
/*
3 2
3 -6 -9

10.5

*/