题目描述:
小Q和牛博士在玩一个石子合并的游戏,初始一共有n堆石子,每堆石子有w[i]个石子。小Q和牛博士他们需要对石子堆进行合并,每次他们可以任意选择两堆石子进行合并。一堆有x个石子的石子堆和一堆有y个石子的石子堆合并将得到一堆x+y个石子的石子堆,这次合并得分为x*y,当只剩下一堆石子的时候游戏结束。
小Q和牛博士希望采取优秀的策略获得最大得分,希望你能来帮他们算算最大得分多少。
思路:
可以发现不管合并的顺序如何,结果都一致,所以以下代码采用的是依次合并。
代码:(C++)
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main()
{
int n;
cin>>n;
int a;
vector<int> A(n);
for(int i=0;i<n;i++)
{
cin>>a;
A[i]=a;
}
int sum=0;
for(int i=0;i<n-1;i++)
{
sum=sum+A[i]*A[i+1];
A[i+1]=A[i]+A[i+1];
}
cout<<sum;
return 0;
}
提交结果: