3499. 序列最大收益

Posted 2021-09-07 幽殇默

摘自: https://www.acwing.com/solution/content/49073/

#include<bits/stdc++.h>
using namespace std;
const int N=210;
int n,m,k;
int f[N][N],a[N],w[N][N],ans;
int main(void)
{
    cin>>n>>k>>m;
    for(int i=1;i<=m;i++) cin>>a[i];
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            cin>>w[i][j];
    memset(f,-0x3f,sizeof f);
    f[1][0]=0;
    for(int i=2;i<=m;i++)
        for(int j=0;j<=k;j++)
            for(int u=1;u<i;u++) 
                if(j>=i-1-u) f[i][j]=max(f[i][j],f[u][j-(i-1-u)]+w[a[u]][a[i]]);
    for(int i=0;i<=k;i++) ans=max(ans,f[m][i]);
    cout<<ans;
    return 0;
}