试题 算法训练 石子游戏
题目描述:
资源限制
时间限制:1.0s 内存限制:256.0MB
问题描述
石子游戏的规则如下:
地上有n堆石子,每次操作可选取两堆石子(石子个数分别为x和y)并将它们合并,操作的得分记为(x+1)×(y+1),对地上的石子堆进行操作直到只剩下一堆石子时停止游戏。
请问在整个游戏过程中操作的总得分的最大值是多少?
输入格式
输入数据的第一行为整数n,表示地上的石子堆数;第二行至第n+1行是每堆石子的个数。
输出格式
程序输出一行,为游戏总得分的最大值。
样例输入
10
5105
19400
27309
19892
27814
25129
19272
12517
25419
4053
样例输出
15212676150
数据规模和约定
1≤n≤1000,1≤一堆中石子数≤50000
解题思路:
把这些数据从大到小排序,每次取最大的两个,即可,不难实现。
注意:
使用long long类型,不然不够。
下面提供两种方法实现,时刻的排序。其中第二种使用优先队列。
第一种AC代码:
#include <stdlib.h>
#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <string.h>
#include <math.h>
using namespace std;
int a[1000];
int main()
{
int n,i;
cin>>n;
long long ans=0;
for(i=0;i<n;i++)
cin>>a[i];
long long max1,max2,pos1,pos2;//注意类型
int t=1;
while(t<n)//执行n-1次即可
{
max1=0;
max2=0;
for(i=0;i<n;i++)//找到最大的那个数字
if(max1<a[i])
{
max1=a[i];
pos1=i;
}
a[pos1]=0;//为了找到第二大的数字,现将第一次找到的索引位置数值为0
for(i=0;i<n;i++)//找到第二大的数字
if(max2<a[i])
{
max2=a[i];
pos2=i;
}
ans+=(max1+1)*(max2+1);
a[pos1]+=max1+max2;
a[pos2]=0;
t++;
}
cout<<ans<<endl;
return 0;
}
第二种AC代码:
#include <stdlib.h>
#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
using namespace std;
int main()
{
//优先队列,从大到小排序(最先出来的是最小的)
priority_queue<long long,vector<long long>,less<long long> >stones;
long long x,y,ans=0;
int n;
cin>>n;
long long temp;
for(int i=0;i<n;i++)
{
cin>>temp;
stones.push(temp);
}
while(stones.size()>=2)
{
x=stones.top();
stones.pop();
y=stones.top();
stones.pop();
ans+=(x+1)*(y+1);
stones.push(x+y);
}
cout<<ans<<endl;
return 0;
}