Codeforces 183D T-shirt【贪心DP】

题目描述：

题目分析：

期望可加。

每多带一件，就用O(n)的时间对对应的那一种做一次f的DP得到新的g值，求增长最大可用set（也可暴力）。

#include<bits/stdc++.h>
#define maxn 3005
#define maxm 305
using namespace std;
int n,m,cnt[maxm];
double p[maxn][maxm],ans,f[maxn][maxm],g[maxn],sum[maxn];
struct node{
	double x;int k; node(){}
	node(double x,int k):x(x),k(k){}
	bool operator < (const node &p)const{return x==p.x?k<p.k:x<p.x;}
};
set<node>s;
double calc(int k){
	for(int i=cnt[k];i<=n;i++) g[i]=f[i][k];
	f[cnt[k]][k]=0,cnt[k]++;
	for(int i=cnt[k];i<=n;i++) f[i][k]=p[i][k]*g[i-1]+(1-p[i][k])*f[i-1][k];
	return sum[k]+=f[n][k];
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) scanf("%lf",&p[i][j]),p[i][j]/=1000;
	for(int j=1;j<=m;j++){
		f[0][j]=1;
		for(int i=1;i<=n;i++) f[i][j]=f[i-1][j]*(1-p[i][j]);
		s.insert(node(sum[j]=f[n][j],j));
	}
	for(int t=1;t<=n;t++){
		node it=*s.begin(); 
		ans+=1-it.x, s.erase(s.begin());
		s.insert(node(calc(it.k),it.k));
	}
	printf("%.12f\n",ans);
}