acwing320能量项链【区间dp】

acwing320能量项链【区间dp】

利用区间dp，注意考虑起始位置就行了，我是用了一个pair数组，first存起点，second存终点：

#include "bits/stdc++.h"
using namespace std;
#define int long long
#define IOS ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
int n;
int dp[205][205];       //记录区间最小
pair<int,int> num[205];           //数列
signed main(){
    IOS
    memset(dp,0,sizeof dp);
    cin>>n;
    for(int i = 1;i <= n;++i)
        cin>>num[i].first,num[i - 1].second = num[i].first;
    num[n].second = num[1].first;
    for(int i = n + 1,j = n * 2;i <= j;++i)
        num[i] = num[i - n];
    //以上是初始化前缀和数组和dp状态
    for(int i = 2;i <= n;++i)   //合并后的长度为2...n
        for(int l = 1,k = 2 * n - i + 1;l <= k;++l) //合并后起点为l，并且起点可以是1 ~ 2*n-i+1
            for(int p = l,r = l + i - 1;p < r;++p)
                dp[l][r] = max(dp[l][p] + dp[p + 1][r] + num[l].first * num[p].second * num[r].second,dp[l][r]);
    int ans = dp[1][n];
    for(int i = 2;i <= n;++i)
        ans = max(dp[i][i + n - 1],ans);
    cout<<ans;
    return 0;
}