【题目描述】
TLE代码
import java.io.*;
import java.lang.Math;
class Main{
static int N = 100010;
public static void main(String args[])throws Exception{
BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(bf.readLine());
int []f = new int[N];
String s[] = bf.readLine().split(" ");
int max = 0;
for(int i = 1; i <= n; i++) {
f[i] = Integer.parseInt(s[i - 1]);
max = Math.max(f[i], max);
}
long ans = Integer.MAX_VALUE;
//枚举货仓选址地点
for(int i = 1; i <=max; i ++){
//计算货仓i到每家商店的距离之和
long tmp = 0;
for(int j = 1; j <= n; j++){
tmp += (long) Math.abs( f[j] - i);
}
ans = Math.min(ans, tmp);
}
System.out.println(ans);
}
}
根据问题规模为10^5,要能AC算法时间复杂度至多为O(n*logn)级别。
import java.io.*;
import java.util.Arrays;
import java.lang.Math;
class Main{
static int N = 100010;
public static void main(String args[])throws Exception{
BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(bf.readLine());
int []f = new int[n];
String s[] = bf.readLine().split(" ");
for(int i = 0; i < n; i++)
f[i] = Integer.parseInt(s[i]);
Arrays.sort(f);
long ans = 0;
if( n % 2 == 1 ){
int k = f[ n / 2 ];
for(int i = 0; i < n; i++) ans += Math.abs(f[i] - k);
}else{
int k = (f[ n / 2 ] + f[ (n - 1) / 2]) / 2;
for(int i = 0; i < n; i++) ans += Math.abs(f[i] - k);
}
System.out.println(ans);
}
}
根据问题规模推测能AC算法的时间复杂度
添加链接描述