【秋招笔试题】多多的平均值

最新推荐文章于 2024-09-27 00:45:01 发布

噜噜噜鹿鹿

最新推荐文章于 2024-09-27 00:45:01 发布

阅读量121

点赞数 3

分类专栏：笔试算法题文章标签： java 开发语言

本文链接：https://blog.csdn.net/m0_52330760/article/details/142494626

版权

笔试算法题专栏收录该内容

9 篇文章 0 订阅

订阅专栏

在这里插入图片描述
解法：抽掉的两个数字之和为2倍的平均数，那么判断一下2倍的平均数是不是整数。然后在搞一个哈希表存取过的值即可。

package com.sky;

import java.util.*;

public class Test1 {

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int T = scanner.nextInt();

        while (T-- > 0) {
            int n = scanner.nextInt();
            long[] a = new long[n];
            long sum = 0;

            for (int i = 0; i < n; i++) {
                a[i] = scanner.nextLong();
                sum += a[i];
            }

            if (2 * sum % n != 0) {
                System.out.println(0);
                continue;
            }

            long target = 2 * sum / n;
            Map<Long, Integer> countMap = new HashMap<>();
            long result = 0;

            for (int i = 0; i < n; i++) {
                long complement = target - a[i];
                if (countMap.containsKey(complement)) {
                    result += countMap.get(complement);
                }
                countMap.put(a[i], countMap.getOrDefault(a[i], 0) + 1);
            }

            System.out.println(result);
        }
    }
}