小C打比赛 wannafly C题 （记忆化搜索）

网址；https://www.nowcoder.com/acm/contest/57/C
小C打比赛
时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld
题目描述
小C现在要参加一场wannafly挑战赛，一场挑战赛一共有n道题，一共有m分钟。
对于第i道题，小C解决它需要恰好j分钟的概率是pi,j。
小C每次会选择某一道没做完的题，然后把它解决（不能中途放弃），之后再决策下一道要做的题是哪道。
求小C在最优策略下，期望能做出几道题。
输入描述:

第一行两个正整数n,m
接下来一共n行，每行有m个小数，第i行的第j个小数表示pi,j（这里假设不存在0分钟A题的dalao）。

输出描述:

输出一个小数，表示期望能做出几道题，保留小数点后五位。

示例1
输入

2 5
0.2 0.2 0.2 0.2 0.2
0 0.25 0.25 0.25 0.25

输出

1.30000

备注:

1≤ n≤ 6,1≤ m≤ 180
每道题的概率和为1(每道题只要时间够一定能做出来)
输入最多四位小数

#include<bits/stdc++.h>
using namespace std;
#define N 200
int n,m;
double p[20][N];
double dp[N][N];
int vis[N][N];
double dfs(int sta,int lef){
    if(lef<=0)return 0;
    if(vis[sta][lef])return dp[sta][lef];
    double ans=0;
    for(int i=1;i<=n;++i){
        if((1<<i)&sta)continue;
        int nxt=(1<<i)|sta;
        double kk=0;
        for(int j=1;j<=lef;++j){
            kk+=p[i][j]*(1+dfs(nxt,lef-j));
        }
        ans=max(ans,kk);
    }
    vis[sta][lef]=1;
    return dp[sta][lef]=ans;
}
int main(){
    //freopen("in.txt","r",stdin);
    cin>>n>>m;
    for(int i=1;i<=n;++i){
        for(int j=1;j<=m;++j){
            cin>>p[i][j];
        }
    }
    dfs(0,m);
    printf("%.5lf\n",dp[0][m]);
}

2018年4月27日 20:50:34 （这么久的坑，我现在可以独立做出来了。。）

#include<bits/stdc++.h>
using namespace std;

#define LL long long
#define N 200010
#define inf 0x3f3f3f3f
const double eps=0.00001;


int n,m;
double p[(1<<6)][200];

double dp[(1<<6) + 10][200];
int vis[(1<<6) + 10][200];
double dfs(int state,int remain){
    if(vis[state][remain])return dp[state][remain];
    vis[state][remain]=1;
    double ret=0;
    for(int i=1;i<=n;++i){
        if(state>>(i-1) & 1)continue;
        double tmp=0;
        for(int j=1;j<=m;++j){
            if(j>remain)break;
            tmp+=p[i][j]* (1 + dfs(state | 1 << (i - 1) ,remain-j));
        }
        dp[state][remain]=max(dp[state][remain],tmp);
    }
    return dp[state][remain];
}
int main(){
    //freopen("in.txt","r",stdin);
    cin>>n>>m;
    for(int i=1;i<=n;++i){
        for(int j=1;j<=m;++j){
            cin>>p[i][j];
        }
    }
    printf("%.5lf\n",dfs(0,m));

}