HDU-5045 Contest 狀態壓縮DP求期望
來源：程序員人生 發布時間：2014-10-14 00:18:57 閱讀次數：3280次
N個人，M道題，M個小時，每個人做一道題需要1個小時。給出一個N*M的矩陣代表每個人做對每道題的概率。然后要求在任何時刻，任意兩個人的敲題時間差不能大于1，也就是說，m道題要分成多段長度為n的最優排列，n為10，2^10=1024 1024*1000 狀壓即可。
#include <iostream>
#include <cstdio>
#include <cmath>
#include <queue>
#include <cstdlib>
#include <cstring>
#include <algorithm>
using namespace std;
int n,m;
double ans;
double p[1111][1111];
double dp[1111][1111];
int main()
{
int t;
int cas=1;
scanf("%d",&t);
while(t--)
{
for(int i=0;i<=1024;i++)
{
for(int j=0;j<=1000;j++)
{
dp[i][j]=-1;
}
}
ans=-1;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
scanf("%lf",&p[i][j]);
}
}
int s=(1<<n)-1;//N個全部選過了
dp[0][0]=0;
for(int j=1;j<=m;j++)
{
for(int i=0;i<s;i++)
{
for(int k=1;k<=n;k++)
{
int temp=1<<(k-1);//選到第i個人的時候
if(dp[i][j-1]<0)
{
continue;
}
if(temp&i)//這個人已經被選過
{
continue;
}
int s1=i|temp;//把這個人標記為選過
if(s1==s)//因為時間只差不能超過1，因此是多段n的排列，然后從0開始
{
s1=0;
}
dp[s1][j]=max(dp[s1][j],dp[i][j-1]+p[k][j]);
if(j==m)
{
ans=max(ans,dp[s1][j]);
}
}
}
}
printf("Case #%d: %.5f
",cas++,ans);
}
return 0;
}
生活不易，碼農辛苦
如果您覺得本網站對您的學習有所幫助,可以手機掃描二維碼進行捐贈
