题意:就5种盒子,给出每个盒子个数,盒子总数,每个人选择这个盒子的概率。求这个人选择哪个盒子取得第一个朋友的概率最大,最大多少
dp[N][sta]表示当前第N个人面临状态sta(选择盒子的状态可以用13进制数表示)时的概率,
那么转移就是dp[N][sta]=sum(dp[N-1][sta-1]*G[n][k]) (表示第N个人选择第k个盒子)
那么答案应该是max(P(第1个人选择i号盒子)/总状态概率)(i<=5)
#include <map> #include <set> #include <list> #include <cmath> #include <ctime> #include <deque> #include <stack> #include <queue> #include <cctype> #include <cstdio> #include <string> #include <vector> #include <climits> #include <cstdlib> #include <cstring> #include <iostream> #include <algorithm> #define LL long long #define PI 3.1415926535897932626 using namespace std; int gcd(int a, int b) {return a % b == 0 ? b : gcd(b, a % b);} #define MAXN 350000 #define MAXD 15 #define MAXB 10 double p[MAXD][MAXB],dp[MAXD][MAXN]; bool vis[MAXD][MAXN]; int N,res[MAXB],gift[MAXB]; double calcu(int cur,int st) { if (vis[cur][st]) return dp[cur][st]; vis[cur][st]=true; double ans=0; int sta=st; for (int i=5;i>0;i--) {res[i]=sta%13;sta/=13;} for (int i=1;i<=5;i++) if (res[i]) { res[i]--; sta=0; for (int j=1;j<=5;j++) sta=sta*13+res[j]; ans+=calcu(cur+1,sta)*p[cur][i]; res[i]++; } return dp[cur][st]=ans; } int main() { //freopen("sample.txt","r",stdin); int T; scanf("%d",&T); while (T--) { scanf("%d",&N); int sta=0; for (int i=1;i<=5;i++) { scanf("%d",&gift[i]);sta=sta*13+gift[i];} for (int i=1;i<=N;i++) for (int j=1;j<=5;j++) scanf("%lf",&p[i][j]); memset(vis,false,sizeof(vis)); vis[N+1][0]=true;dp[N+1][0]=1; double tmp=calcu(1,sta); double ans=-1.0,sym;int ide; for (int i=1;i<=5;i++) if (gift[i]) { gift[i]--; sta=0; for (int j=1;j<=5;j++) sta=sta*13+gift[j]; gift[i]++; sym=dp[2][sta]*p[1][i]/gift[i]; if (sym/tmp>ans) { ans=sym/tmp; ide=i; } } printf("%d %.3lf\n",ide,ans); } return 0; }
时间: 2024-10-14 00:35:53