可以使用三种方法:
- 回溯法,回溯法可以遍历每一种可能,属于暴力解法。
- 分治法,分治法首先要得到状态转移方程。
- dp动态规划
首先为数组增加首位
fun(i,j)相当于不戳边界 i,j。戳其中的任意一个气球得到的硬币数。
对于确定的最后戳中间k索引的气球fun(i,j)=fun(i,k)+fun(k,j)+nums[i]*nums[k]*nums[j];
所以有fun(i, j)=max{fun(i,k)+fun(k,j)+nums[i]*nums[k]*nums[j]};对于所有的k来说。
使用cache[i][j]来保存结果,避免重复计算。
public int maxCoins(int[] nums) {
int[] n_helper=new int[nums.length+2];
n_helper[0]=1;n_helper[n_helper.length-1]=1;
System.arraycopy(nums, 0, n_helper, 1, nums.length);
int[][] cache=new int[nums.length+2][nums.length+2];
return helper(n_helper, 0, n_helper.length-1, cache);
}
public int helper(int[] nums,int start,int end,int[][] cache) {
if(start==end-1)return 0;
if(cache[start][end]!=0)return cache[start][end];
int max=0;
for(int k=start+1;k<end;k++) {
max=Integer.max(max, helper(nums, start, k, cache)+helper(nums, k, end, cache)+nums[start]*nums[k]*nums[end]);
}
cache[start][end]=max;
return max;
}
若使用dp,即将上面的cache转化为dp,dp的计算顺序为先计算i,j之差小的,再推到大的。
public int maxCoins(int[] nums) {
int[] n_helper=new int[nums.length+2];
System.arraycopy(nums, 0, n_helper, 1, nums.length);
n_helper[0]=1;
n_helper[n_helper.length-1]=1;
int[][] dp=new int[n_helper.length][n_helper.length];
for(int cha=2;cha<n_helper.length;cha++) {
for(int i=0;i+cha<n_helper.length;i++) {
for(int k=i+1;k<i+cha;k++) {
dp[i][i+cha]=Integer.max(dp[i][i+cha], dp[i][k]+dp[k][i+cha]+n_helper[i]*n_helper[k]*n_helper[i+cha]);
}
}
}
return dp[0][n_helper.length-1];
}