寒假每日一题打卡day1——ACWing 104. 货仓选址

【题目描述】

ACWing 104. 货仓选址

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算法的时间复杂度
添加链接描述