BZOJ 2201: 彩色圆环 期望

2201: 彩色圆环

Time Limit: 10 Sec  Memory Limit: 259 MB
Submit: 175  Solved: 91
[Submit][Status][Discuss]

Description

Input

仅有一行，该行给出依次两个正整数N, M，分别表示宝石的个数和宝石在变化时可能变成的颜色种类数。

Output

应仅有一行，该行给出一个实数E(R)，表示圆环的“美观程度”的期望值。

Sample Input

8 1

Sample Output

8.00000
【数据规模和约定
100%的数据满足1 ≤ N ≤ 200, 1 ≤ M ≤ 10^9。

感觉 没做出来这个题 是挺shame的...

f[i][2]表示长度为i+1的序列收尾颜色是否与开头相同的期望值

转移很简单看code就行

之后就是最后统计答案

枚举与1号位置连续的个数i 因为是个环 所以有i种方案 这时候我们把这一段看成一个点

所以每一段的贡献是 i*i*f[i][0]*p(i)

加到一起就行了

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;

typedef double db;
typedef long long ll;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=210;

db f[N][2],g[N];

int main()
{
	int n=read(),m=read();
	register int i,j;
	db p=1.0/m;
	g[1]=1;f[0][1]=1;
	for(i=2;i<=n;++i) g[i]=g[i-1]*p;
	for(i=1;i<=n;++i)
		for(j=0;j<n;++j)
		{
			f[i][0]+=(i-j)*g[i-j]*f[j][0]*(1-2*p)+f[j][1]*(i-j)*g[i-j]*(1-p);
			f[i][1]+=(i-j)*g[i-j]*f[j][0]*p;
		}
	db ans=g[n]*n;
	for(i=1;i<n;++i) ans+=f[n-i][0]*g[i]*i*i;
	printf("%.5lf",ans);
	return 0;
}