题目背景
某次编程俱乐部拥有了一层楼并购置了一台高性能服务器,为了让每层楼都能连接到服务器,我们需要对每间房子进行接线。
题目描述
可以把这层楼看成一个数轴,房间不均匀的分布在这个数轴的整数点上,问,该服务器放置在哪件房中可以使总接线的长度最小?
输入格式:
第一行输入整数N,表示房间数量。
第二行N个整数A1~An.
输出格式:
输出一个整数,表示距离之和的最小值。
输入样例#1:
4
6 2 9 1
输出样例#1:
12
说明
1≤N≤105
代码+题解
#include<stdio.h>
#include<algorithm>
using namespace std;
long long int arr[100005],l[100005]={0}; //arr[]用来存储输入的数,l[]用来记录结果。
int main()
{
int N;
scanf("%d",&N);
for(int i=0;i<N;i++)
scanf("%lld",&arr[i]);
sort(arr,arr+N); //将房间号从小到大排序。
for(int i=1;i<N;i++)
l[0]+=(arr[i]-arr[0]); //求出第一间房到各个房间的总距离。
for(int i=1;i<N;i++)
{
l[i]=0;
l[i]=l[i-1]+(2*i-N)*(arr[i]-arr[i-1]); //往后求每个房间到其他房间的总距离:
//l[i]=l[i-1]+(i-1)*(arr[i]-arr[i-1])-(n-i-1)*(arr[i]-arr[i-1])。
}
long long int ma=1e9;
for(int i=0;i<N;i++)
ma=min(ma,l[i]); //找到最小值。
printf("%lld",ma);
return 0;
}