牛客AB52题解析:环形序列合并的动态规划解法
一、题目解读
牛客AB52题要求处理一个环形序列,通过合并相邻珠子释放能量,计算最大可释放能量值。题目核心在于环形结构的处理与动态规划的应用,需将环形问题转化为线性问题求解。
二、解题思路
采用动态规划(DP)解决该问题。首先,通过复制原数组形成长度为2n的线性序列,将环形结构转化为线性区间计算。随后,利用区间DP思想,定义dp[i][j]为合并区间[i,j]的最大能量,通过枚举分割点k优化状态转移。最终遍历所有n长度区间取最大值,得到答案。
三、解题步骤
1. 输入与初始化:读入序列n及元素,构建2n长度的arr数组。
2. 定义DP状态:dp[i][j]表示合并i到j珠子的最大能量。
3. 区间DP计算:
○ 枚举区间长度len=2~n,确保覆盖所有合法区间。
○ 枚举起点i,计算终点j,并遍历分割点k(i~j-1)。
○ 状态转移方程:dp[i][j] = max(dp[i][j], dp[i][k] + dp[k+1][j] + arr[i]*arr[k+1]*arr[j+1]),即合并能量为子区间能量之和+当前区间首尾元素乘积。
4. 结果获取:遍历所有n长度区间dp[i][i+n-1],取最大值res作为答案。
四、代码与注释
#include <iostream> #include <vector> #include <algorithm> using namespace std; int main() { int n; cin >> n; vector<int> nums(n); for(int i = 0; i < n; ++i) { cin >> nums[i]; } // 环形转线性:复制数组接在后面 vector<int> arr(nums.begin(), nums.end()); arr.insert(arr.end(), nums.begin(), nums.end()); // dp[i][j]为合并i到j的最大能量 vector<vector<int>> dp(2*n, vector<int>(2*n, 0)); // 区间DP:枚举长度len和分割点k for(int len = 2; len <= n; ++len) { for(int i = 0; i + len - 1 < 2*n; ++i) { int j = i + len - 1; for(int k = i; k < j; ++k) { // 状态转移方程:合并能量=子区间能量+首尾乘积 dp[i][j] = max(dp[i][j], dp[i][k] + dp[k+1][j] + arr[i]*arr[k+1]*arr[j+1]); } } } // 找出所有n长度区间的最大值 int res = 0; for(int i = 0; i < n; ++i) { res = max(res, dp[i][i+n-1]); } cout << res << endl; return 0; }
五、总结
本解法关键在于环形结构到线性问题的转化,通过复制数组突破环形边界限制。区间DP通过枚举子区间分割点,有效解决组合优化问题。时间复杂度为O(n³),空间复杂度O(n²),适用于中小规模数据。掌握此类动态规划模型可高效解决类似环形序列优化问题。
原创内容 转载请注明出处