POJ 2443 Set Operation -- 位运算 + bitset

最新推荐文章于 2023-03-16 16:54:41 发布

x314542916

最新推荐文章于 2023-03-16 16:54:41 发布

阅读量1.3k

点赞数

分类专栏：【杂】

本文链接：https://blog.csdn.net/x314542916/article/details/12358975

版权

【杂】专栏收录该内容

1 篇文章 0 订阅

订阅专栏

/*
	http://poj.org/problem?id=2443 
	POJ2443 Set Operation -- 位运算 + bitset
	询问两个元素在不在同一个集合内
	(在 ? yes : no)
*/

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
#define CLR(c,v) memset(c,v,sizeof(c));
int const N = 1e5 +10;

double loge[N];
double nCr(int n,int r){
	if(n==0)return 0.;
	return loge[n] - loge[r] - loge[n-r];
}
int a[N],sum[N];

int main(){
	//freopen("in.txt","r",stdin);
	loge[0] = 0;
	for(int i = 1 ; i < N ; i++)
		loge[i] += loge[i-1] + log(i*1.0);
	
	int n,k;
	while(cin >> n >> k,n||k){
		CLR(sum,0);
		for(int i = 0 ; i < n ; i++){
			scanf("%d",&a[i]);
			sum[i+1] += sum[i] + a[i];
		}
		double ans = 0;
		for(int i = k ; i <= n ; i++){ // seq length
			double p = exp( nCr(i-2,k-2) - nCr(n,k) );
			ans += p * ( (sum[n]-sum[i-1])-(sum[n-i+1]-sum[0]) );
		}
		printf("%.3f\n" , ans);
	}
	return 0;
}