Submission #70986274

Source Code Expand

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        long[] candies = new long[n];
        long sum = 0;
        
        for (int i = 0; i < n; i++) {
            candies[i] = sc.nextLong();
            sum += candies[i];
        }
        
        if (sum % n != 0) {
            System.out.println(-1);
            return;
        }
        
        long target = sum / n;
        
        List<Integer> sources = new ArrayList<>();
        List<Integer> sinks = new ArrayList<>();
        long[] excess = new long[n];
        long[] deficit = new long[n];
        
        for (int i = 0; i < n; i++) {
            if (candies[i] > target) {
                sources.add(i);
                excess[i] = candies[i] - target;
            } else if (candies[i] < target) {
                sinks.add(i);
                deficit[i] = target - candies[i];
            }
        }
        
        if (sources.isEmpty() && sinks.isEmpty()) {
            System.out.println(0);
            return;
        }
        
        List<long[]> operations = new ArrayList<>();
        
        int srcIdx = 0, sinkIdx = 0;
        while (srcIdx < sources.size() && sinkIdx < sinks.size()) {
            int src = sources.get(srcIdx);
            int sink = sinks.get(sinkIdx);
            
            long transfer = Math.min(excess[src], deficit[sink]);
            operations.add(new long[]{src + 1, sink + 1, transfer});
            
            excess[src] -= transfer;
            deficit[sink] -= transfer;
            
            if (excess[src] == 0) srcIdx++;
            if (deficit[sink] == 0) sinkIdx++;
        }
        
        System.out.println(operations.size());
        for (long[] op : operations) {
            System.out.println(op[0] + " " + op[1] + " " + op[2]);
        }
        
        sc.close();
    }
}

Submission Info

