题目
链接:https://www.nowcoder.com/acm/contest/90/J
来源:牛客网
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld
题目描述
牛客网是IT求职神器,提供海量C++、JAVA、前端等职业笔试题库,在线进行百度阿里腾讯网易等互联网名企笔试面试模拟考试练习,和牛人一起讨论经典试题,全面提升你的编程。作为acmer的小A,牛客网是他首选的题库。
小A是一个中度强迫症患者,每次做数组有关的题目都异常难受,他十分希望数组的每一个元素都一样大,这样子看起来才是最棒的,所以他决定通过一些操作把这个变成一个看起来不难受的数组,但他又想不要和之前的那个数组偏差那么大,所以他每次操作只给这个数组的其中n-1个元素加1,但是小A并不能很好的算出最优的解决方案,如果你能帮他解决这个问题,小A就能送你一个气球
输入描述:
第一行一个整数T(T<=100),表示组数
对于每组数据有一个n,表示序列的长度(0< n <100000)
下面一行有n个数,表示每个序列的值(0
分析
先考虑纯暴力,只要序列内元素不都相等,就循环操作:【将序列排个序,然后将序列前n-1个元素都加1】。这样的时间复杂度为O(m*nlogn),m为循环的次数。m的次数是有可能达到n次以上的,所以暴力不可解。
虽然暴力不可解,但我们可以在模拟暴力解法过程中发现一些规律。
对数组1,5,7.
我们可以将7看做不存在,将1累加4次加到5。这样对于数组长度为2的序列,问题已经解决了。但是实际上7它是存在的,我们可以看作7也累加了4次,然后我们只需将5,5都累加到11即可。
可以验证这样操作与暴力操作所得的答案是一致的。
那么如果对数组1,5,7,21呢?同理。
如果数组更长呢?
将数组排好序后,
如果数组长度为1,则操作次数为d[1]=0,数组元素ans=a[1];
如果数组长度为2,则操作次数为d[2]=d[1]+d[1]+a[1]-ans,数组元素为ans=d[1]+a[2].
…
如果数组长度为i,则操作次数为d[i]=d[i-1]+d[i-1]+a[i-1]-ans,数组元素为ans=d[i-1]+a[i]。
而d[i]的空间复杂度可以通过滚动数组来优化。
代码
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <queue>
#include <map>
using namespace std;
const int maxn=1e5+100;
typedef long long ll;
ll a[maxn];
ll d,ans;
int main()
{
ll T;
scanf("%lld",&T);
while(T--)
{
ll n;
scanf("%lld",&n);
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
sort(a+1,a+n+1);
d=0;
ans=a[1];
ll tmp;
for(int i=2;i<=n;i++)
{
tmp=d;
d=2*d+a[i]-ans;
ans=tmp+a[i];
}
printf("%lld %lld\n",d,ans);
}
return 0;
}