AcWIng夏季每日一题--序列最大收益

https://www.acwing.com/problem/content/3502/

 

 

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
int n,k,m;
const int N = 210;
int a[N],w[N][N];
int f[N][N];
int main()
{
    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];
    //初始化为f[0][0]=0;
    for(int i=1;i<=m;i++){
        for(int j=0;j<=k;j++){
            for(int u=1;u<i;u++){
                if(j>=(i-u-1)){
                    f[i][j]=max(f[i][j],f[u][j-(i-u-1)]+w[a[u]][a[i]]);
                }
            }
        }
    }
    int res=0;
    for(int i=0;i<=m;i++){
        for(int j=0;j<=k;j++){
            res=max(f[i][j],res);
        }
    }
    cout<<res;
    return 0;
}