Submission Time
Task F - Candy Redistribution
User addy
Language Java24 (OpenJDK 24.0.2)
Score 0
Code Size 2005 Byte
Status WA
Exec Time 81 ms
Memory 41368 KiB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 525
Status
AC × 2
AC × 55
WA × 39
Set Name Test Cases
Sample 00-sample-01.txt, 00-sample-02.txt
All 00-sample-01.txt, 00-sample-02.txt, 01-01.txt, 01-02.txt, 01-03.txt, 01-04.txt, 01-05.txt, 01-06.txt, 01-07.txt, 01-08.txt, 01-09.txt, 01-10.txt, 01-11.txt, 01-12.txt, 01-13.txt, 01-14.txt, 01-15.txt, 01-16.txt, 01-17.txt, 01-18.txt, 01-19.txt, 01-20.txt, 01-21.txt, 01-22.txt, 01-23.txt, 01-24.txt, 01-25.txt, 01-26.txt, 01-27.txt, 01-28.txt, 01-29.txt, 01-30.txt, 01-31.txt, 01-32.txt, 01-33.txt, 01-34.txt, 01-35.txt, 01-36.txt, 01-37.txt, 01-38.txt, 01-39.txt, 01-40.txt, 01-41.txt, 01-42.txt, 01-43.txt, 01-44.txt, 01-45.txt, 01-46.txt, 01-47.txt, 01-48.txt, 01-49.txt, 01-50.txt, 01-51.txt, 01-52.txt, 01-53.txt, 01-54.txt, 01-55.txt, 01-56.txt, 01-57.txt, 01-58.txt, 01-59.txt, 01-60.txt, 01-61.txt, 01-62.txt, 01-63.txt, 01-64.txt, 01-65.txt, 01-66.txt, 01-67.txt, 01-68.txt, 01-69.txt, 01-70.txt, 01-71.txt, 01-72.txt, 01-73.txt, 01-74.txt, 01-75.txt, 01-76.txt, 01-77.txt, 01-78.txt, 01-79.txt, 01-80.txt, 01-81.txt, 01-82.txt, 01-83.txt, 01-84.txt, 01-85.txt, 01-86.txt, 01-87.txt, 01-88.txt, 01-89.txt, 01-90.txt, 01-91.txt, 01-92.txt
Case Name Status Exec Time Memory
00-sample-01.txt AC 81 ms 41168 KiB
00-sample-02.txt AC 64 ms 40860 KiB
01-01.txt AC 76 ms 40772 KiB
01-02.txt AC 76 ms 40712 KiB
01-03.txt AC 69 ms 40356 KiB
01-04.txt AC 67 ms 40264 KiB
01-05.txt AC 79 ms 41140 KiB
01-06.txt AC 67 ms 40892 KiB
01-07.txt AC 73 ms 40724 KiB
01-08.txt AC 62 ms 40588 KiB
01-09.txt AC 70 ms 40500 KiB
01-10.txt AC 61 ms 40204 KiB
01-11.txt AC 70 ms 40904 KiB
01-12.txt AC 62 ms 40408 KiB
01-13.txt AC 69 ms 40960 KiB
01-14.txt AC 79 ms 41264 KiB
01-15.txt AC 72 ms 40756 KiB
01-16.txt AC 71 ms 40936 KiB
01-17.txt AC 78 ms 40716 KiB
01-18.txt AC 77 ms 41068 KiB
01-19.txt AC 66 ms 40644 KiB
01-20.txt AC 66 ms 40132 KiB
01-21.txt WA 77 ms 41040 KiB
01-22.txt WA 78 ms 40620 KiB
01-23.txt AC 79 ms 41368 KiB
01-24.txt WA 75 ms 40864 KiB
01-25.txt WA 76 ms 40780 KiB
01-26.txt WA 76 ms 40476 KiB
01-27.txt WA 79 ms 41100 KiB
01-28.txt WA 75 ms 40716 KiB
01-29.txt WA 72 ms 40480 KiB
01-30.txt WA 76 ms 40872 KiB
01-31.txt WA 74 ms 40780 KiB
01-32.txt WA 72 ms 40744 KiB
01-33.txt AC 73 ms 41040 KiB
01-34.txt WA 74 ms 41148 KiB
01-35.txt WA 73 ms 41052 KiB
01-36.txt WA 69 ms 40588 KiB
01-37.txt WA 73 ms 40856 KiB
01-38.txt WA 71 ms 41144 KiB
01-39.txt AC 70 ms 40732 KiB
01-40.txt WA 72 ms 41004 KiB
01-41.txt WA 73 ms 41120 KiB
01-42.txt AC 70 ms 40416 KiB
01-43.txt AC 73 ms 40728 KiB
01-44.txt WA 70 ms 41132 KiB
01-45.txt WA 69 ms 40996 KiB
01-46.txt WA 72 ms 40732 KiB
01-47.txt WA 72 ms 40848 KiB
01-48.txt WA 71 ms 40848 KiB
01-49.txt WA 70 ms 40992 KiB
01-50.txt AC 66 ms 40856 KiB
01-51.txt AC 63 ms 40552 KiB
01-52.txt WA 73 ms 41048 KiB
01-53.txt WA 70 ms 41012 KiB
01-54.txt WA 72 ms 41100 KiB
01-55.txt WA 70 ms 40980 KiB
01-56.txt WA 72 ms 40708 KiB
01-57.txt WA 70 ms 40844 KiB
01-58.txt WA 72 ms 41152 KiB
01-59.txt WA 72 ms 40776 KiB
01-60.txt WA 71 ms 40836 KiB
01-61.txt WA 73 ms 40764 KiB
01-62.txt WA 73 ms 41084 KiB
01-63.txt WA 71 ms 41280 KiB
01-64.txt WA 69 ms 40716 KiB
01-65.txt WA 73 ms 41124 KiB
01-66.txt WA 73 ms 41184 KiB
01-67.txt AC 69 ms 40968 KiB
01-68.txt AC 71 ms 41096 KiB
01-69.txt AC 74 ms 41180 KiB
01-70.txt AC 79 ms 40612 KiB
01-71.txt AC 78 ms 40840 KiB
01-72.txt AC 77 ms 40768 KiB
01-73.txt AC 75 ms 40872 KiB
01-74.txt AC 75 ms 40752 KiB
01-75.txt AC 71 ms 40752 KiB
01-76.txt AC 74 ms 40612 KiB
01-77.txt AC 72 ms 41016 KiB
01-78.txt AC 71 ms 41284 KiB
01-79.txt AC 71 ms 40708 KiB
01-80.txt AC 75 ms 41068 KiB
01-81.txt AC 69 ms 40696 KiB
01-82.txt AC 72 ms 41116 KiB
01-83.txt AC 72 ms 41348 KiB
01-84.txt AC 72 ms 40820 KiB
01-85.txt AC 72 ms 41088 KiB
01-86.txt AC 70 ms 40632 KiB
01-87.txt AC 70 ms 40548 KiB
01-88.txt AC 71 ms 41096 KiB
01-89.txt AC 75 ms 41004 KiB
01-90.txt AC 74 ms 41136 KiB
01-91.txt AC 76 ms 41300 KiB
01-92.txt AC 76 ms 40996 KiB