思路:dp[i][j]:=第i轮打完后,决定以j阵容打下一轮
保持原有阵容:dp[ i ][ j ] = dp[ i - 1 ][ j ] * p [ j ][ s [ i ] ]
换成第i轮怪的阵容:
for(int k=0;k<r;++k)dp[i][j]=max(dp[i][j],dp[i-1][k]*p[k][s[i]])
优化:用滚动数组,把 i 换成 now
#include <iostream> #include <cstring> #include <cstdio> #include <algorithm> #define INF 0x7f7f7f7f using namespace std; typedef long long LL; int s[10005]; double p[125][125]; double dp[2][125]; int main(){int n;while(cin>>n){int r=n*(n-1)*(n-2)/6;for(int i=0;i<r;++i)for(int j=0;j<r;++j) scanf("%lf",&p[i][j]);int m;cin>>m;for(int i=0;i<m;++i)scanf("%d",&s[i]);for(int i=0;i<125;++i)dp[1][i]=1;int now=0;for(int i=0;i<m;++i){for(int j=0;j<r;++j){dp[now][j]=dp[now^1][j]*p[j][s[i]];if(s[i]==j)for(int k=0;k<r;++k)dp[now][j]=max(dp[now][j],dp[now^1][k]*p[k][s[i]]);}now^=1; }double ans=0;for(int i=0;i<r;++i) ans=max(ans,dp[now^1][i]);printf("%.7lf\n",ans);}return 0; }