Submission #74941488

Source Code Expand

import java.io.*;
import java.util.*;

public class Main {
    static long calc(long[] d, long[] pref, int n, long k, long s) {
        long end = s + k - 1;
        
        int left = lowerBound(d, s);       // first >= s
        int right = upperBound(d, end);    // first > end
        
        long leftCost = s * left - pref[left];
        
        long rightCount = n - right;
        long rightCost = (pref[n] - pref[right]) - end * rightCount;
        
        return leftCost + rightCost;
    }

    static int lowerBound(long[] a, long x) {
        int l = 0, r = a.length;
        while (l < r) {
            int m = (l + r) / 2;
            if (a[m] < x) l = m + 1;
            else r = m;
        }
        return l;
    }

    static int upperBound(long[] a, long x) {
        int l = 0, r = a.length;
        while (l < r) {
            int m = (l + r) / 2;
            if (a[m] <= x) l = m + 1;
            else r = m;
        }
        return l;
    }

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] first = br.readLine().split(" ");
        
        int n = Integer.parseInt(first[0]);
        long k = Long.parseLong(first[1]);
        
        String[] s = br.readLine().split(" ");
        long[] d = new long[n];
        
        for(int i = 0; i < n; i++) d[i] = Long.parseLong(s[i]);
        
        Arrays.sort(d);
        
        long[] pref = new long[n+1];
        for(int i = 0; i < n; i++) pref[i+1] = pref[i] + d[i];
        
        long ans = Long.MAX_VALUE;
        
        for(int i = 0; i < n; i++) {
            ans = Math.min(ans, calc(d, pref, n, k, d[i]));
            ans = Math.min(ans, calc(d, pref, n, k, d[i] - k + 1)); // 🔥 missing part
        }
        
        System.out.println(ans);
    }
}

Submission Info

Submission Time
Task D - Operation Plan for Distribution Center
User KrishnaPriya_16
Language Java24 (OpenJDK 24.0.2)
Score 400
Code Size 1927 Byte
Status AC
Exec Time 364 ms
Memory 70524 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 400 / 400
Status
AC × 3
AC × 78
Set Name Test Cases
Sample sample01.txt, sample02.txt, sample03.txt
All sample01.txt, sample02.txt, sample03.txt, in01.txt, in02.txt, in03.txt, in04.txt, in05.txt, in06.txt, in07.txt, in08.txt, in09.txt, in10.txt, in11.txt, in12.txt, in13.txt, in14.txt, in15.txt, in16.txt, in17.txt, in18.txt, in19.txt, in20.txt, in21.txt, in22.txt, in23.txt, in24.txt, in25.txt, in26.txt, in27.txt, in28.txt, in29.txt, in30.txt, in31.txt, in32.txt, in33.txt, in34.txt, in35.txt, in36.txt, in37.txt, in38.txt, in39.txt, in40.txt, in41.txt, in42.txt, in43.txt, in44.txt, in45.txt, in46.txt, in47.txt, in48.txt, in49.txt, in50.txt, in51.txt, in52.txt, in53.txt, in54.txt, in55.txt, in56.txt, in57.txt, in58.txt, in59.txt, in60.txt, in61.txt, in62.txt, in63.txt, in64.txt, in65.txt, in66.txt, in67.txt, in68.txt, in69.txt, in70.txt, in71.txt, in72.txt, in73.txt, in74.txt, in75.txt
Case Name Status Exec Time Memory
in01.txt AC 53 ms 38500 KiB
in02.txt AC 42 ms 38304 KiB
in03.txt AC 61 ms 39724 KiB
in04.txt AC 55 ms 39488 KiB
in05.txt AC 48 ms 38628 KiB
in06.txt AC 325 ms 70524 KiB
in07.txt AC 334 ms 69308 KiB
in08.txt AC 52 ms 38636 KiB
in09.txt AC 59 ms 39676 KiB
in10.txt AC 324 ms 70008 KiB
in11.txt AC 339 ms 70172 KiB
in12.txt AC 356 ms 69436 KiB
in13.txt AC 325 ms 69796 KiB
in14.txt AC 348 ms 69656 KiB
in15.txt AC 364 ms 70092 KiB
in16.txt AC 342 ms 69408 KiB
in17.txt AC 235 ms 68064 KiB
in18.txt AC 237 ms 70492 KiB
in19.txt AC 47 ms 38168 KiB
in20.txt AC 326 ms 69340 KiB
in21.txt AC 269 ms 67820 KiB
in22.txt AC 254 ms 69788 KiB
in23.txt AC 270 ms 67440 KiB
in24.txt AC 266 ms 69312 KiB
in25.txt AC 269 ms 69236 KiB
in26.txt AC 346 ms 69648 KiB
in27.txt AC 345 ms 69748 KiB
in28.txt AC 344 ms 69660 KiB
in29.txt AC 49 ms 38612 KiB
in30.txt AC 51 ms 38764 KiB
in31.txt AC 46 ms 38488 KiB
in32.txt AC 45 ms 38536 KiB
in33.txt AC 45 ms 38456 KiB
in34.txt AC 47 ms 38644 KiB
in35.txt AC 47 ms 38596 KiB
in36.txt AC 47 ms 38576 KiB
in37.txt AC 47 ms 38560 KiB
in38.txt AC 47 ms 38208 KiB
in39.txt AC 175 ms 53276 KiB
in40.txt AC 50 ms 38496 KiB
in41.txt AC 48 ms 38284 KiB
in42.txt AC 50 ms 38448 KiB
in43.txt AC 52 ms 38496 KiB
in44.txt AC 64 ms 40288 KiB
in45.txt AC 340 ms 69456 KiB
in46.txt AC 319 ms 69012 KiB
in47.txt AC 336 ms 69464 KiB
in48.txt AC 340 ms 69260 KiB
in49.txt AC 323 ms 69096 KiB
in50.txt AC 271 ms 69580 KiB
in51.txt AC 252 ms 67496 KiB
in52.txt AC 333 ms 68808 KiB
in53.txt AC 292 ms 67492 KiB
in54.txt AC 358 ms 69156 KiB
in55.txt AC 330 ms 69016 KiB
in56.txt AC 338 ms 68256 KiB
in57.txt AC 320 ms 69496 KiB
in58.txt AC 342 ms 69080 KiB
in59.txt AC 322 ms 69336 KiB
in60.txt AC 47 ms 38476 KiB
in61.txt AC 46 ms 38632 KiB
in62.txt AC 47 ms 38292 KiB
in63.txt AC 47 ms 38260 KiB
in64.txt AC 47 ms 38052 KiB
in65.txt AC 48 ms 38772 KiB
in66.txt AC 253 ms 67860 KiB
in67.txt AC 48 ms 38372 KiB
in68.txt AC 47 ms 38476 KiB
in69.txt AC 46 ms 38620 KiB
in70.txt AC 47 ms 38596 KiB
in71.txt AC 46 ms 38336 KiB
in72.txt AC 46 ms 38360 KiB
in73.txt AC 47 ms 38460 KiB
in74.txt AC 48 ms 38332 KiB
in75.txt AC 49 ms 38564 KiB
sample01.txt AC 49 ms 38644 KiB
sample02.txt AC 48 ms 38356 KiB
sample03.txt AC 47 ms 38368 KiB