蓝桥杯2022年第十三届省赛真题-最少刷题数

题目描述

小蓝老师教的编程课有 N 名学生，编号依次是 1 . . . N。第 i 号学生这学期刷题的数量是 Ai。

对于每一名学生，请你计算他至少还要再刷多少道题，才能使得全班刷题比他多的学生数不超过刷题比他少的学生数。

输入格式

第一行包含一个正整数 N。

第二行包含 N 个整数：A1, A2, A3, . . . , AN.

输出格式

输出 N 个整数，依次表示第 1 . . . N 号学生分别至少还要再刷多少道题。

样例输入

复制

5
12 10 15 20 6

样例输出

复制

0 3 0 0 7

import java.util.Arrays;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = sc.nextInt();
        }
        sc.close();

        int[] sortarr = arr.clone();
        Arrays.sort(sortarr);

        int mid = sortarr[n / 2];
        int more = 0, less = 0;
        for (int i : arr) {
            if (i < mid) less++;
            else if (i > mid) more++;
        }

        int flag = 0, midflag = 0;
        if (more >= less) flag = 1;
        if (more > less) midflag = 1;

        for (int i = 0; i < n; i++) {  //遍历判断每个数是小于、等于还是大于mid
            if (arr[i] < mid) {
                System.out.print(mid + flag - arr[i]);    //flag=1时是case2,3  =0时是case1
            } else if (arr[i] == mid && midflag == 1) {
                System.out.print(mid + midflag - arr[i]); //midflag=1时是case2  =0时是case1,3
            } else {
                System.out.print("0");
            }

            if (i < n - 1) System.out.print(" ");
        }
    }
}