题目描述
在Mars星球上,每个Mars人都随身佩带着一串能量项链。在项链上有N颗能量珠。能量珠是一颗有头标记与尾标记的珠子,这些标记对应着某个正整数。并且,对于相邻的两颗珠子,前一颗珠子的尾标记一定等于后一颗珠子的头标记。因为只有这样,通过吸盘(吸盘是Mars人吸收能量的一种器官)的作用,这两颗珠子才能聚合成一颗珠子,同时释放出可以被吸盘吸收的能量。如果前一颗能量珠的头标记为m,尾标记为r,后一颗能量珠的头标记为r,尾标记为n,则聚合后释放的能量为:m×r×n(Mars单位),新产生的珠子的头标记为m,尾标记为n。
需要时,Mars人就用吸盘夹住相邻的两颗珠子,通过聚合得到能量,直到项链上只剩下一颗珠子为止。显然,不同的聚合顺序得到的总能量是不同的,请你设计一个聚合顺序,使一串项链释放出的总能量最大。
例如:设N=4,4颗珠子的头标记与尾标记依次为(2,3) (3,5) (5,10) (10,2)。我们用记号⊕表示两颗珠子的聚合操作,(j⊕k)表示第j,k两颗珠子聚合后所释放的能量。则第4、1两颗珠子聚合后释放的能量为:
(4⊕1)=10×2×3=60。
这一串项链可以得到最优值的一个聚合顺序所释放的总能量为
((4⊕1)⊕2)⊕3)=10×2×3+10×3×5+10×5×10=710。
算法思想(区间DP)
根据题目描述,相邻的两颗珠子才能聚合成一颗珠子,同时释放出可以被吸盘吸收的能量。这样就需要使用区间型DP的思想,枚举区间长度len
、合并的起点i
、终点j
,以及最后的合并位置k
,计算从i
合并到j
时,所能释放的最大总能量f[i][j]
。
因为是在项链中进行合并,所以可以使用拆环为链的思想,将其变为一个直线型区间合并问题。
状态表示
f[i][j]
表示从第i
颗珠子合并到第j
颗珠子所释放的最大总能量。
状态转移方程
f[i][j] = max(f[i][j], f[i][k] + f[k + 1][j] + h[i] * h[k + 1] * r[j]);
其中k
为最后一次合并的位置。
时间复杂度
O ( n 3 ) O(n^3) O(n3)
代码实现
#include <iostream>
using namespace std;
const int N = 210;
int h[N], r[N];
int f[N][N];
int main()
{
int n;
cin >> n;
for(int i = 1; i <= n; i ++) cin >> h[i];
//拆环为链
for(int i = 1; i <= n; i ++) h[i + n] = h[i];
for(int i = 1; i <= 2 * n - 1; i ++) r[i] = h[i + 1];
//len表示区间长度,最少合并两颗珠子才能释放能量
for(int len = 2; len <= n; len ++)
{
//i为合并的起点
for(int i = 1; i <= 2 * n - len; i ++)
{
int j = i + len - 1; //j为合并的终点
//枚举最后一次合并位置
for(int k = i; k < j; k ++)
{
f[i][j] = max(f[i][j], f[i][k] + f[k + 1][j] + h[i] * h[k + 1] * r[j]);
}
}
}
int ans = 0;
//打擂台求所有开始位置的最大值
for(int i = 1; i <= n; i ++)
ans = max(ans, f[i][i + n - 1]);
cout << ans << endl;
return 0;
}